v3

Author: bot910

main.py

@@ -1,19 +1,25 @@
-from collections import Counter
-from typing import Literal
-import json
-import time
-import sys
-import os
+from collections import Counter #counter for counting resources
+from typing import Literal #type hinting for literals
+import time #time for sleep
+import sys #sys for clearing console
+import ast #ast for parsing a python dict
+import os #os for clearing console
 
 players = {}
 nextprint = ""
 
-def get_numbers(input: str):
+def clear(): #clear console
+    if sys.platform.startswith("win"):
+        os.system("cls")
+    else:
+        os.system("clear")
+
+def get_numbers(input: str): #get list of numbers from input
     user_input = input
     numbers = [int(num.strip()) for num in user_input.split(",")]
     return numbers
 
-def num_to_resource(num: int):
+def num_to_resource(num: int): #convert number to resource name
     if num == 1:
         return "Wood"
     elif num == 2:
@@ -24,19 +30,28 @@ def num_to_resource(num: int):
         return "Sheep"
     elif num == 5:
         return "Ore"
-    
-def add_player(name: str):
+
+def get_amount_of_resources(name: str, resource: int): #get amount of specific resource a player has
+    global players
+    resources = players[name].get("resources", [])
+
+    if resource is None:
+        return 0
+    else:
+        return resources.count(int(resource))
+
+def add_player(name: str): #add player to game
     global players
     players[name] = {"order":len(players)+1}
 
-def get_player_order(order: int):
+def get_player_order(order: int): #get player name by order
     global players
     for name, info in players.items():
         if info.get("order") == order:
             return name
     return None
 
-def get_player_resources(name: str):
+def get_player_resources(name: str): #get list of resources a player has
     global players
     if name not in players:
         raise ValueError(f"Player {name} does not exist.")
@@ -45,22 +60,23 @@ def get_player_resources(name: str):
     else:
         return players[name]["resources"]
 
-def format_resources(name: str, resources: list):
+def format_resources(name: str, resources: list): #format resources for printing
     global players
     counts = Counter(resources)
     formatted = " ".join(f"{num}:{num_to_resource(num)} = {counts[num]}," for num in sorted(counts))
     return f"{name}: {formatted}"
 
-def add_resource(name: str, resources: list):
+def add_resource(name: str, resources: list, dontshow=False): #add resources to player
     global players
+    global nextprint
     processed = []
     for r in resources:
         if r > 6 or r < 1:
-            raise ValueError("Invalid resource")
+            print(f"Invalid resource {r}. Skipping.")
         elif r == 6:
             choice = input(f"Resource '(6) gold' for {name} detected! Enter choice: ")
             if not choice or choice == 6:
-                raise ValueError("Invalid replacement")
+                print("No valid choice made. Skipping.")
             processed.append(int(choice))
         else:
             processed.append(r)
@@ -69,22 +85,36 @@ def add_resource(name: str, resources: list):
         players[name]["resources"] = processed.copy()
     else:
         players[name]["resources"].extend(processed)
+    if not dontshow:
+        nextprint += f"Added {processed} to {name}\n"
 
-def remove_resource(name: str, resource: list):
+def remove_resource(name: str, resource: list): #remove resources from player
     global players
     if name not in players:
-        raise ValueError(f"Player {name} does not exist.")
+        print(f"Player {name} does not exist.")
     elif "resources" not in players[name]:
-        raise ValueError(f"Player {name} has no resources.")
+        print(f"Player {name} has no resources.")
     else:
         for r in resource:
             if r > 0 and r < 7:
                 players[name]["resources"].remove(r)
             else:
-                raise ValueError("Invalid resource")
+                print(f"Invalid resource {r}.")
 
-def trade(name1: str, name2: str, resource1: list, resource2: list):
+def trade(name1: str, name2: str, resource1: list, resource2: list): #trade resources between players
     global players
+    global nextprint
+    try:
+        name1 = int(name1)
+    except ValueError:
+        pass
+    try:
+        name2 = int(name2)
+    except ValueError:
+        pass
+    if type(name1) == int and type(name2) == int:
+        name1 = get_player_order(name1)
+        name2 = get_player_order(name2)
     if name1 not in players or name2 not in players:
         print("Player does not exist.")
         return
@@ -99,13 +129,13 @@ def trade(name1: str, name2: str, resource1: list, resource2: list):
         if r not in players[name2]["resources"]:
             print(f"{name2} does not have resource {r}. Trade canceled.")
             return
