import time

from typing import List, Any, Optional, Union

from fastapi import APIRouter, Depends, Request, HTTPException, Body
from sqlalchemy.orm import Session
from sqlalchemy import and_, or_

from presupuestov1 import model, schema, constant, schemas, models, search


router = APIRouter(
	prefix='/entidad',
	tags=['entidad'],
)


###############################################################################
# GET
###############################################################################

@router.get('/{entity_id:int}', response_model=schemas.entity.ClasInstitucionalIngresosGastos)
def fetch_entity(
	entity_id: int,
	db: Session = Depends(model.get_db),
):
	TreemapObjeto = models.treemap.TreemapObjeto
	gastos = db.query(TreemapObjeto).filter(
		TreemapObjeto.entidad == entity_id,
		TreemapObjeto.objeto == 0
	).all()

	ClasInstitucional = models.classifier.ClasInstitucional
	entidad = db.query(ClasInstitucional).filter(
		ClasInstitucional.entidad == entity_id
	).one()

	setattr(entidad, 'gastos', gastos)

	return entidad


@router.get('/clasificador', response_model=List[schemas.classifier.ClasInstitucional])
def fetch_classifier(
	db: Session = Depends(model.get_db),
):
	return db.query(models.classifier.ClasInstitucional).all()


@router.get('/{entity_id:int}/objetos', response_model=List[schemas.treemap.TreemapObjeto])
def fetch_entity_objetos(
	entity_id: int,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	TreemapObjeto = models.treemap.TreemapObjeto
	return models.treemap.get_entity_connections(
		entity_id, db, query, TreemapObjeto, 'objeto'
	)


@router.get('/{entity_id:int}/finfuns', response_model=List[schemas.treemap.TreemapFinFun])
def fetch_entity_finfuns(
	entity_id: int,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	TreemapFinFun = models.treemap.TreemapFinFun
	return models.treemap.get_entity_connections(
		entity_id, db, query, TreemapFinFun, 'finfun'
	)


@router.get('/{entity_id:int}/actecos', response_model=List[schemas.treemap.TreemapActeco])
def fetch_entity_actecos(
	entity_id: int,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	TreemapActeco = models.treemap.TreemapActeco
	return models.treemap.get_entity_connections(
		entity_id, db, query, TreemapActeco, 'acteco'
	)