External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-23 03:45:01 -02:30
Code Issues Packages Projects Releases Wiki Activity

data generator schema update for 3.2

Browse Source
This commit is contained in:
AlanCoding
2017-08-16 17:14:56 -04:00
parent de2aa2792a
commit d48c1bd71d
1 changed files with 25 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
tools/data_generators/rbac_dummy_data_generator.py
View File

@@ -185,19 +185,22 @@ def mock_save(self, *args, **kwargs):
return super(PrimordialModel, self).save(*args, **kwargs) return super(PrimordialModel, self).save(*args, **kwargs)
def mock_update(self):
return
def mock_computed_fields(self, **kwargs): def mock_computed_fields(self, **kwargs):
pass pass
PrimordialModel.save = mock_save PrimordialModel.save = mock_save
Project.update = mock_update
startTime = datetime.now() startTime = datetime.now()
try: def make_the_data():
with disable_activity_stream():
with transaction.atomic():
with batch_role_ancestor_rebuilding(), disable_computed_fields(): with batch_role_ancestor_rebuilding(), disable_computed_fields():
admin, created = User.objects.get_or_create(username = 'admin', is_superuser=True) admin, created = User.objects.get_or_create(username = 'admin', is_superuser=True)
if created: if created:
@@ -479,10 +482,6 @@ try:
if org_inv_count > 0: if org_inv_count > 0:
inventory = project.organization.inventories.all()[inv_idx % org_inv_count] inventory = project.organization.inventories.all()[inv_idx % org_inv_count]
extra_kwargs = {} extra_kwargs = {}
if ids['job_template'] % 5 == 0:
extra_kwargs['cloud_credential'] = next(credential_gen)
if ids['job_template'] % 7 == 0:
extra_kwargs['network_credential'] = next(credential_gen)
job_template, _ = JobTemplate.objects.get_or_create( job_template, _ = JobTemplate.objects.get_or_create(
name='%s Job Template %d Project %d' % (prefix, job_template_id, project_idx), name='%s Job Template %d Project %d' % (prefix, job_template_id, project_idx),
@@ -495,6 +494,10 @@ try:
playbook="debug.yml", playbook="debug.yml",
**extra_kwargs) **extra_kwargs)
) )
if ids['job_template'] % 7 == 0:
job_template.extra_credentials.add(next(credential_gen))
if ids['job_template'] % 5 == 0: # formerly cloud credential
job_template.extra_credentials.add(next(credential_gen))
job_template._is_new = _ job_template._is_new = _
job_templates.append(job_template) job_templates.append(job_template)
inv_idx += 1 inv_idx += 1
@@ -646,9 +649,9 @@ try:
status=job_stat, name="%s-%d" % (job_template.name, job_i), status=job_stat, name="%s-%d" % (job_template.name, job_i),
project=job_template.project, inventory=job_template.inventory, project=job_template.project, inventory=job_template.inventory,
credential=job_template.credential, credential=job_template.credential,
cloud_credential=job_template.cloud_credential,
network_credential=job_template.network_credential
) )
for ec in job_template.extra_credentials.all():
job.extra_credentials.add(ec)
job._is_new = _ job._is_new = _
jobs.append(job) jobs.append(job)
job_i += 1 job_i += 1
@@ -707,11 +710,18 @@ try:
if n: if n:
print('') print('')
if options['pretend']: if options['pretend']:
raise Rollback() with transaction.atomic():
except Rollback: try:
print('Rolled back changes') make_the_data()
pass raise Rollback()
except Rollback:
print('Rolled back changes')
pass
else:
make_the_data()
print('') print('')
print('script execution time: {}'.format(datetime.now() - startTime)) print('script execution time: {}'.format(datetime.now() - startTime))