summaryrefslogtreecommitdiff
path: root/program/management
diff options
context:
space:
mode:
Diffstat (limited to 'program/management')
-rw-r--r--program/management/commands/check_automation_ids.py35
-rw-r--r--program/management/commands/remove_automation_id.py16
2 files changed, 51 insertions, 0 deletions
diff --git a/program/management/commands/check_automation_ids.py b/program/management/commands/check_automation_ids.py
new file mode 100644
index 0000000..be57ae2
--- /dev/null
+++ b/program/management/commands/check_automation_ids.py
@@ -0,0 +1,35 @@
+import json
+from os.path import join
+
+from django.conf import settings
+from django.core.management.base import NoArgsCommand
+
+from program.models import ProgramSlot
+
+
+class Command(NoArgsCommand):
+ help = 'checks the automation_ids used by program slots against the exported'
+
+ def handle_noargs(self, **options):
+ cache_dir = getattr(settings, 'AUTOMATION_CACHE_DIR', 'cache')
+ cached_shows = join(cache_dir, 'shows.json')
+ with open(cached_shows) as shows_json:
+ shows = json.loads(shows_json.read())['shows']
+
+ automation_ids = []
+ for show in shows:
+ automation_ids.append(show['id'])
+ automation_ids.sort()
+
+ automation_ids2 = []
+ for programslot in ProgramSlot.objects.filter(automation_id__isnull=False):
+ automation_ids2.append(int(programslot.automation_id))
+ automation_ids2.sort()
+
+ for automation_id in automation_ids:
+ if automation_id not in automation_ids2:
+ print '+', automation_id
+
+ for automation_id in automation_ids2:
+ if automation_id not in automation_ids:
+ print '-', automation_id
diff --git a/program/management/commands/remove_automation_id.py b/program/management/commands/remove_automation_id.py
new file mode 100644
index 0000000..4fb8ee7
--- /dev/null
+++ b/program/management/commands/remove_automation_id.py
@@ -0,0 +1,16 @@
+from django.core.management.base import BaseCommand, CommandError
+
+from program.models import ProgramSlot
+
+
+class Command(BaseCommand):
+ help = 'removes the automation_id from the program slots'
+ args = '<automation_id>'
+
+ def handle(self, *args, **options):
+ if len(args) == 1:
+ automation_id = args[0]
+ else:
+ raise CommandError('you must provide the automation_id')
+
+ ProgramSlot.objects.filter(automation_id=automation_id).update(automation_id=None)