diff options
| -rw-r--r-- | program/management/commands/importnotes.py | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/program/management/commands/importnotes.py b/program/management/commands/importnotes.py new file mode 100644 index 0000000..faf3ea3 --- /dev/null +++ b/program/management/commands/importnotes.py @@ -0,0 +1,62 @@ +from django.contrib.auth.models import User +from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned +from django.core.management.base import NoArgsCommand +from django.utils.html import clean_html, strip_tags + +import MySQLdb + +from program.models import Note, Show, TimeSlot + +USER = 'helsinki' +PASSWD = 'helsinki' +DB = 'helsinki' + +OWNER = User.objects.get(pk=1) + +class Command(NoArgsCommand): + help = 'Import notes from current program' + + def handle_noargs(self, **options): + connection = MySQLdb.connect(user=USER, passwd=PASSWD, db=DB) + cursor = connection.cursor() + + cursor.execute("""SELECT n.titel, n.datum, s.titel, n.notiz +FROM notizen AS n JOIN sendungen AS s ON n.sendung_id=s.id +WHERE n.sendung_id in (SELECT id FROM sendungen WHERE letzter_termin > current_date) AND n.titel != ''""") + + counter = 0 + for ntitel, datum, stitel, notiz in cursor.fetchall(): + ntitel = strip_tags(ntitel.decode('latin1').encode('utf8')) + stitel = strip_tags(stitel.decode('latin1').encode('utf8')) + notiz = clean_html(notiz.decode('latin1').encode('utf8')) + + if stitel.endswith('(Wiederholung)'): + stitel = stitel[:-15] + + if datum: + year, month, day = datum.year, datum.month, datum.day + try: + show = Show.objects.get(name=stitel) + + try: + timeslot = TimeSlot.objects.get(programslot__show=show, start__year=year, start__month=month, start__day=day) + except ObjectDoesNotExist: + print 'no timeslot found for sendung "%s" and datum "%s"' % (stitel, datum) + except MultipleObjectsReturned: + print 'multiple timeslots found for sendung "%s" and datum "%s"' % (stitel, datum) + else: + note = Note(timeslot=timeslot, owner=OWNER, title=ntitel, content=notiz) + + try: + note.save() + except: + print 'could not save note "%s" for show "%s"' % (ntitel, stitel) + else: + counter += 1 + except ObjectDoesNotExist: + print 'show with name "%s" not found' % stitel + + cursor.close() + connection.close() + + print '%i notes imported' % counter
\ No newline at end of file |