market_control_for_reasearch_and_analytics/app/database.py

"""Database module"""

from datetime import datetime

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


def get_new_market_track(session):
    """Get ner market track"""
    market_track = MarketTrack()
    market_track.date_time = datetime.now()
    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)
    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)
    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():
        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.total_offers = len(offers)
        market_stat.amount = item_dict['amount']
        market_stat.price = item_dict['price']
        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:
        print(item_dict)
        market_stat = StateMarketStat()
        market_stat.market_track_id = market_track.id
        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