From 3c4de3c293ceae7fab8428b980cc36d7ad4fc07d Mon Sep 17 00:00:00 2001 From: yomguy Date: Wed, 6 Jun 2012 11:28:52 +0200 Subject: [PATCH] fix PB import, no course for code='X' --- teleforma/management/commands/teleforma-import-users-pb.py | 7 +++---- teleforma/views.py | 2 +- 2 files changed, 4 insertions(+), 5 deletions(-) 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 -- 2.39.5