Skip to content
Snippets Groups Projects
Commit 12f59051 authored by Carlos Vega's avatar Carlos Vega
Browse files

removed SS number and added superuser at the end of the script

parent 7536a235
No related branches found
No related tags found
1 merge request!185Hotfix/voucher templates
......@@ -460,7 +460,7 @@ class smashProvider(BaseProvider):
phone_number = self.fake.phone_number()
if social_security_number is None:
social_security_number = self.fake.ean(length=8)
social_security_number = ''
if sex is None:
sex = self.fake.word(
......@@ -534,7 +534,7 @@ class smashProvider(BaseProvider):
def createSmashWorker(self, first_name=None, last_name=None, username=None, email=None,
specialization=None, unit=None, phone_number=None, password='password1234', role=None,
locations=None, languages=None):
locations=None, languages=None, is_super=False):
'''
first_name: string: Sets the first_name. By default None. Random values are generated. Provide a value to overwrite random generation
last_name: string: Sets the last_name. By default None. Random values are generated. Provide a value to overwrite random generation
......@@ -578,6 +578,12 @@ class smashProvider(BaseProvider):
user = self.createSmashUser(
username=username, email=email, password=password)
if is_super:
user.is_superuser = True
user.is_staff = True
user.is_admin = True
user.save()
# create worker
defaults = {'first_name': first_name, 'last_name': last_name,
'email': email, 'unit': unit, 'specialization': specialization,
......@@ -712,4 +718,4 @@ if __name__ == "__main__":
fake.createSmashAppointments()
fake.createSmashWorker(first_name=u'System', last_name=u'Admin',
email=u'carlos.vega@uni.lu', role=ROLE_CHOICES_TECHNICIAN, password='smashtest007',
locations=fake.getAllCreatedLocations(), languages=fake.getAllCreatedLanguages())
locations=fake.getAllCreatedLocations(), languages=fake.getAllCreatedLanguages(), is_super=True)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment