"""Database module"""

from datetime import datetime

from app import SESSION, functions, MAX_OFFER
from app.models import Region, Player, MarketTrack, StateMarketStat, PlayerMarketStat


def get_new_market_track(session, resources, state_resources, items):
    """Get ner market track"""
    market_track = MarketTrack()
    market_track.date_time = datetime.now()
    market_track.resources = resources
    market_track.state_resources = state_resources
    market_track.items = items
    session.add(market_track)
    return market_track

def save_resource_market(player_market, state_market):
    """Save factories to database"""
    session = SESSION()

    market_track = get_new_market_track(session, True, True, False)
    market_track.player_resources = True
    market_track.state_resources = True
    _save_player_market(session, market_track, player_market)
    _save_state_market(session, market_track, state_market)

    session.commit()
    session.close()

def save_player_market(market):
    """Save player market"""
    session = SESSION()
    market_track = get_new_market_track(session, True, False, False)
    session.add(market_track)
    _save_player_market(session, market_track, market)

def _save_player_market(session, market_track, market):
    """Save player market to database"""
    for item_type, offers in market.items():
        if not offers:
            continue
        item_dict = offers[0]
        market_stat = PlayerMarketStat()
        player = session.query(Player).get(item_dict['player_id'])
        if not player:
            player = save_player(session, item_dict)
        market_stat.player_id = player.id
        market_stat.item_type = item_type
        market_stat.amount = item_dict['amount']
        market_stat.price = item_dict['price']

        market_stat.total_offers = len(offers)
        one_t_average = functions.calculate_purchage_amount(offers, 1e12) / 100
        base_average = 3 * MAX_OFFER[item_type] * one_t_average
        market_stat.base_average = base_average
        market_stat.one_average = functions.calculate_purchage_amount(offers, 1 * base_average)
        market_stat.two_average = functions.calculate_purchage_amount(offers, 2 * base_average)
        market_stat.three_average = functions.calculate_purchage_amount(offers, 3 * base_average)
        market_stat.four_average = functions.calculate_purchage_amount(offers, 4 * base_average)
        market_stat.five_average = functions.calculate_purchage_amount(offers, 5 * base_average)
        market_stat.six_average = functions.calculate_purchage_amount(offers, 6 * base_average)
        market_stat.seven_average = functions.calculate_purchage_amount(offers, 7 * base_average)
        market_stat.eight_average = functions.calculate_purchage_amount(offers, 8 * base_average)
        market_stat.nine_average = functions.calculate_purchage_amount(offers, 9 * base_average)
        market_stat.ten_average = functions.calculate_purchage_amount(offers, 10 * base_average)

        market_stat.market_track_id = market_track.id
        session.add(market_stat)

def _save_state_market(session, market_track, market):
    """Save state market"""
    for item_dict in market:
        market_stat = StateMarketStat()
        region = session.query(Region).get(item_dict['region_id'])
        if not region:
            region = save_region(session, item_dict)
        market_stat.region_id = region.id
        market_stat.market_track_id = market_track.id
        market_stat.item_type = item_dict['item_type'] - 1000
        market_stat.amount = item_dict['amount']
        market_stat.price = item_dict['price']
        session.add(market_stat)

def save_player(session, item_dict):
    """Save player to database"""
    player = Player()
    player.id = item_dict['player_id']
    player.name = item_dict['player_name']
    session.add(player)
    return player

def save_region(session, item_dict):
    """Save player to database"""
    region = Region()
    region.id = item_dict['region_id']
    region.name = item_dict['region_name']
    session.add(region)
    return region