From 147becca89251222461b41f1f3efac0533ca59fa Mon Sep 17 00:00:00 2001
From: Wholesomedonut <wholesomedonut@tuta.io>
Date: Mon, 14 Sep 2020 23:57:37 -0600
Subject: [PATCH] Started work on adding mechs

---
 clean_dict.py |  4 ++++
 main.py       | 35 ++++++++++++++++++++++++++++++++---
 mech.py       | 40 ++++++++++++++++++++++++++--------------
 3 files changed, 62 insertions(+), 17 deletions(-)
 create mode 100644 clean_dict.py

diff --git a/clean_dict.py b/clean_dict.py
new file mode 100644
index 0000000..87952d3
--- /dev/null
+++ b/clean_dict.py
@@ -0,0 +1,4 @@
+def clean(dirty_dict):
+  # gets rid of None items in dictionaries
+  clean_dict = {k:v for k,v in dirty_dict.items() if v is not None}
+  return clean_dict
diff --git a/main.py b/main.py
index d14ad9b..02223b8 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,36 @@
 import mech
+from collections import defaultdict
+import clean_dict as c
 
-test1 = mech.Mech('Wolverine WVR-7K',2,3,2)
 
-test1.get_card()
+def add_player(lobby_dict):
+  # Set the name
+  temp = input("Input your name: ")
+  # and immediately add it and a first mech into the lobby
+  test1 = mech.Mech(quiet=False,debug=True)
+  lobby_dict[temp] = test1
+  # cleans None from lobby
+  lobby_dict = c.clean(lobby_dict)
+  return lobby_dict
 
-test1.set_skill(4)
+
+def add_mech(lobby_dict):
+  # takes the lobby the player is in
+  player = input("Input your player ID: ")
+  # will need to authenticate player input (this would be serverside IRL)
+  temp = [lobby_dict[player]]
+  # temporary copy of current player's lobby info'
+  temp.append(mech.Mech(quiet=False,debug=True))
+  # adds new mech onto list
+  return temp
+
+lobby_dict = defaultdict(list)
+
+test2 = mech.Mech(quiet=True,debug=True)
+
+add_player(lobby_dict)
+
+for player in lobby_dict.items():
+  print('Player: ',player)
+  
+lobby_dict[player] = add_mech(lobby_dict)
diff --git a/mech.py b/mech.py
index 2c18f71..22d6c50 100644
--- a/mech.py
+++ b/mech.py
@@ -1,18 +1,31 @@
+import clean_dict as c
+
 class Mech:
-    def __init__(self, name='NewMech', size=1, skill=1, tmm=1):
+    def __init__(self, name='NewMech', dez='NEW-1',size=1, skill=1, tmm=1, debug=True,quiet=True):
+      if not quiet:
+        self.name = input("Mech name: ")
+        self.size = input("Mech size: ")
+        self.skill = input("Pilot skill: ")
+        self.tmm = input("Total Movement Modifier: ")
+        self.dez = input("Variant name: ")
+        self.debug = debug
+      else:
         self.name = name
+        self.dez = dez
         self.size = size
         self.skill = skill
         self.tmm = tmm
+        self.debug = debug
 
         self.card = {
-            "name": self.name,
-            "size": self.size,
-            "skill": self.skill,
-            "tmm": self.tmm
+            "name": self.get_name(),
+            "size": self.get_size(),
+            "skill": self.get_skill(),
+            "tmm": self.get_tmm(),
+            "dez": self.get_dez()
         }
 
-        print(self.card)
+        if debug: print(self.card)
 
     def set_name(self, name):
         self.name = name
@@ -31,17 +44,16 @@ class Mech:
         self.card["tmm"] = tmm
 
     def get_name(self):
-        return self.__name
+        return self.name
+
+    def get_dez(self):
+        return self.dez
 
     def get_size(self):
-        return self.__size
+        return self.size
 
     def get_skill(self):
-        return self.__skill
+        return self.skill
 
     def get_tmm(self):
-        return self.__tmm
-
-    def get_card(self):
-        return self.card
-
+        return self.tmm