ajurna/cbwebreader
1
0
Fork 0
mirror of https://github.com/ajurna/cbwebreader.git synced 2025-12-06 06:17:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix for issue #2.

Browse Source
This commit is contained in:
Ajurna
2017-11-22 15:56:39 +00:00
parent 28a2f29f5e
commit c330c6812b
5 changed files with 52 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
comic/util.py
View File

@@ -98,16 +98,24 @@ class DirFile:
self.type = 'directory'
def populate_comic(self, comic, user):
self.icon = 'glyphicon-book'
self.name = comic.file_name
status, created = ComicStatus.objects.get_or_create(comic=comic, user=user)
if created:
status.save()
self.selector = urlsafe_base64_encode(comic.selector.bytes).decode()
self.location = '/comic/read/{0}/{1}/'.format(self.selector,
status.last_read_page)
self.label = generate_label(comic, status)
self.type = 'book'
if type(comic) == str:
self.icon = 'glyphicon-remove'
self.name = comic
self.selector = '0'
self.location = '/'
self.label = '<center><span class="label label-danger">Error</span></center>'
self.type = 'book'
else:
self.icon = 'glyphicon-book'
self.name = comic.file_name
status, created = ComicStatus.objects.get_or_create(comic=comic, user=user)
if created:
status.save()
self.selector = urlsafe_base64_encode(comic.selector.bytes).decode()
self.location = '/comic/read/{0}/{1}/'.format(self.selector,
status.last_read_page)
self.label = generate_label(comic, status)
self.type = 'book'
def generate_directory(user, directory=False):