from collections import defaultdict
from datetime import date, datetime
from decimal import Decimal

from flask import current_app, jsonify, request, url_for
from flask_login import current_user, login_required, login_user
from sqlalchemy import extract
from sqlalchemy.exc import IntegrityError

from . import api
from .common import accessible_vehicles, boolv, fail, payload, response, role_required, station_dict, user_dict, vehicle_dict
from .. import THEMES
from ..api_tokens import create_access_token
from ..extensions import db
from ..models import AppSetting, CompanySettings, FuelCard, FuelCardPolicy, FuelCardStationRule, FuelEntry, FuelStationCompany, FuelStationPoint, OrlenPrice, User, Vehicle
from ..services import POLISH_REGIONS, aggregate_ure_companies, calculate_costs, fetch_orlen_price, fetch_orlen_range, fetch_ure_stations, invoice_period
from ..station_catalog import sync_station_catalog
from ..orlen_sync import sync_orlen_prices

FUEL_TYPES = ("PB95", "PB98", "DIESEL", "LPG")

@api.get('/orlen/prices')
@login_required
def orlen_prices():
    year=request.args.get('year',type=int) or date.today().year; fuels=[x.upper() for x in request.args.getlist('fuel') if x.upper() in FUEL_TYPES] or ['PB95']; regions=[x.lower() for x in request.args.getlist('region') if x.lower() in POLISH_REGIONS]
    rows=OrlenPrice.query.filter(OrlenPrice.fuel_type.in_(fuels),extract('year',OrlenPrice.effective_date)==year).order_by(OrlenPrice.effective_date).all()
    return response([{'fuel_type':r.fuel_type,'date':r.effective_date.isoformat(),'price_per_liter':float(r.price_per_liter),'region':r.region,'source':r.source} for r in rows if r.fuel_type!='LPG' or not regions or r.region in regions])

@api.post('/orlen/sync')
@role_required('boss','admin')
def orlen_sync():
    d = payload()
    fuels = d.get('fuels', [])
    fuels = [fuels] if isinstance(fuels, str) else fuels
    fuels = [fuel.upper() for fuel in fuels if fuel.upper() in FUEL_TYPES]
    regions = d.get('regions', [])
    regions = [regions] if isinstance(regions, str) else regions
    regions = [str(region).strip().lower() for region in regions if str(region).strip().lower() in POLISH_REGIONS]
    if not fuels:
        return fail('Wybierz co najmniej jedno paliwo')

    try:
        year = int(d.get('year', date.today().year))
        settings = CompanySettings.query.first()
        result, total = sync_orlen_prices(
            fuels=fuels,
            year=year,
            regions=regions,
            default_region=settings.region,
            full_refresh=boolv(d, 'full_refresh'),
        )
        db.session.commit()
        return response(result, f'Przetworzono {total} rekordów')
    except Exception as exc:
        db.session.rollback()
        return fail(str(exc), 502)

@api.get('/orlen/preview')
@login_required
def orlen_preview():
    try:return response(fetch_orlen_price(request.args['fuel_type'],datetime.fromisoformat(request.args['date']).date(),request.args.get('region') or CompanySettings.query.first().region))
    except Exception as exc:return fail(str(exc),502)