diff options
Diffstat (limited to 'helsinki/program/management/commands')
5 files changed, 261 insertions, 0 deletions
diff --git a/helsinki/program/management/commands/__init__.py b/helsinki/program/management/commands/__init__.py new file mode 100644 index 0000000..3aa4ca5 --- /dev/null +++ b/helsinki/program/management/commands/__init__.py @@ -0,0 +1,2 @@ +__author__ = 'ers' +
\ No newline at end of file diff --git a/helsinki/program/management/commands/importhosts.py b/helsinki/program/management/commands/importhosts.py new file mode 100644 index 0000000..39a8afd --- /dev/null +++ b/helsinki/program/management/commands/importhosts.py @@ -0,0 +1,34 @@ +from django.core.management.base import NoArgsCommand + +import MySQLdb + +from helsinki.program.models import Host + +USER = 'helsinki' +PASSWD = 'helsinki' +DB = 'helsinki' + +class Command(NoArgsCommand): + help = 'Import hosts from current program' + + def handle_noargs(self, **options): + connection = MySQLdb.connect(user=USER, passwd=PASSWD, db=DB) + cursor = connection.cursor() + + cursor.execute("""SELECT DISTINCT macher +FROM sendungen +WHERE letzter_termin > current_date AND macher != '' AND titel NOT LIKE 'Musikprogramm'""") + + counter = 0 + + for row in cursor.fetchall(): + for macher in row[0].split(','): + host = Host(name=macher.strip()) + host.save() + + counter += 1 + + cursor.close() + connection.close() + + print '%i hosts imported' % counter diff --git a/helsinki/program/management/commands/importnotes.py b/helsinki/program/management/commands/importnotes.py new file mode 100644 index 0000000..19ddfed --- /dev/null +++ b/helsinki/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 helsinki.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) + 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 diff --git a/helsinki/program/management/commands/importprogramslots.py b/helsinki/program/management/commands/importprogramslots.py new file mode 100644 index 0000000..52aa73a --- /dev/null +++ b/helsinki/program/management/commands/importprogramslots.py @@ -0,0 +1,96 @@ +from django.core.exceptions import ObjectDoesNotExist +from django.core.management.base import NoArgsCommand +from django.utils.html import strip_tags + +from datetime import time +import MySQLdb + +from helsinki.program.models import Show, ProgramSlot, RRule + +USER = 'helsinki' +PASSWD = 'helsinki' +DB = 'helsinki' + +RRULES = { + 0: RRule.objects.get(pk=1), + 7: RRule.objects.get(pk=3), + 14: RRule.objects.get(pk=4), + 28: RRule.objects.get(pk=5) +} + +class Command(NoArgsCommand): + help = 'Import programslots from the current program' + + def handle_noargs(self, **options): + connection = MySQLdb.connect(user=USER, passwd=PASSWD, db=DB) + cursor = connection.cursor() + + cursor.execute("""SELECT titel, beginn, ende, erster_termin, letzter_termin, rhytmus, termin +FROM sendungen +WHERE letzter_termin > current_date AND titel NOT LIKE 'Musikprogramm' AND titel NOT LIKE '%%(Wiederholung)'""") + + counter = 0 + + for titel, beginn, ende, erster_termin, letzter_termin, rhytmus, termin in cursor.fetchall(): + titel = strip_tags(titel) + + hours, seconds = divmod(beginn.seconds, 3600) + minutes, seconds = divmod(seconds, 60) + tstart = time(hour=hours, minute=minutes, second=seconds) + + hours, seconds = divmod(ende.seconds, 3600) + minutes, seconds = divmod(seconds, 60) + tend = time(hour=hours, minute=minutes, second=seconds) + + try: + rrule = RRULES[rhytmus] + try: + show = Show.objects.get(name=titel) + except ObjectDoesNotExist: + print 'show with name "%s" not found' % titel + else: + programslot = ProgramSlot(rrule=rrule, byweekday=termin, show=show, dstart=erster_termin, tstart=tstart, + tend=tend, until=letzter_termin) + try: + programslot.save() + counter += 1 + except: + pass + except KeyError: + print 'rhythmus "%i" is not supported for sendung "%s"' % (rhytmus, titel) + + cursor.execute("""SELECT titel, beginn, ende, erster_termin, letzter_termin, rhytmus, termin +FROM sendungen +WHERE letzter_termin > current_date AND titel LIKE '%%(Wiederholung)'""") + + for titel, beginn, ende, erster_termin, letzter_termin, rhytmus, termin in cursor.fetchall(): + titel = strip_tags(titel[:-15]) + + hours, seconds = divmod(beginn.seconds, 3600) + minutes, seconds = divmod(seconds, 60) + tstart = time(hour=hours, minute=minutes, second=seconds) + + hours, seconds = divmod(ende.seconds, 3600) + minutes, seconds = divmod(seconds, 60) + tend = time(hour=hours, minute=minutes, second=seconds) + + try: + rrule = RRULES[rhytmus] + try: + show = Show.objects.get(name=titel) + except ObjectDoesNotExist: + print 'show with name "%s" not found' % titel + else: + programslot = ProgramSlot(rrule=rrule, byweekday=termin, show=show, dstart=erster_termin, tstart=tstart, tend=tend, until=letzter_termin, is_repetition=True) + try: + programslot.save() + counter += 1 + except: + pass + except KeyError: + print 'rhythmus "%i" is not supported for sendung "%s"' % (rhytmus, titel) + + cursor.close() + connection.close() + + print '%i programslots imported' % counter diff --git a/helsinki/program/management/commands/importshows.py b/helsinki/program/management/commands/importshows.py new file mode 100644 index 0000000..b013dab --- /dev/null +++ b/helsinki/program/management/commands/importshows.py @@ -0,0 +1,67 @@ +from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned +from django.core.management.base import NoArgsCommand +from django.template.defaultfilters import slugify +from django.utils.html import clean_html, strip_tags + +import MySQLdb + +from helsinki.program.models import BroadcastFormat, Host, Show + +USER = 'helsinki' +PASSWD = 'helsinki' +DB = 'helsinki' + +TALK = BroadcastFormat.objects.get(pk=1) + +class Command(NoArgsCommand): + help = 'Import shows from the current program' + + def handle_noargs(self, **options): + connection = MySQLdb.connect(user=USER, passwd=PASSWD, db=DB) + cursor = connection.cursor() + + cursor.execute("""SELECT titel, beschreibung, web, macher +FROM sendungen +WHERE letzter_termin > current_date AND titel NOT LIKE 'Musikprogramm' AND titel NOT LIKE '%%(Wiederholung)' +ORDER BY titel, beginn, ende""") + + counter = 0 + + for titel, beschreibung, web, macher in cursor.fetchall(): + titel = strip_tags(titel) + beschreibung = clean_html(beschreibung) + + slug = slugify(titel) + + hosts = [] + + for macher in macher.split(','): + macher = macher.strip() + try: + host = Host.objects.get(name=macher) + except MultipleObjectsReturned: + print 'multiple hosts with name "%s" found' % macher + except ObjectDoesNotExist: + print 'host with name "%s" not found' % macher + else: + hosts.append(host) + + try: + show = Show.objects.get(name=titel) + print 'sendung "%s" already imported as show "%s"' % (titel, show) + except ObjectDoesNotExist: + show = Show(broadcastformat=TALK, name=titel, slug=slug, short_description='FIXME', description=beschreibung) + try: + show.save() + counter += 1 + except: + print 'sendung "%s" could not be imported' % titel + else: + for h in hosts: + show.hosts.add(h) + show.save() + + cursor.close() + connection.close() + + print '%i shows imported' % counter |