def test_series_fn(self):
""" Тестирование функций addseries, addbseries """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertEqual(book.series.count(), 1)
ser = book.series.all()[0]
self.assertEqual(ser.ser, "mywork")
self.assertEqual(bseries.objects.get(ser=ser).ser_no, 1)
def inpx_callback(self, inpx, inp, meta_data):
name = "%s.%s" % (meta_data[inpx_parser.sFile],
meta_data[inpx_parser.sExt])
lang = meta_data[inpx_parser.sLang].strip(strip_symbols)
title = meta_data[inpx_parser.sTitle].strip(strip_symbols)
annotation = ''
docdate = meta_data[inpx_parser.sDate].strip(strip_symbols)
rel_path_current = os.path.join(self.rel_path,
meta_data[inpx_parser.sFolder])
if opdsdb.findbook(name, rel_path_current, 1) == None:
cat = opdsdb.addcattree(rel_path_current, opdsdb.CAT_INP)
book = opdsdb.addbook(name, rel_path_current, cat,
meta_data[inpx_parser.sExt], title,
annotation, docdate, lang,
meta_data[inpx_parser.sSize], opdsdb.CAT_INP)
self.books_added += 1
self.books_in_archives += 1
self.logger.debug("Book " + rel_path_current + "/" + name +
" Added ok.")
for a in meta_data[inpx_parser.sAuthor]:
author = opdsdb.addauthor(a.replace(',', ' '))
opdsdb.addbauthor(book, author)
for g in meta_data[inpx_parser.sGenre]:
opdsdb.addbgenre(
book, opdsdb.addgenre(g.lower().strip(strip_symbols)))
for s in meta_data[inpx_parser.sSeries]:
ser = opdsdb.addseries(s.strip())
opdsdb.addbseries(book, ser, 0)
def test_author_fn(self):
""" Тестирование функций addauthor, addbauthor """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertEqual(book.authors.count(), 1)
self.assertEqual(
book.authors.get(full_name="Test Author").search_full_name,
"TEST AUTHOR")
def inpx_callback(self, inpx, inp, meta_data):
name = "%s.%s"%(meta_data[inpx_parser.sFile],meta_data[inpx_parser.sExt])
lang=meta_data[inpx_parser.sLang].strip(strip_symbols)
title=meta_data[inpx_parser.sTitle].strip(strip_symbols)
annotation=''
docdate=meta_data[inpx_parser.sDate].strip(strip_symbols)
rel_path_current = os.path.join(self.rel_path,meta_data[inpx_parser.sFolder])
if opdsdb.findbook(name, rel_path_current, 1) == None:
cat = opdsdb.addcattree(rel_path_current, opdsdb.CAT_INP)
book=opdsdb.addbook(name,rel_path_current,cat,meta_data[inpx_parser.sExt],title,annotation,docdate,lang,meta_data[inpx_parser.sSize],opdsdb.CAT_INP)
self.books_added+=1
self.books_in_archives+=1
self.logger.debug("Book "+rel_path_current+"/"+name+" Added ok.")
for a in meta_data[inpx_parser.sAuthor]:
author=opdsdb.addauthor(a.replace(',',' '))
opdsdb.addbauthor(book,author)
for g in meta_data[inpx_parser.sGenre]:
opdsdb.addbgenre(book,opdsdb.addgenre(g.lower().strip(strip_symbols)))
for s in meta_data[inpx_parser.sSeries]:
ser=opdsdb.addseries(s.strip())
opdsdb.addbseries(book,ser,0)
def test_series_fn(self):
""" Тестирование функций addseries, addbseries """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertEqual(book.series.count(), 1)
ser = book.series.all()[0]
self.assertEqual(ser.ser, "mywork")
self.assertEqual(bseries.objects.get(ser=ser).ser_no, 1)
def test_genre_fn(self):
""" Тестирование функций addgenre, addbgenre """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertEqual(book.genres.count(), 1)
self.assertEqual(
book.genres.get(genre="fantastic").section, opdsdb.unknown_genre)
self.assertEqual(
book.genres.get(genre="fantastic").subsection, "fantastic")
def processfile(self,name,full_path,file,cat,archive=0,file_size=0):
(n, e) = os.path.splitext(name)
if e.lower() in config.SOPDS_BOOK_EXTENSIONS.split():
rel_path=os.path.relpath(full_path,config.SOPDS_ROOT_LIB)
self.logger.debug("Attempt to add book "+rel_path+"/"+name)
try:
if opdsdb.findbook(name, rel_path, 1) == None:
if archive==0:
cat=opdsdb.addcattree(rel_path,archive)
try:
book_data = create_bookfile(file, name)
except Exception as err:
book_data = None
self.logger.warning(rel_path + ' - ' + name + ' Book parse error, skipping... (Error: %s)'%err)
self.bad_books += 1
if book_data:
lang = book_data.language_code.strip(strip_symbols) if book_data.language_code else ''
title = book_data.title.strip(strip_symbols) if book_data.title else n
annotation = book_data.description if book_data.description else ''
annotation = annotation.strip(strip_symbols) if isinstance(annotation, str) else annotation.decode('utf8').strip(strip_symbols)
docdate = book_data.docdate if book_data.docdate else ''
book=opdsdb.addbook(name,rel_path,cat,e[1:],title,annotation,docdate,lang,file_size,archive)
self.books_added+=1
if archive!=0:
self.books_in_archives+=1
self.logger.debug("Book "+rel_path+"/"+name+" Added ok.")
for a in book_data.authors:
author_name = a.get('name',_('Unknown author')).strip(strip_symbols)
# Если в имени автора нет запятой, то фамилию переносим из конца в начало
if author_name and author_name.find(',')<0:
author_names = author_name.split()
author_name = ' '.join([author_names[-1],' '.join(author_names[:-1])])
author=opdsdb.addauthor(author_name)
opdsdb.addbauthor(book,author)
for genre in book_data.tags:
opdsdb.addbgenre(book,opdsdb.addgenre(genre.lower().strip(strip_symbols)))
if book_data.series_info:
ser = opdsdb.addseries(book_data.series_info['title'])
ser_no = book_data.series_info['index'] or '0'
ser_no = int(ser_no) if ser_no.isdigit() else 0
opdsdb.addbseries(book,ser,ser_no)
else:
self.books_skipped+=1
self.logger.debug("Book "+rel_path+"/"+name+" Already in DB.")
except UnicodeEncodeError as err:
self.logger.warning(rel_path + ' - ' + name + ' Book UnicodeEncodeError error, skipping... (Error: %s)' % err)
self.bad_books += 1
def processfile(self,name,full_path,file,cat,archive=0,file_size=0):
(n, e) = os.path.splitext(name)
if e.lower() in config.SOPDS_BOOK_EXTENSIONS.split():
rel_path=os.path.relpath(full_path,config.SOPDS_ROOT_LIB)
self.logger.debug("Attempt to add book "+rel_path+"/"+name)
try:
if opdsdb.findbook(name, rel_path, 1) == None:
if archive==0:
cat=opdsdb.addcattree(rel_path,archive)
try:
book_data = create_bookfile(file, name)
except Exception as err:
book_data = None
self.logger.warning(rel_path + ' - ' + name + ' Book parse error, skipping... (Error: %s)'%err)
self.bad_books += 1
if book_data:
lang = book_data.language_code.strip(strip_symbols) if book_data.language_code else ''
title = book_data.title.strip(strip_symbols) if book_data.title else n
annotation = book_data.description if book_data.description else ''
annotation = annotation.strip(strip_symbols) if isinstance(annotation, str) else annotation.decode('utf8').strip(strip_symbols)
docdate = book_data.docdate if book_data.docdate else ''
book=opdsdb.addbook(name,rel_path,cat,e[1:],title,annotation,docdate,lang,file_size,archive)
self.books_added+=1
if archive!=0:
self.books_in_archives+=1
self.logger.debug("Book "+rel_path+"/"+name+" Added ok.")
for a in book_data.authors:
author_name = a.get('name',_('Unknown author')).strip(strip_symbols)
# Если в имени автора нет запятой, то фамилию переносим из конца в начало
if author_name and author_name.find(',')<0:
author_names = author_name.split()
author_name = ' '.join([author_names[-1],' '.join(author_names[:-1])])
author=opdsdb.addauthor(author_name)
opdsdb.addbauthor(book,author)
for genre in book_data.tags:
opdsdb.addbgenre(book,opdsdb.addgenre(genre.lower().strip(strip_symbols)))
if book_data.series_info:
ser = opdsdb.addseries(book_data.series_info['title'])
ser_no = book_data.series_info['index'] or '0'
ser_no = int(ser_no) if ser_no.isdigit() else 0
opdsdb.addbseries(book,ser,ser_no)
else:
self.books_skipped+=1
self.logger.debug("Book "+rel_path+"/"+name+" Already in DB.")
except UnicodeEncodeError as err:
self.logger.warning(rel_path + ' - ' + name + ' Book UnicodeEncodeError error, skipping... (Error: %s)' % err)
self.bad_books += 1
def test_book_fn(self):
""" Тестирование функций addbook, findbook """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertIsNotNone(book)
self.assertEqual(book.filename, "testbook.fb2")
self.assertEqual(book.path, "root/child")
self.assertEqual(book.catalog.cat_name, "child")
self.assertEqual(book.catalog.cat_type, 0)
self.assertEqual(book.format, ".fb2")
self.assertEqual(book.title, "Test Book")
self.assertEqual(book.annotation, "Annotation")
self.assertEqual(book.docdate, "01.01.2016")
self.assertEqual(book.lang, "ru")
self.assertEqual(book.filesize, 500)
self.assertEqual(book.cat_type, 0)
def test_book_fn(self):
""" Тестирование функций addbook, findbook """
book = opdsdb.findbook("testbook.fb2","root/child")
self.assertIsNotNone(book)
self.assertEqual(book.filename,"testbook.fb2")
self.assertEqual(book.path, "root/child")
self.assertEqual(book.catalog.cat_name, "child")
self.assertEqual(book.catalog.cat_type, 0)
self.assertEqual(book.format, ".fb2")
self.assertEqual(book.title, "Test Book")
self.assertEqual(book.annotation, "Annotation")
self.assertEqual(book.docdate, "01.01.2016")
self.assertEqual(book.lang, "ru")
self.assertEqual(book.filesize, 500)
self.assertEqual(book.cat_type, 0)
def test_genre_fn(self):
""" Тестирование функций addgenre, addbgenre """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertEqual(book.genres.count(), 1)
self.assertEqual(book.genres.get(genre="fantastic").section, opdsdb.unknown_genre)
self.assertEqual(book.genres.get(genre="fantastic").subsection, "fantastic")
def test_author_fn(self):
""" Тестирование функций addauthor, addbauthor """
book = opdsdb.findbook("testbook.fb2", "root/child")
self.assertEqual(book.authors.count(), 1)
self.assertEqual(book.authors.get(full_name="Test Author").search_full_name, "TEST AUTHOR")
