Exemplo n.º 1
0
Arquivo: dao.py Projeto: Ziem0/Library
 def load_library():
     if not os.path.isfile('static/books.csv'):
         raise FileNotFoundError
     else:
         with open('static/books.csv', 'r') as f:
             for item in f:
                 item = item.strip('\n')
                 item = item.split(',')
                 if item[0] == 'book':
                     author = item[1]
                     title = item[2]
                     book = Book(author, title)
                     book.add_copy(int(item[3]))
                     for copy in book.copies:
                         if item[5] == 'y':
                             copy.is_short_term = True
                     for copy in book.copies[:int(item[4])]:
                         copy.is_borrowed = True
                 else:
                     title = item[1]
                     number = item[2]
                     journal = Journal(title, number)
                     if item[3] == 'y':
                         journal.is_borrowed = True