simco-construction-calc/main.py

90 lines
3 KiB
Python
Raw Normal View History

import os
import csv
import pickle
from pprint import pprint
try:
from wget import download
except ImportError as e:
print('Error: Please install wget using "pip install wget"')
exit(1)
2022-12-23 19:09:23 -06:00
def data_download(buildings_file_name, debug=False):
if 'data' not in os.listdir():
os.mkdir('data')
os.chdir('data')
if debug:
print(os.listdir())
# Buildings download
if f'{buildings_file_name[:-4]}.pickle' in os.listdir():
pickle.load(open(f'{buildings_file_name[:-4]}.pickle', 'rb'))
elif buildings_file_name not in os.listdir():
if debug:
print('Downloading building info...')
download(
'https://docs.google.com/spreadsheets/d/16J269YAFTVy_IPuzGUXfV_4-Rplzk1LVk7YpsvDcEVY/export?format=tsv',
buildings_file_name)
if debug:
print('Building info downloaded')
else:
print('Building info already downloaded, but not pickled')
# Buildings tsv processing
with open(buildings_file_name) as file:
reader = csv.reader(file, delimiter='\t') # Changes delimiter to tab for .tsv files
buildings = {}
for row in reader:
2022-12-23 19:09:23 -06:00
if row[0] != 'Building' and row[0] != '':
temp = list(row)
buildings[temp[0].lower()] = [row[i].lower() for i in range(1, len(row))]
pickle.dump(buildings, open(f'{buildings_file_name[:-4]}.pickle', 'wb'))
return buildings
os.chdir('..')
print('Downloading and processing buildings...')
2022-12-23 19:09:23 -06:00
buildings = data_download('buildings.tsv') # In format {building_name: [brick, plank, rc, cu]}
print('Buildings downloaded and processed')
print()
2022-12-23 19:09:23 -06:00
# Can't keep track of hours because it varies by building, and it would be such a pain to figure out
total = [0, 0, 0, 0] # Number of [rc, bricks, planks, cu] needed for upgrade (total)
running = True
while running:
print('What building do you want to upgrade?')
buildings_keys = list(buildings.keys())
for i in range(len(buildings)):
print(f'{i + 1}: {buildings_keys[i]}')
i = int(input('Enter the number corresponding to the building:\n')) - 1
current_level = int(input('What is the current level of the building? Enter 0 if it is not built yet.\n'))
target_level = int(input('What is the target level of the building?\n'))
print()
for j in range(current_level, target_level):
if j == 0:
temp = 1
else:
temp = j
total[0] += int(buildings[buildings_keys[i]][3]) * temp
total[1] += int(buildings[buildings_keys[i]][4]) * temp
total[2] += int(buildings[buildings_keys[i]][5]) * temp
total[3] += int(buildings[buildings_keys[i]][6]) * temp
print('Would you like to upgrade another building? (y/N)')
if input().lower() != 'y':
running = False
print()
print()
print('Total resources needed:')
print(f' Reinforced Concrete: {total[0]}')
print(f' Bricks: {total[1]}')
print(f' Planks: {total[2]}')
print(f' Construction Units: {total[3]}')