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)

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:
            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...')
buildings = data_download('buildings.tsv')  # In format {building_name: [brick, plank, rc, cu]}
print('Buildings downloaded and processed')
print()

# 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]}')