From: yomguy <yomguy@parisson.com>
Date: Wed, 6 Jun 2012 09:28:52 +0000 (+0200)
Subject: fix PB import, no course for code='X'
X-Git-Tag: 0.5^2~21^2
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=3c4de3c293ceae7fab8428b980cc36d7ad4fc07d;p=teleforma.git

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 efb8cf6a..1e3f3234 100644
--- 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 4b9192a7..72d7e473 100755
--- 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