fixed tests not working

This commit is contained in:
ajurna committed 2019-02-07 15:17:28 +00:00
1 parent bc6504088f
commit e52ba3f393
2 files changed
+27 -29

No files matched your search

+6
View File
@@ -117,6 +117,9 @@ class DirFile:
self.label = generate_label(comic, status)
self.type = 'book'
def __repr__(self):
return f'<DirFile: {self.name}: {self.type}>'
def generate_directory(user, directory=False):
"""
@@ -147,6 +150,7 @@ def generate_directory(user, directory=False):
df.populate_directory(directory_obj, user)
files.append(df)
dir_list.remove(directory_obj.name)
with atomic():
for file_obj in file_list_obj:
df = DirFile()
@@ -169,6 +173,8 @@ def generate_directory(user, directory=False):
df = DirFile()
df.populate_comic(book, user)
files.append(df)
files.sort(key=lambda x: x.name)
files.sort(key=lambda x: x.type, reverse=True)
return files