Add reminder function

[teilnehmertool.git] / teilnehmertool.py

diff --git a/teilnehmertool.py b/teilnehmertool.py
old mode 100644 (file)
new mode 100755 (executable)
index 4961e70..35a47dc
--- a/teilnehmertool.py
+++ b/teilnehmertool.py
@@ -30,6 +30,32 @@ def import_teilnehmer(input):
 
     conn.commit()
             
+def mark_pay(outdir):
+    betrag = int(raw_input("Betrag: "))
+
+    template = env.get_template('eingangsbestaetigung.eml')
+
+    while True:
+        uid = raw_input("User: ")
+        if uid == "":
+            break
+        uid = int(uid)
+        vorname, nachname, email, bezahlt = \
+                 conn.execute("SELECT vorname, nachname, email, bezahlt FROM teilnehmer WHERE id = ?", [uid]).fetchone()
+
+        if bezahlt == betrag:
+            continue
+
+        conn.execute("UPDATE teilnehmer SET bezahlt = ? WHERE id = ?", [betrag, uid])
+
+
+        with open(os.path.join(outdir, email), 'w') as f:
+            f.write(template.render(vorname=vorname.decode('utf-8'),
+                                    nachname=nachname.decode('utf-8'),
+                                    email=email.decode('utf-8'),
+                                    betrag=betrag).encode('utf-8'))
+
+    conn.commit()
 
 def create_mail(outdir='output'):
     def gen_rechnung(tshirt, zipper):
@@ -43,7 +69,7 @@ def create_mail(outdir='output'):
         kosten += ' '*2 + 29*'-' + '+' + 6*'-' + '\n'
         kosten += u"%s | %s\n" % (u"  Summe".ljust(30), "%2d.00" % preis)
 
-        return kosten
+        return kosten, preis
 
     def gen_token(id, vorname, nachname):
         return u"%s-%s-%s" % (id, nachname.decode('utf-8').upper(), vorname.decode('utf-8').upper())
@@ -53,14 +79,54 @@ def create_mail(outdir='output'):
     for id, vorname, nachname, email, tshirt, zipper in \
         conn.execute("SELECT id, vorname, nachname, email, tshirt, zipper FROM teilnehmer WHERE emailsent = 0"):
 
+        rechnung, preis = gen_rechnung(tshirt, zipper)
+
+        with open(os.path.join(outdir, email), 'w') as f:
+            f.write(template.render(vorname=vorname.decode('utf-8'),
+                                    nachname=nachname.decode('utf-8'),
+                                    email=email.decode('utf-8'),
+                                    token=gen_token(id, vorname, nachname),
+                                    rechnung=rechnung).encode('utf-8'))
+
+        conn.execute("UPDATE teilnehmer set emailsent = ?, betrag = ? WHERE id = ?", [datetime.datetime.now().isoformat(),
+                                                                                      preis, id])
+    conn.commit()
+
+def create_remind(outdir='output'):
+    def gen_rechnung(tshirt, zipper):
+        preis = 25
+        kosten = u""
+        kosten += u"%s | %s\n" % (u"  Konferenzbeitrag".ljust(30), "25.00")
+        kosten += u"%s | %s\n" % ((u"  T-Shirt (Größe %s)" % tshirt).ljust(30), " 0.00")
+        if zipper != 'Kein Pulli':
+            kosten += u"%s | %s\n" % ((u"  Kapuzenzipper (Größe %s)" % zipper).ljust(30), "25.00")
+            preis += 25
+        kosten += ' '*2 + 29*'-' + '+' + 6*'-' + '\n'
+        kosten += u"%s | %s\n" % (u"  Summe".ljust(30), "%2d.00" % preis)
+
+        return kosten, preis
+
+    def gen_token(id, vorname, nachname):
+        return u"%s-%s-%s" % (id, nachname.decode('utf-8').upper(), vorname.decode('utf-8').upper())
+
+    template = env.get_template('erinnerung.eml')
+
+    for id, vorname, nachname, email, tshirt, zipper in \
+        conn.execute("SELECT id, vorname, nachname, email, tshirt, zipper FROM teilnehmer " +
+                     "WHERE bezahlt < 25 AND datetime(emailsent) < datetime(?)",
+                     [(datetime.datetime.now() - datetime.timedelta(14)).isoformat()]):
+
+        rechnung, preis = gen_rechnung(tshirt, zipper)
+
         with open(os.path.join(outdir, email), 'w') as f:
             f.write(template.render(vorname=vorname.decode('utf-8'),
                                     nachname=nachname.decode('utf-8'),
                                     email=email.decode('utf-8'),
                                     token=gen_token(id, vorname, nachname),
-                                    rechnung=gen_rechnung(tshirt, zipper)).encode('utf-8'))
+                                    rechnung=rechnung).encode('utf-8'))
 
-        conn.execute("UPDATE teilnehmer set emailsent = ? WHERE id = ?", [datetime.datetime.now().isoformat(), id])
+        conn.execute("UPDATE teilnehmer set emailsent = ?, betrag = ? WHERE id = ?", [datetime.datetime.now().isoformat(),
+                                                                                      preis, id])
     conn.commit()
 
 def create_nametag(outdir='output'):
@@ -86,7 +152,11 @@ 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')
+    group.add_argument('--remind', action='store_true',
+                       help='Generiere Zahlungsaufforderungserinnerung')
+    
     # Argumente
     parser.add_argument('--db', default='teilnehmer.sqlite',
                         help='Teilnehmerdatenbank')
@@ -101,7 +171,7 @@ def main():
     conn = sqlite3.connect(args.db)
     conn.text_factory = str
 
-    if args.nametag or args.email:
+    if args.nametag or args.email or args.pay or args.remind:
         assert(args.output)
         if not os.path.exists(args.output):
             os.mkdir(args.output)
@@ -110,6 +180,10 @@ def main():
             create_nametag(args.output)
         elif args.email:
             create_mail(args.output)
+        elif args.pay:
+            mark_pay(args.output)
+        elif args.remind:
+            create_remind(args.output)
 
     elif args.importcsv:
         assert(args.input)