| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100 |
- """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
|
