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 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)
stitel = strip_tags(stitel)
notiz = clean_html(notiz)
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" and datum "%s"' % (ntitel, stitel, datum)
else:
counter += 1
except ObjectDoesNotExist:
print 'show with name "%s" not found' % stitel
cursor.close()
connection.close()
print '%i notes imported' % counter
|
