arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-18 05:52:15 +00:00
Code Issues Projects Releases Wiki Activity

Fix sample data command

Browse Source
This commit is contained in:
FreneticScribbler
2022-01-04 12:28:25 +00:00
parent 215c51e718
commit 7aa37c225d
3 changed files with 9440 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
RIGS/management/commands/generateSampleData.py
View File

@@ -12,5 +12,4 @@ class Command(BaseCommand):
call_command('generateSampleUserData')
call_command('generateSampleRIGSData')
call_command('generateSampleAssetsData')
call_command('import_old_db')
call_command('generate_sample_training_users')
call_command('generateSampleTrainingData')

9656
package-lock.json generated
View File

File diff suppressed because it is too large Load Diff

10
training/management/commands/generateSampleTrainingData.py
View File

@@ -32,7 +32,7 @@ class Command(BaseCommand):
self.setup_categories()
self.setup_items()
self.setup_levels()
call_command('generate_sample_training_users')
# call_command('generate_sample_training_users')
print("Done generating training data")
def setup_categories(self):
@@ -144,7 +144,13 @@ class Command(BaseCommand):
"Wiki Editing"]
for i, name in enumerate(names):
item = models.TrainingItem.objects.create(category=random.choice(self.categories), reference_number=random.randint(0, 100), name=name)
category = random.choice(self.categories)
previous_item = models.TrainingItem.objects.filter(category=category).last()
if previous_item is not None:
number = previous_item.reference_number + 1
else:
number = 0
item = models.TrainingItem.objects.create(category=category, reference_number=number, name=name)
self.items.append(item)
def setup_levels(self):