teilnehmertool.py

   1 #!/usr/bin/python
   2 # -*- coding: utf-8 -*-
   3
   4 import argparse
   5 import sqlite3
   6 import datetime
   7 import csv
   8 import string
   9 import jinja2
  10 import os
  11 import os.path
  12
  13 conn = None
  14
  15
  16 env = jinja2.Environment(loader=jinja2.FileSystemLoader('templates'))
  17
  18 def import_teilnehmer(input):
  19     ids = set([ i[0] for i in conn.execute('SELECT id FROM teilnehmer').fetchall() ])
  20
  21     with open(input) as f:
  22         reader = csv.reader(f, delimiter=';', quotechar='"')
  23         for row in reader:
  24             if row[0] == '#':
  25                 continue
  26             if not int(row[0]) in ids:
  27                 print "Importing %s" % (row, )
  28                 conn.execute("INSERT INTO teilnehmer VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
  29                              row + [0, 0, 0])
  30
  31     conn.commit()
  32
  33
  34 def create_mail(outdir='output'):
  35     def gen_rechnung(tshirt, zipper):
  36         preis = 25
  37         kosten = u""
  38         kosten += u"%s | %s\n" % (u"  Konferenzbeitrag".ljust(30), "25.00")
  39         kosten += u"%s | %s\n" % ((u"  T-Shirt (Größe %s)" % tshirt).ljust(30), " 0.00")
  40         if zipper != 'Kein Pulli':
  41             kosten += u"%s | %s\n" % ((u"  Kapuzenzipper (Größe %s)" % zipper).ljust(30), "25.00")
  42             preis += 25
  43         kosten += ' '*2 + 29*'-' + '+' + 6*'-' + '\n'
  44         kosten += u"%s | %s\n" % (u"  Summe".ljust(30), "%2d.00" % preis)
  45
  46         return kosten
  47
  48     def gen_token(id, vorname, nachname):
  49         return u"%s-%s-%s" % (id, nachname.decode('utf-8').upper(), vorname.decode('utf-8').upper())
  50
  51     template = env.get_template('teilnehmerbeitrag.eml')
  52
  53     for id, vorname, nachname, email, tshirt, zipper in \
  54         conn.execute("SELECT id, vorname, nachname, email, tshirt, zipper FROM teilnehmer WHERE emailsent = 0"):
  55
  56         with open(os.path.join(outdir, email), 'w') as f:
  57             f.write(template.render(vorname=vorname.decode('utf-8'),
  58                                     nachname=nachname.decode('utf-8'),
  59                                     email=email.decode('utf-8'),
  60                                     token=gen_token(id, vorname, nachname),
  61                                     rechnung=gen_rechnung(tshirt, zipper)).encode('utf-8'))
  62
  63         conn.execute("UPDATE teilnehmer set emailsent = ? WHERE id = ?", [datetime.datetime.now().isoformat(), id])
  64     conn.commit()
  65
  66 def create_nametag(outdir='output'):
  67     template = env.get_template('nametag.svg')
  68
  69     for vorname, nachname, namensschild, hochschule in \
  70         conn.execute("SELECT vorname, nachname, namensschild, hochschule FROM teilnehmer"):
  71
  72         with open(os.path.join(outdir, "%s.svg" % namensschild), 'w') as f:
  73             f.write(template.render(name=u"%s %s" % (vorname.decode('utf-8'),
  74                                                      nachname.decode('utf-8')),
  75                                     nick=namensschild.decode('utf-8'),
  76                                     uni1=hochschule.decode('utf-8')).encode('utf-8'))
  77
  78 def main():
  79     parser = argparse.ArgumentParser(description="Ultimate Teilnehmertool")
  80
  81     # Modus
  82     group = parser.add_mutually_exclusive_group(required=True)
  83     group.add_argument('--importcsv', action='store_true',
  84                        help='Importiere neue Teilnehmer aus CSV')
  85     group.add_argument('--nametag', action='store_true',
  86                        help='Generiere Nametags')
  87     group.add_argument('--email', action='store_true',
  88                        help='Generiere E-Mails mit Überweisungsinformationen')
  89
  90     # Argumente
  91     parser.add_argument('--db', default='teilnehmer.sqlite',
  92                         help='Teilnehmerdatenbank')
  93     parser.add_argument('-i', '--input',
  94                         help='Eingabedatei')
  95     parser.add_argument('-o', '--output', default='output',
  96                         help='Ausgabeverzeichnis')
  97
  98     args = parser.parse_args()
  99
 100     global conn
 101     conn = sqlite3.connect(args.db)
 102     conn.text_factory = str
 103
 104     if args.nametag or args.email:
 105         assert(args.output)
 106         if not os.path.exists(args.output):
 107             os.mkdir(args.output)
 108
 109         if args.nametag:
 110             create_nametag(args.output)
 111         elif args.email:
 112             create_mail(args.output)
 113
 114     elif args.importcsv:
 115         assert(args.input)
 116         import_teilnehmer(args.input)
 117
 118 if __name__ == '__main__':
 119     main()
 120