def gen_books(): books_list = [] data = pd.read_csv('catalogue.csv') data = data[(data.resource_type == 'Book')] data = data.loc['title', 'author', 'code', 'year', 'total_copies', 'available_copies'].values for i in range(len(data)): obj_book = Book() obj_book.title = data[i, 0] obj_book.author = data[i, 1] obj_book.code = data[i, 2] obj_book.year = data[i, 3] obj_book.total_copies = data[i, 4] obj_book.available_copies = data[i, 5] return books_list