Add --pay subcommand

diff --git a/teilnehmertool.py b/teilnehmertool.py
index 4961e70a04e166f37baa2bd966ecb87a2147ebed..098f44f211a1e7f3e4bd40ffbbb4011fabda0df4 100644 (file)
--- a/teilnehmertool.py
+++ b/teilnehmertool.py
@@ -30,6 +30,17 @@ def import_teilnehmer(input):
 
     conn.commit()
             
+def mark_pay():
+    betrag = int(raw_input("Betrag: "))
+
+    while True:
+        uid = raw_input("User: ")
+        if uid == "":
+            break
+        uid = int(uid)
+        conn.execute("UPDATE teilnehmer SET bezahlt = ? WHERE id = ?", [betrag, uid])
+
+    conn.commit()
 
 def create_mail(outdir='output'):
     def gen_rechnung(tshirt, zipper):
@@ -86,7 +97,9 @@ def main():
                        help='Generiere Nametags')
     group.add_argument('--email', action='store_true',
                        help='Generiere E-Mails mit Überweisungsinformationen')
-
+    group.add_argument('--pay', action='store_true',
+                       help='Speichere Zahlungseingang')
+    
     # Argumente
     parser.add_argument('--db', default='teilnehmer.sqlite',
                         help='Teilnehmerdatenbank')
@@ -115,6 +128,9 @@ def main():
         assert(args.input)
         import_teilnehmer(args.input)
 
+    elif args.pay:
+        mark_pay()
+
 if __name__ == '__main__':
     main()