Loop through release files properly

This commit is contained in:
Ruud committed 2014-05-11 17:40:55 +02:00
1 parent 0f8ab05fd4
commit 5a2a9bbf9a
1 file changed
+10 -5
+10 -5
View File
@@ -180,15 +180,20 @@ class Release(Plugin):
try: try:
db = get_db() db = get_db()
rel = db.get('id', release_id) rel = db.get('id', release_id)
raw_files = rel.get('files')
if len(rel.get('files')) == 0: if len(raw_files) == 0:
self.delete(rel['_id']) self.delete(rel['_id'])
else: else:
files = [] files = {}
for release_file in rel.get('files'): for file_type in raw_files:
if os.path.isfile(ss(release_file['path'])):
files.append(release_file) for release_file in raw_files.get(file_type, []):
if os.path.isfile(ss(release_file)):
if file_type not in files:
files[file_type] = []
files[file_type].append(release_file)
rel['files'] = files rel['files'] = files
db.update(rel) db.update(rel)