From 87c918514af5f31d6a5d29d0870a0278926af2f3 Mon Sep 17 00:00:00 2001 From: Christoph Egger Date: Thu, 12 Sep 2013 13:43:55 +0200 Subject: [PATCH] Finish received-payment functionality --- teilnehmertool.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/teilnehmertool.py b/teilnehmertool.py index 60f36ef..b11b688 100755 --- a/teilnehmertool.py +++ b/teilnehmertool.py @@ -30,15 +30,24 @@ def import_teilnehmer(input): conn.commit() -def mark_pay(): +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) conn.execute("UPDATE teilnehmer SET bezahlt = ? WHERE id = ?", [betrag, uid]) + vorname, nachname, email = conn.execute("SELECT vorname, nachname, email FROM teilnehmer WHERE id = ?", [uid]).fetchone() + + 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() @@ -117,7 +126,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: assert(args.output) if not os.path.exists(args.output): os.mkdir(args.output) @@ -126,14 +135,13 @@ def main(): create_nametag(args.output) elif args.email: create_mail(args.output) + elif args.pay: + mark_pay(args.output) elif args.importcsv: assert(args.input) import_teilnehmer(args.input) - elif args.pay: - mark_pay() - if __name__ == '__main__': main() -- 2.39.2