code de-duplication

1 files changed, 9 insertions, 14 deletions

diff --git a/geldschieberbot.py b/geldschieberbot.py
index d6fb9b1..20561c0 100644
--- a/geldschieberbot.py
+++ b/geldschieberbot.py
@@ -135,6 +135,13 @@ def create_members():
         r += m + ": " + name2num[m] + "\n"
     return r
 
+def add_to_balance(name):
+    nb = {}
+    for m in balance:
+        balance[m][name] = 0
+        nb[m] = 0
+    balance[name] = nb
+
 def create_help():
     return """
 Usage: send a message starting with '!' followed by a command
@@ -187,12 +194,7 @@ def register(sender, args, msg):
     num2name[sender] = name
     name2num[name] = sender
 
-    # add to balance
-    nb = {}
-    for m in balance:
-        balance[m][name] = 0
-        nb[m] = 0
-    balance[name] = nb
+    add_to_balance(name)
 
     # add changes list
     changes[name] = []
@@ -414,14 +416,7 @@ def cars(sender, args, msg):
             return None, "service-charge must be a positive number"
 
         available_cars[car] = service_charge
-
-        # add car special user
-        # add to balance
-        nb = {}
-        for m in balance:
-            balance[m][car] = 0
-            nb[m] = 0
-        balance[car] = nb
+        add_to_balance(car)
         return 'added "{}" as an available car'.format(car), None
     else:
         return None, 'cmd not in form "{} [cmd]"'.format(args[0])