From: Christoph Egger <Christoph.Egger@fau.de>
Date: Mon, 9 Sep 2013 19:40:38 +0000 (+0200)
Subject: Add --pay subcommand
X-Git-Url: https://git.siccegge.de//index.cgi?a=commitdiff_plain;h=e4b88a95401b0f30909d43cccc2211757a1acbed;p=teilnehmertool.git

Add --pay subcommand
---

diff --git a/teilnehmertool.py b/teilnehmertool.py
index 4961e70..098f44f 100644
--- 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()