-    add_resource(name1, resource2)
+    nextprint += f"Traded {resource1} from {name1} to {name2} for {resource2}\n"
+    add_resource(name1, resource2, dontshow=True)
     remove_resource(name2, resource2)
-    add_resource(name2, resource1)
+    add_resource(name2, resource1, dontshow=True)
     remove_resource(name1, resource1)
 
-
-def add_source(name: str, number: Literal[2,3,4,5,6,7,8,9,10,11,12], resource: Literal[1,2,3,4,5,6]):
+def add_source(name: str, number: Literal[2,3,4,5,6,7,8,9,10,11,12], resource: Literal[1,2,3,4,5,6]): #add resource source to player
     global players
     if resource > 0 and resource < 7 and number > 1 and number < 13 and name in players:
         if "sources" not in players[name]:
@@ -116,7 +146,7 @@ def add_source(name: str, number: Literal[2,3,4,5,6,7,8,9,10,11,12], resource: L
 
         players[name]["sources"][number].append(resource)
 
-def new_roll(p, number):
+def new_roll(p, number): #process new roll and add resources to players
     global players
     global nextprint
     number = int(number)
@@ -125,20 +155,21 @@ def new_roll(p, number):
         sources = data.get("sources", {})
         if number in sources:
             resources_to_add = sources[number].copy()
-            add_resource(name, resources_to_add)
-            nextprint = nextprint +f"Added {resources_to_add, num_to_resource(resources_to_add) } to {name} for roll {number}\n"
+            add_resource(name, resources_to_add, dontshow=True)
+            names_to_add = []
+            for i in range(len(resources_to_add)):
+                names_to_add.append(num_to_resource(resources_to_add[i]))
+            nextprint = nextprint +f"Added {resources_to_add}:{names_to_add} to {name} for roll {number}\n"
         else:
             pass
-def main():
+
+def main(): #main game loop
     global players
     turn = 1
     while True:
         nextturn = 0
         while nextturn == 0:
-            if sys.platform.startswith("win"):
-                os.system("cls")  #
-            else:
-                os.system("clear")  
+            clear()
             print(nextprint)
             player = 0
 
@@ -157,16 +188,23 @@ def main():
     2 - Trade
     3 - Steal
     4 - Split
-    5 - Add source
-    6 - Add resources
-    7 - remove latest source
+    5 - Play action card
+    6 - Add source
+    7 - Add resources
     8 - Next turn
-    9 - Input full data\n""")
+    9 - Advanced options\n""")
 
             print("Debug (full list of players and data):")
             print(players)
 
-            choice = int(input("Enter choice (1-8): "))
+            choice = input("Enter choice (1-9): ").strip()
+            try:
+                choice = int(choice)
+            except ValueError:
+                print("Invalid choice. Please try again.")
+                time.sleep(0.75)
+                continue
+
             if choice == 1: #build
                 buildingchoice = input("Enter building (VI, CI, CA, ST, SH or 1-5): ")
                 if buildingchoice == "q":
@@ -183,34 +221,69 @@ def main():
                     remove_resource(player, [1,4])
                 else:
                     remove_resource(player, [buildingchoice])
+
             elif choice == 2: #trade
                 trade(input("Enter player 1: "), input("Enter player 2: "), get_numbers(input("Enter resources 1 (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")), get_numbers(input("Enter resources 2 (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")))
+            
             elif choice == 3: #Steal
                 victim = input("Enter victim: ")
                 if victim == "q":
                     continue
                 else:
                     card = input("Enter card (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")
                     remove_resource(victim, [card])
+            
             elif choice == 4: #split
                 player_a = input("Enter amount of players: ")
                 if player_a == "q":
                     continue
                 else:
                     for i in range(int(player_a)):
                         remove_resource(input("Enter player name: "), get_numbers(input("Enter resources (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")))
-            elif choice == 5: #add source
+            
+            elif choice == 5: #play action card
+                card_a = input("Enter card (1:mon,2:inv): ")
+                if card_a == "q":
+                    continue
+                elif card_a in ("1", "mon", "mo", "monopoly"): 
+                    card_chosen = input("Enter card (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")
+                    total_amount = 0
+                    for name in players:
+                        amount_of_card_chosen = get_amount_of_resources(name,int(card_chosen))
+                        total_amount += amount_of_card_chosen
+                        for i in range(amount_of_card_chosen):
+                            remove_resource(name, [int(card_chosen)])
+                    add_resource(player, [int(card_chosen)] * total_amount)
+                    
+                elif card_a in ("2", "inv", "in", "invention"):
+                    player_name = input("Enter player name: ")
+                    add_resource(player_name, input("Enter resources (1:wood 2:stone 3:wheet 4:sheep 5:ore): "))
+                    add_resource(player_name, input("Enter resources (1:wood 2:stone 3:wheet 4:sheep 5:ore): "))
+            
+            elif choice == 6: #add source
                 source_a = input("Enter amount of sources: ")
                 if source_a == "q":
                     continue
                 else:
                     source_a = int(source_a)
                     for i in range(source_a):
