feat: stack size in new hand dialog

Author: truenicoco

hh_creator/dialog.py

@@ -7,7 +7,7 @@
 from PyQt5.QtCore import pyqtSlot
 
 from . import config, hh
-from .util import AutoUI, amount_validator, decimal_conversion
+from .util import AutoUI, decimal_conversion, amount_validator
 
 if typing.TYPE_CHECKING:
     from .player import PlayerItemGroup
@@ -28,6 +28,7 @@ class NewHandDialog(QtWidgets.QDialog, AutoUI):
         "Straddle": Field(int, True),
         "Ante": Field(Decimal, True),
         "BBAnte": Field(Decimal, True),
+        "Stack": Field(Decimal, False),
         "Players": Field(int, False),
         "Currency": Field(str, True),
         "Variant": Field(str, False, "comboBox", "CurrentText"),
@@ -44,8 +45,12 @@ def __init__(self, *a, **kw):
 
         conf = config.config["new_hand"]
         for name, field in self.FIELDS.items():
+            if name == "Stack":
+                default = conf.get("default_stack_in_bb")
+            else:
+                default = conf.get(name)
             widget = self._get_widget(name)
-            getattr(widget, f"set{field.input_property}")(conf.get(name))
+            getattr(widget, f"set{field.input_property}")(default)
             if field.checkable:
                 checkbox = self._get_checkbox(name)
                 checked = conf.getboolean(f"{name}_checked")
@@ -166,7 +171,9 @@ def update_ok(self):
         ante = self.get_field_value("Ante", default=0)
         players = self.get_field_value("Players", default=0)
         decimals = self.get_field_value("Decimals", default=0)
+        stack = self.get_field_value("Stack", default=0)
 
+        stack_ok = stack > 0
         sb_ok = 0 <= sb <= bb
         bb_ok = 0 < bb
         straddle_ok = (
@@ -178,7 +185,7 @@ def update_ok(self):
         decimals_ok = decimals >= 0
 
         self.ok_button.setEnabled(
-            all((sb_ok, bb_ok, straddle_ok, ante_ok, players_ok, decimals_ok))
+            all((stack_ok, sb_ok, bb_ok, straddle_ok, ante_ok, players_ok, decimals_ok))
         )
 
 

hh_creator/main_window.py

@@ -152,8 +152,7 @@ def on_actionNew_triggered(self):
             self.scene.n_seats = dialog.get_field_value("Players")
 
             self.scene.set_all_stacks(
-                config.config["new_hand"].getint("default_stack_in_bb")
-                * dialog.get_field_value("BB")
+                dialog.get_field_value("Stack") * dialog.get_field_value("BB")
             )
 
             self.scene.set_n_cards(dialog.get_n_cards())
@@ -166,7 +165,11 @@ def on_actionNew_triggered(self):
 
             conf = dialog.conf
             for name, field in dialog.FIELDS.items():
-                conf[name] = config.escape_dollar(dialog.get_field_value(name))
+                if name == "Stack":
+                    conf_name = "default_stack_in_bb"
+                else:
+                    conf_name = name
+                conf[conf_name] = config.escape_dollar(dialog.get_field_value(name))
                 if field.checkable:
                     conf[f"{name}_checked"] = str(
                         dialog._get_checkbox(name).isChecked()

hh_creator/resource/ui/NewHandDialog.ui

@@ -221,6 +221,20 @@
        </item>
       </layout>
      </item>
+     <item row="11" column="0">
+      <widget class="QLabel" name="label_11">
+       <property name="text">
+        <string>Stack</string>
+       </property>
+      </widget>
+     </item>
+     <item row="11" column="1">
+      <layout class="QHBoxLayout">
+       <item>
+        <widget class="QLineEdit" name="lineEditStack"/>
+       </item>
+      </layout>
+     </item>
     </layout>
    </item>
    <item>

hh_creator/util.py

@@ -105,7 +105,7 @@ def __init__(self):
 
 def decimal_conversion(text):
     try:
-        return Decimal(text)
+        return Decimal(text.replace(",", "."))
     except InvalidOperation:
         log.warning(f"Error converting {text} to decimal, returning 0")
         return Decimal(0)