From 29653da9d04d015468c2051447e7e25318f2a114 Mon Sep 17 00:00:00 2001
From: Carlos Vega <carlos.vega@uni.lu>
Date: Mon, 22 Oct 2018 17:01:41 +0200
Subject: [PATCH] added nd_number
---
smash/create_dummy_data.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/smash/create_dummy_data.py b/smash/create_dummy_data.py
index e6eb0cce..79ca1310 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()
--
GitLab