summaryrefslogtreecommitdiff
path: root/program/management/commands
diff options
context:
space:
mode:
authorErnesto Rico-Schmidt <e.rico.schmidt@gmail.com>2011-03-21 14:46:02 (GMT)
committerErnesto Rico-Schmidt <e.rico.schmidt@gmail.com>2011-03-21 14:46:02 (GMT)
commit4b4324459e4e6669519d2c2904b9e1de83b30944 (patch)
tree2d2607382629f1eca9c57389e1e603c23362bfc1 /program/management/commands
parentbbbb87cf6bd3bee25f3fe777357dc195029d8d38 (diff)
data migration, added Note import.
Diffstat (limited to 'program/management/commands')
-rw-r--r--program/management/commands/importnotes.py62
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