예제 #1
0
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()