th3r00t/pyShelf
1
0
Fork 0
mirror of https://github.com/th3r00t/pyShelf.git synced 2026-04-28 01:59:35 -04:00
Code Issues Packages Projects Releases Wiki Activity

Adjust file extension grabbing

Browse Source
This commit is contained in:
th3r00t
2020-02-01 23:17:26 -05:00
parent 9edf182c2b
commit 8767dd4c26
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/lib/library.py
View File

@@ -56,7 +56,7 @@ class Catalogue:
:returns self._book_list_expanded: json string containing all book metadata
"""
self.scan_folder()
self.scan_folder() # Populate file list
regx = re.compile(r"\.epub")
try:
self.books = list(filter(regx.search, filter(None, self.file_list)))

3
src/interface/views.py
View File

@@ -123,4 +123,5 @@ def hr_name(book):
"""
Nicer file names
"""
return "{0}.{1}".format(slugify(book.title), book.file_name.split(".")[1])
# return "{0}.{1}".format(slugify(book.title), book.file_name.split(".")[1])
return "{0}.{1}".format(slugify(book.title), os.path.splitext(book.file_name.split))