summaryrefslogtreecommitdiff
path: root/program/management/commands/importhosts.py
blob: 0f1b1548c2e59a1fe8a69cf38e7e457a6ed0195c (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
from django.core.management.base import NoArgsCommand

import MySQLdb

from 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].decode('latin1').encode('utf8').split(','):
                host = Host(name=macher.strip())
                host.save()

                counter += 1

        cursor.close()
        connection.close()

        print '%i hosts imported' % counter