add option menu to select troops gui

README.md

@@ -31,7 +31,9 @@ You also need a running redis server (used db number is `0`, soon configurable).
 
 ## Troops selection
 
-Not developed yet
+Start troops selection GUI with:
+
+    python select_troops.py --country URSS --country DE
 
 ## Troops Placement phase
 

opencombat/strategy/manager.py

@@ -2,7 +2,7 @@
 from synergine2.config import Config
 from synergine2.log import get_logger
 
-from opencombat.strategy.selection.troops import TroopClassBuilder
+from opencombat.strategy.troops import TroopClassBuilder
 
 
 class TroopManager(object):
@@ -17,4 +17,7 @@ class TroopManager(object):
 
         builder = TroopClassBuilder(config)
         self._unit_stash = builder.get_unit_stash(units_file_path)
-        self._team_stash = builder.get_team_stash(teams_file_path)
+        self._team_stash = builder.get_team_stash(
+            units_file_path,
+            teams_file_path,
+        )

opencombat/strategy/selection/gui.py

@@ -1,6 +1,8 @@
 # coding: utf-8
+import typing
 from tkinter import Tk
-from tkinter import Label
+from tkinter import StringVar
+from tkinter import OptionMenu
 from tkinter import W
 
 from synergine2.config import Config
@@ -15,10 +17,25 @@ class SelectTroopsGui(Gui):
         config: Config,
         master: Tk,
         troop_manager: TroopManager,
+        countries: typing.List[str],
     ) -> None:
         super().__init__(config, master)
         self._master.title('Troops selection')
-        self.label = Label(master, text="Hello")
+
+        # Widgets
+        self.selected_country_var = StringVar(self._master)
+        self.selected_country_var.set(countries[0])
+        self.selected_country_var.trace('w', self.change_country)
+        self.select_country_menu = OptionMenu(
+            self._master,
+            self.selected_country_var,
+            *countries,
+        )
 
         # Layout
-        self.label.grid(row=0, column=0, sticky=W)
+        self.select_country_menu.grid(row=0, column=0, sticky=W)
+
+    def change_country(self, *args, **kwargs) -> None:
+        self._logger.info('Change country to "{}"'.format(
+            self.selected_country_var.get(),
+        ))

opencombat/strategy/troops.py

@@ -33,6 +33,7 @@ class TroopClassBuilder(object):
 
     def get_team_stash(
         self,
+        units_file_path: str,
         teams_file_path: str,
     ) -> UnitStash:
         class_address = self._config.resolve(
@@ -43,7 +44,10 @@ class TroopClassBuilder(object):
             self._config,
             class_address,
         )
+
+        unit_stash = self.get_unit_stash(units_file_path)
         return class_(
             self._config,
             teams_file_path,
+            unit_stash=unit_stash,
         )

select_troops.py

@@ -28,6 +28,7 @@ def main(
         config,
         master=master,
         troop_manager=troop_manager,
+        countries=countries,
     )
     master.mainloop()
 
@@ -50,7 +51,6 @@ if __name__ == '__main__':
         '--country',
         action='append',
         dest='countries',
-        default=['URSS', 'DE'],
     )
     args = parser.parse_args()