def testGetThirdChapter(self):
nb = Notabenoid('19980')
for l in nb.get_chapter(3):
pass
self.assertEqual(l,
'.\r\n',
'wrong first chapter')
def testGetBookInfoByID(self):
nb = Notabenoid('19980')
self.assertEqual(nb.get_book_name(),
'A DANCE WITH DRAGONS / ТАНЕЦ С ДРАКОНАМИ',
'wrong book name receive')
self.assertEqual(nb.get_book_img(),
'http://notabenoid.com/i//book/1/19980-45199.jpg',
'wrong book image url receive')
class Notabenoid2FB2():
'''
Class for creation FB2 from Notabenoid.
'''
def __init__(self, book_id):
'''
Constructor
'''
self.nb = Notabenoid(book_id)
name = self.nb.get_book_name()
name = name.split('/')[0]
name = name.strip()
name = name.replace(' ', '_')
name = name.lower()
self.file_name = name
fb2 = FB2Creator(name)
fb2.set_generator(self.nb.content())
fb2.set_notifier(self.print_chapter)
fb2.set_end_of_chapter_symbol(self.nb.end_of_chapter())
self.fb2 = fb2
self.current_chapter = 0
def print_chapter(self, s):
'''
Print log for each chapter.
'''
self.current_chapter += 1
print(s + ' ' + str(self.current_chapter))
def get_file_name(self):
'''
Get output file name.
'''
return self.fb2.get_file_name()
def generate(self):
'''
Generate book.
'''
self.fb2.create_file()
def get_chapter_number(self):
'''
Return number of chapter in book.
'''
return self.nb.get_chapter_number()
def set_notifier(self, n):
'''
Set notifier called for each chapter.
'''
self.fb2.set_notifier(n)
def __init__(self, book_id):
'''
Constructor
'''
self.nb = Notabenoid(book_id)
name = self.nb.get_book_name()
name = name.split('/')[0]
name = name.strip()
name = name.replace(' ', '_')
name = name.lower()
self.file_name = name
fb2 = FB2Creator(name)
fb2.set_generator(self.nb.content())
fb2.set_notifier(self.print_chapter)
fb2.set_end_of_chapter_symbol(self.nb.end_of_chapter())
self.fb2 = fb2
self.current_chapter = 0
def testGetChapterNumber(self):
nb = Notabenoid('19980')
self.assertEqual(nb.get_chapter_number(),
75,
'wrong chapter number')
def testGetFirstLine(self):
nb = Notabenoid('19980')
self.assertEqual(nb.content().__next__(),
'ПОЯСНЕНИЯ К ХРОНОЛОГИИ\r\n',
'wrong first book line')
def testGetChapter(self):
nb = Notabenoid('19980')
links = nb.get_chapters_links()
self.assertEqual(links[0],
'/book/19980/65224/ready',
'wrong chapters link receive')
