fix PB import, no course for code='X'

diff --git a/teleforma/management/commands/teleforma-import-users-pb.py b/teleforma/management/commands/teleforma-import-users-pb.py
index efb8cf6a920acf31fc786099eef59e3a02ca9487..1e3f32344b7ebeb0d8998a0fa039d2bf87975634 100644 (file)
--- a/teleforma/management/commands/teleforma-import-users-pb.py
+++ b/teleforma/management/commands/teleforma-import-users-pb.py
@@ -20,16 +20,15 @@ class Command(BaseCommand):
     def get_courses(self, code):
         courses = Course.objects.filter(code=code)
         if courses:
-            return [courses[0]]
+            return courses[0]
         else:
-            raise BaseException('You should first create a course with this code: ' + code)
+            return None
 
     def get_training(self, code):
         platform_only = False
         if 'I' in code[0:2]:
             platform_only = True
-            code = code[3:]
-            print code
+            code = code[4:]
             training = Training.objects.get(code=code)
         else:
             training = Training.objects.get(code=code)

diff --git a/teleforma/views.py b/teleforma/views.py
index 4b9192a7cd1c3c7fbd377f6653fab3a67504823c..72d7e4739e97386ab6f94e50e3cd59bc965669ba 100755 (executable)
--- a/teleforma/views.py
+++ b/teleforma/views.py
@@ -41,7 +41,7 @@ def render(request, template, data = None, mimetype = None):
 
 def get_course(obj):
     course = []
-    if obj:
+    if obj and obj.code != 'X':
         course = Course.objects.filter(id=obj.id)
     return course