support !split person amount

1 files changed, 15 insertions, 6 deletions

diff --git a/geldschieberbot.py b/geldschieberbot.py
index cc0b0cc..135fa33 100644
--- a/geldschieberbot.py
+++ b/geldschieberbot.py
@@ -269,21 +269,30 @@ def split(sender, args, msg):
 
     try:
         amount = to_cent(args[1])
+        persons = args[2:]
     except:
-        return None, "amount must be a positive number"
+        # support !split name amount
+        if len(args) == 3:
+            try:
+                amount = to_cent(args[2])
+                persons = [args[1]]
+            except:
+                return None, "amount must be a positive number"
+        else:
+            return None, "amount must be a positive number"
 
-    # len(args) - cmd - amount + sender
-    persons = len(args) - 2 + 1
-    amount_per_person = int(amount/persons)
+    # persons + sender
+    npersons = len(persons) + 1
+    amount_per_person = int(amount/npersons)
 
     if sender in num2name:
         recipient = num2name[sender]
     else:
         return None, "you must register first"
 
-    output = f"Split {to_euro(amount)} between {persons} -> {to_euro(amount_per_person)} each\n"
+    output = f"Split {to_euro(amount)} between {npersons} -> {to_euro(amount_per_person)} each\n"
     change = [args]
-    for p in args[2:]:
+    for p in persons:
         if p in name2num:
             if p == recipient:
                 output += f"{p}, you will be charged multiple times. This may not be what you want\n"