summaryrefslogtreecommitdiff
path: root/program/management/commands/importnotes.py
blob: faf3ea3d3af7d7901d7c75f2295a1acab6d136b9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
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