Added periodic CompTIA update command

1 files changed, 38 insertions, 0 deletions

diff --git a/servo/management/commands/periodic.py b/servo/management/commands/periodic.py
new file mode 100644
index 0000000..5895e3c
--- /dev/null
+++ b/servo/management/commands/periodic.py
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+
+import os
+import sys
+import gsxws
+import subprocess
+from time import strftime
+from django.conf import settings
+from django.core.cache import caches
+from django.core.management.base import BaseCommand
+
+from servo.models import GsxAccount
+
+
+class Command(BaseCommand):
+    help = 'Peforms periodic commands'
+    VERBS = ('comptia',)
+
+    def add_arguments(self, parser):
+        parser.add_argument('verb', type=str, choices=self.VERBS, help='Periodic command to perform')
+
+    def handle(self, *args, **options):
+        if 'comptia' in options['verb']:
+            # Update raw CompTIA data (all product groups)
+            try:
+                act = GsxAccount.fallback()
+            except Exception as e:
+                print >> sys.stderr, 'Failed to connect to GSX (%s)' % e
+                sys.exit(-1)
+            
+            try:
+                codes = gsxws.comptia.fetch()
+                caches['comptia'].set('codes', codes)
+            except Exception as e:
+                print >> sys.stderr, 'Failed to fetch CompTIA codes (%s)' % e
+                sys.exit(-1)
+
+        exit(0)