Implemented the return of unique files for storage

This commit is contained in:
Raelon Masters
2019-10-14 17:52:16 -04:00
parent a1bb0aa101
commit f821eb02cb
15 changed files with 87 additions and 76 deletions

View File

@@ -2,10 +2,12 @@
import os
import zipfile
from config import Config
from library import Catalogue
from storage import Storage
from lib.library import Catalogue
from lib.storage import Storage
config = Config()
Storage = Storage()
class InitFiles:
"""First run file creation operations"""
def __init__(self, file_array):
@@ -18,7 +20,7 @@ class InitFiles:
"""Create the file"""
if not os.path.isdir(os.path.split(_pointer)[0]):
os.mkdir(os.path.split(_pointer)[0])
f = open(_pointer, "w+")
f = open(_pointer, "w+")
f.close()
@@ -29,8 +31,9 @@ class Epub:
self.book_path = config.book_path
self.Catalogue = Catalogue()
def import_books(self):
book_list = self.Catalogue.filter_books()
def import_books(self, list=None):
if list is not None: book_list = list
else: book_list = self.Catalogue.filter_books()
for book in book_list:
extracted = self.Catalogue.extract_metadata(book_list[book])
Storage.insert_book(extracted)