summaryrefslogtreecommitdiff
path: root/helsinki.program/management/commands/importhosts.py
blob: 39a8afd56e73f4a94493333fcf58b77e3d8dc313 (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 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