diff --git a/smash/create_dummy_data.py b/smash/create_dummy_data.py index e6eb0cce8efd673c1eca633a9cc755ff16413eec..79ca13101bae29b9cd0cf4bb10147c44d6793374 100644 --- a/smash/create_dummy_data.py +++ b/smash/create_dummy_data.py @@ -103,6 +103,8 @@ class smashProvider(BaseProvider): alreadyCreatedWorkers = [] #rooms alreadyCreatedRooms = [] + #nd_number_ctr + nd_number_ctr = 0 def getLuxembourgTown(self): return self.fake.word(ext_word_list=self.luxtowns) @@ -383,10 +385,14 @@ class smashProvider(BaseProvider): study = Study.objects.filter(id=GLOBAL_STUDY_ID)[0] return [self.createSmashStudySubject(study=study) for _ in xrange(n)] - def createSmashStudySubject(self, subject=None, study=None, postponed=None, + def createSmashStudySubject(self, nd_number=None, subject=None, study=None, postponed=None, datetime_contact_reminder=None, type=None, default_location=None, flying_team=None, screening_number=None): # complete code and args... + if nd_number is None: + nd_number = 'nd{:05}'.format(self.nd_number_ctr) + self.nd_number_ctr+=1 + if subject is None: subject = self.createSmashSubject()