import time

from typing import List, Any, Optional, Union

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

from presupuestov1 import model, schema, constant, schemas, models, search
from presupuestov1.routers._decorators import not_found


router_entidad = APIRouter(
	prefix='/entidad',
	tags=['entidad'],
)
router_ubigeo = APIRouter(
	prefix='/ubigeo',
	tags=['ubigeo'],
)
router_da = APIRouter(
	prefix='/direccion_administrativa',
	tags=['direccion_administrativa'],
)


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

# clas

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


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



# base

@router_entidad.get('/{entity_id:str}', response_model=schemas.entity.ClasInstitucionalIngresosGastos)
@not_found
def fetch_entity(
	entity_id: str,
	db: Session = Depends(model.get_db),
):
	ClasInstitucional = models.classifier.ClasInstitucional
	entidad = db.query(ClasInstitucional).filter(
		ClasInstitucional.entidad == entity_id
	).one()

	gastos_ingresos = get_resumen(db, 'entidad', entity_id)
	setattr(entidad, 'gastos_ingresos', gastos_ingresos)

	return entidad


@router_ubigeo.get('/{ubigeo_id:str}', response_model=schemas.entity.ClasGeograficoIngresosGastos)
@not_found
def fetch_entity(
	ubigeo_id: str,
	db: Session = Depends(model.get_db),
):
	ClasGeografico = models.classifier.ClasGeografico
	ubigeo = db.query(ClasGeografico).filter(
		ClasGeografico.ubigeo == ubigeo_id
	).one()

	gastos_ingresos = get_resumen(db, 'municipio_ubigeo', ubigeo_id)
	setattr(ubigeo, 'gastos_ingresos', gastos_ingresos)

	return ubigeo


# objetos

@router_entidad.get('/{entity_id:str}/objetos', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_objetos(
	entity_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'entidad', entity_id, 'objeto', query)


@router_ubigeo.get('/{ubigeo_id:str}/objetos', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_objetos(
	ubigeo_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'municipio_ubigeo', ubigeo_id, 'objeto', query)


@router_entidad.get('/{entity_id:str}/finfuns', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_finfuns(
	entity_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'entidad', entity_id, 'finfun', query)


@router_ubigeo.get('/{ubigeo_id:str}/finfuns', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_finfuns(
	ubigeo_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'municipio_ubigeo', ubigeo_id, 'finfun', query)


@router_entidad.get('/{entity_id:str}/actecos', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_actecos(
	entity_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'entidad', entity_id, 'acteco', query)


@router_ubigeo.get('/{ubigeo_id:str}/actecos', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_actecos(
	ubigeo_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'municipio_ubigeo', ubigeo_id, 'acteco', query)


@router_entidad.get('/{entity_id:str}/rubros', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_rubros(
	entity_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'entidad', entity_id, 'rubro', query)


@router_entidad.get('/{entity_id:str}/organismos', response_model=List[schemas.entity.EntidadDistribucion])
def fetch_entity_organismos(
	entity_id: str,
	db: Session = Depends(model.get_db),
	query: schema.QueryParams = Depends(),
):
	return get_distribuciones(db, 'entidad', entity_id, 'organismo', query)


###############################################################################
# private
###############################################################################

@not_found
def get_distribuciones(db, tipo_codigo, entity_id, dimension, query):
	EntidadDistribucionesSchema = schemas.entity.EntidadDistribucion
	EntidadDistribuciones = models.entity.EntidadDistribuciones

	schema_fields = EntidadDistribucionesSchema.model_fields.keys()
	cols = [getattr(EntidadDistribuciones, name) for name in schema_fields]

	filters = []
	if query.gestion is not None:
		filters.append(EntidadDistribuciones.gestion == query.gestion)

	filters.extend([
		EntidadDistribuciones.tipo_codigo == tipo_codigo,
		EntidadDistribuciones.codigo == entity_id,
		EntidadDistribuciones.dimension == dimension,
	])

	return db.execute(
		select(*cols).where(*filters)
	).mappings().all()


def get_resumen(db, tipo_codigo, codigo):
	EntidadResumenes = models.entity.EntidadResumenes
	EntidadResumenSchema = schemas.entity.EntidadResumen

	schema_fields = EntidadResumenSchema.model_fields.keys()
	cols = [getattr(EntidadResumenes, name) for name in schema_fields]

	return db.execute(
		select(*cols).where(
			EntidadResumenes.tipo_codigo == tipo_codigo,
			EntidadResumenes.codigo == codigo,
		)
	).all()