-                        add_source(player, int(input(f"Enter source ({i+1}/{source_a}) number (2-12): ")), int(input("Enter resource (1:wood 2:stone 3:wheet 4:sheep 5:ore 6:custom): ")))
-            elif choice == 6: #add resources
+                        source_n = str(input(f"Enter source ({i+1}/{source_a}) number (2-12): "))
+                        if source_n == "q" or source_a == "q":
+                            continue
+                        source_r = str(input("Enter resource (1:wood 2:stone 3:wheet 4:sheep 5:ore 6:custom): "))
+                        if source_r == "q":
+                            continue
+                        else:
+                            source_n = int(source_n)
+                            for i in range(len(source_r)):
+                                if source_r[i] == ",":
+                                    pass
+                                else:
+                                    add_source(player, source_n, int(source_r[i]))
+            
+            elif choice == 7: #add resources
                 add_resource(player, get_numbers(input("Enter resources (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")))
-            elif choice == 7: #remove latest source
-                players[player]["sources"].popitem()
+            
             elif choice == 8: #next turn
                 roll = input("Enter roll (2-12): ")
                 if roll == "q":
@@ -219,29 +292,62 @@ def main():
                     new_roll(player, roll)
                     turn += 1
                     nextturn = 1
+
             elif choice == 9: #input custom data
-                data = input("Enter full data (The data must me in the correct format):")
-                if data == "q":
-                    continue
+                clear()
+                if turn == 1:
+                    player = get_player_order(1)
+                    print("Starting player:", player)
                 else:
-                    players = json.loads(data)
+                    player = get_player_order((turn - 1) % len(players) + 1)
+                    print("Current player:", player)
+
+                for name in players:
+                    print(format_resources(name, get_player_resources(name)))
+                print("""\nAdvanced options:
+    1. Remove resources
+    2. Remove latest source
+    3. Import full data (Python dict format)\n""")
+                choice_a = input("Enter choice (1-3): ").strip()
+                if choice_a == "q":
+                    continue
+                elif choice_a == "1":
+                    remove_resource(player, get_numbers(input("Enter resources (1:wood 2:stone 3:wheet 4:sheep 5:ore): ")))
+                elif choice_a == "2": #remove latest source
+                    if "sources" not in players[player] or len(players[player]["sources"]) == 0:
+                        print("No sources to remove.")
+                    else:
+                        players[player]["sources"].popitem()
+                elif choice_a == "3":
+                    data = input("Enter full data (The data must me in the correct format):")
+                    if data == "q":
+                        continue
+                    else:
+                        try:
+                            parsed = ast.literal_eval(data)
+                            if not isinstance(parsed, dict):
+                                print("Parsed value is not a dict.")
+                            else:
+                                players = parsed
+                                print("Imported players successfully.")
+                        except Exception as e:
+                            print("Failed to parse input as Python literal:", e)
+            
             else:
-                print("Invalid choice.")
-            time.sleep(0.5)
+                print("Invalid choice. Please try again.")
+            
+            time.sleep(0.75)
 
-def init_game():
+def init_game(): #initialize game and players
     print("""╔═╗╔═╗╔╦╗╔═╗╔╗╔  ┌┬┐┬─┐┌─┐┌─┐┬┌─┌─┐┬─┐  
 ║  ╠═╣ ║ ╠═╣║║║   │ ├┬┘├─┤│  ├┴┐├┤ ├┬┘  
-╚═╝╩ ╩ ╩ ╩ ╩╝╚╝   ┴ ┴└─┴ ┴└─┘┴ ┴└─┘┴└─  V2.0
+╚═╝╩ ╩ ╩ ╩ ╩╝╚╝   ┴ ┴└─┴ ┴└─┘┴ ┴└─┘┴└─  V3.0
 ──────────────────────────────────────
 A simple irl Settlers of Catan card and player tracker. (With the seafarers gold resource)\n""")
     amount_of_players = int(input("Enter the amount of players: "))
     for i in range(amount_of_players):
         name = input(f"Enter player name ({i+1}/{amount_of_players}): ")
         add_player(name)
 
-
-
-
 init_game()
-main()
+main()