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)} 
    '''