def load_file(name): file_reader = csv.reader(open(path + name), delimiter=',') stock = Stock(name.split('.')[0], True) stock.save_to_db() bought_stocks1 = BoughtStocks(user1.id, stock.id, 1.0) bought_stocks2 = BoughtStocks(user2.id, stock.id, 1.0) bought_stocks3 = BoughtStocks(user3.id, stock.id, 1.0) bought_stocks1.save_to_db() bought_stocks2.save_to_db() bought_stocks3.save_to_db() for idx, row in enumerate(file_reader): if idx == 0: continue stock_data = StockData(stock.id, row[0], row[1], row[2], row[3], row[4], row[5], row[6]) stock_data.save_to_db()