diff --git a/main.py b/main.py index 8a3cafe..e78d250 100644 --- a/main.py +++ b/main.py @@ -124,22 +124,19 @@ def main(): print(" ",end="") print(f"{i+1}: {data.businesses[i]['pretty_name']}: {data.currency}{millify(data.businesses[i]['cost'])}, Earning: {data.currency}{millify(data.businesses[i]['earning'])}/hour, Threat: {data.businesses[i]['threat']}%") user_input = input_command() - try: - user_input_int = int(user_input) - 1 - if user_input_int == -1: - continue - elif user_input_int < len(data.businesses): - if save.money >= data.businesses[user_input_int]['cost']: - print(f"Buying {data.businesses[user_input_int]['pretty_name']}") - save.businesses.append(data.businesses[user_input_int]) - save.businesses[len(save.businesses)-1]['cost'] *= 0.9 # Remove 10% of it's value so the sell price is slightly less - save.money -= data.businesses[user_input_int]['cost'] - else: - print("You can't afford that!") + user_input_int = int(user_input) - 1 + if user_input_int == -1: + continue + elif user_input_int < len(data.businesses): + if save.money >= data.businesses[user_input_int]['cost']: + print(f"Buying {data.businesses[user_input_int]['pretty_name']}") + save.businesses.append(data.businesses[user_input_int].copy()) + save.businesses[len(save.businesses)-1]['cost'] *= 0.9 # Remove 10% of it's value so the sell price is slightly less + save.money -= data.businesses[user_input_int]['cost'] else: - print("Invalid number") - except: - print("Didn't input a number") + print("You can't afford that!") + else: + print("Invalid number") ### Sell an owned business elif user_input.lower() == "sell":