예제 #1
0
파일: data_utils.py 프로젝트: LEVLLN/trades
def save_trades(element_list, stock_code, insider_code):
    for element in element_list:
        trade = Trade()
        trade.stock = stock_code
        trade.insider = insider_code
        trade.relation = element[2]
        trade_last_date = datetime.date(
            datetime.strptime(element[3], '%m/%d/%Y'))
        trade.last_date = trade_last_date
        trade.transaction_type = element[4]
        trade.owner_type = element[5]
        shares_traded = element[6].replace(',', '')
        if shares_traded == '':
            shares_traded = 0
        trade.shares_traded = shares_traded
        last_price = element[7]
        if last_price == '':
            last_price = 0.0
        trade.last_price = last_price
        shares_held = element[8].replace(',', '')
        if shares_held == '':
            shares_held = 0
        trade.shares_held = shares_held
        try:
            trade.save()
            logger.info(
                f'Trade object of {stock_code} on last date {element[2]} saved in db'
            )
        except Exception as exception:
            logger.error(f'{stock_code} is allready exist: {exception}')
            db.rollback()