from datetime import datetime
from enum import Enum

from api import PortfoliosRisk, PortfoliosType
from framework import read, write, where, format_date, mapper_columns

__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 = mapper_columns(datas=datas, columns=__COLUMNS__)
    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)} 
    '''