rr_functions/calculate_loot.py

"""Test file for construction costs"""

from rival_regions_calc import ConstructionCosts, Building

ZUID_BUILDINGS = {
    'hospital': 1208,
    'military base': 815,
    'school': 775,
    'missile system': 1250,
    'sea port': 978,
    'power plant': 1400,
    'spaceport': 3,
    'airport': 1100,
    'house fund': 4050,
}

OOST_BUILDINGS = {
    'hospital': 1226,
    'military base': 813,
    'school': 775,
    'missile system': 4550,
    'sea port': 2475,
    'power plant': 1461,
    'spaceport': 201,
    'airport': 3750,
    'house fund': 3795,
}

WEST_BUILDINGS = {
    # 'hospital': 2426,
    # 'military base': 1640,
    # 'school': 1325,
    # 'missile system': 1500,
    # 'sea port': 1250,
    # 'power plant': 2500,
    # 'spaceport': 601,
    # 'airport': 1250,
    # 'house fund': 7765,
    # 'hospital': 1213,
    # 'military base': 820,
    # 'school': 663,
    # 'missile system': 720,
    # 'sea port': 625,
    # 'power plant': 1250,
    # 'spaceport': 301,
    # 'airport': 625,
    # 'house fund': 3883,
#    'hospital': 2572,
#    'military base': 1701,
#    'school': 1616,
#    'missile system': 1500,
#    'sea port': 2000,
#    'power plant': 2500,
#    'spaceport': 1115,
#    'airport': 2000,
#    'house fund': 8371,
    'military academy': 48,
    'hospital': 2465,
    'military base': 617,
    'school': 1580,
    'missile system': 550,
    'power plant': 1450,
    'spaceport': 50,
    'airport': 500,
    'house fund': 8242,
}

def main():
    """Main method"""
    # print('oost')
    # calculate(OOST_BUILDINGS)
    # print('zuid')
    # calculate(ZUID_BUILDINGS)
    print('west')
    calculate(WEST_BUILDINGS)

def calculate(buildings):
    """calculate and print resources"""
    print('cash,gold,oil,ore,uranium,diamond')
    for building_name, amount in buildings.items():
        amount = int(amount/4)
        building = Building(building_name)
        construction_costs = ConstructionCosts(building, amount)
        construction_costs.calculate(amount)
        print('{},{},{},{},{},{}'.format(
            construction_costs.cash,
            construction_costs.gold,
            construction_costs.oil,
            construction_costs.ore,
            construction_costs.uranium,
            construction_costs.diamond,
        ))


def bucks(integer):
    """Format number"""
    return '{:,}'.format(integer).replace(',', '.')

if __name__ == "__main__":
    main()