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

added argument to return filtered book list

Browse Source
This commit is contained in:
Mike
2019-09-26 23:33:32 -04:00
parent 5d26a7d49f
commit bb06acd687
5 changed files with 10 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
library.py
View File

@@ -15,8 +15,7 @@ class Catalogue:
self.catalogue = json.load(f)
self.current_files = self.scan_folder()
except Exception:
with open(config.book_shelf, 'w') as f:
json.dump(self.filter_books(), f)
self.filter_books()
def scan_folder(self, folder=config.book_path):
for f in os.listdir(folder):
@@ -27,12 +26,15 @@ class Catalogue:
self.file_list.append(self.scan_folder(_path))
self.file_list.append(_path)
def filter_books(self):
scan = self.scan_folder()
def filter_books(self, ret=0):
self.scan_folder()
regx = re.compile(r"\.epub")
self.books = list(filter(regx.search, filter(None, self.file_list)))
with open(config.book_shelf, 'w') as f:
json.dump(self.books, f)
if ret == 0:
with open(config.book_shelf, 'w') as f:
json.dump(self.books, f)
else:
return self.books
def compare_shelf_current(self):
try: