arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-27 10:22:17 +00:00
Code Issues Projects Releases Wiki Activity

Initial work on requirements importer

Browse Source
This commit is contained in:
FreneticScribbler
2021-12-21 13:07:45 +00:00
parent 522837c64e
commit ce5efff268
1 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
training/management/commands/import_old_db.py
View File

@@ -20,6 +20,7 @@ class Command(BaseCommand):
self.import_TrainingItemQualification() self.import_TrainingItemQualification()
self.import_TrainingLevel() self.import_TrainingLevel()
self.import_TrainingLevelQualification() self.import_TrainingLevelQualification()
self.import_TrainingLevelRequirements()
@staticmethod @staticmethod
def xml_path(file): def xml_path(file):
@@ -172,9 +173,13 @@ class Command(BaseCommand):
level = models.TrainingLevel.TA level = models.TrainingLevel.TA
depString = None depString = None
desc = ""
if child.find('Desc'):
desc = child.find('Desc').text
obj, created = models.TrainingLevel.objects.update_or_create( obj, created = models.TrainingLevel.objects.update_or_create(
pk=int(child.find('ID').text), pk=int(child.find('ID').text),
description = name, description = desc,
level = level level = level
) )
if depString is not None: if depString is not None:
@@ -218,3 +223,24 @@ class Command(BaseCommand):
print("Training Level Qualification #{} is duplicate. ಠ_ಠ".format(child.find('ID').text)) print("Training Level Qualification #{} is duplicate. ಠ_ಠ".format(child.find('ID').text))
print('TrainingLevelQualifications - Updated: {}, Created: {}'.format(tally[0], tally[1])) print('TrainingLevelQualifications - Updated: {}, Created: {}'.format(tally[0], tally[1]))
def import_TrainingLevelRequirements(self):
tally = [0, 0]
root = self.parse_xml(self.xml_path('Training Level Requirements.xml'))
for child in root:
try:
item = child.find('Item').text.split(".")
obj, created = models.TrainingLevelRequirement.objects.update_or_create(level=models.TrainingLevel.objects.get(pk=int(child.find('Level').text)),item=models.TrainingItem.objects.get(reference_number=item[1], category=models.TrainingCategory.objects.get(reference_number=item[0])), depth=int(child.find('Depth').text))
if created:
tally[1] += 1
else:
tally[0] += 1
except models.TrainingItem.DoesNotExist:
print("Item with number {} does not exist".format(item))
except models.TrainingItem.MultipleObjectsReturned:
print(models.TrainingItem.objects.filter(reference_number=item[1], category=models.TrainingCategory.objects.get(reference_number=item[0])))
print('TrainingLevelRequirements - Updated: {}, Created: {}'.format(tally[0], tally[1]))