from datetime import datetime from enum import Enum from api import PortfoliosRisk, PortfoliosType from framework import read, write, where, format_date __COLUMNS__ = { 'rmp_id': 'id', 'rmp_date': 'date', 'rmp_risk': 'risk', 'rmp_type': 'type', 'rmp_rolve': 'solve', 'rmp_portfolio': 'portfolio', 'rmp_cvar': 'cvar', } @write def insert(datas): datas = {x[0]: datas[x[1]] for x in __COLUMNS__.items() if x[1] in datas and datas[x[1]] is not None} datas = { **datas, **{x[0]: format_date(x[1]) for x in datas.items() if isinstance(x[1], datetime)}, **{x[0]: x[1].value for x in datas.items() if isinstance(x[1], Enum)} } return f''' insert into robo_mpt_portfolios({','.join([x for x in datas.keys()])}) values ({','.join([f"'{x[1]}'" for x in datas.items()])}) ''' @read(one=True) def get_one(day, type: PortfoliosType, risk: PortfoliosRisk): return f''' select {','.join([f"{x[0]} as {x[1]}" for x in __COLUMNS__.items()])} from robo_mpt_portfolios {where(rmp_date=day, rmp_risk=risk, rmp_type=type)} '''