keannu125/Borrowing-TrackerBackend
1
0
Fork 0
mirror of https://github.com/lemeow125/Borrowing-TrackerBackend.git synced 2024-11-16 22:09:27 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improved postmigration data seeding for equipments

Browse source
This commit is contained in:
Keannu Bernasol 2023-12-26 19:47:51 +08:00
parent 9443631102
commit 6d25157c5f
1 changed files with 38 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
equipment_tracker/equipments/models.py
View file

@ -43,15 +43,41 @@ class EquipmentInstance(models.Model):
@receiver(post_migrate)
def create_superuser(sender, **kwargs):
if sender.name == 'equipments':
EQUIPMENT, CREATED = Equipment.objects.get_or_create(
name="Pyrex Beaker", description="", category="Glassware")
EQUIPMENT_INSTANCE, CREATED = EquipmentInstance.objects.get_or_create(
equipment=EQUIPMENT, status="Available", remarks="First beaker of equipment tracker!")
EQUIPMENT, CREATED = Equipment.objects.get_or_create(
name="Bunsen Burner", description="", category="Miscellaneous")
EQUIPMENT_INSTANCE, CREATED = EquipmentInstance.objects.get_or_create(
equipment=EQUIPMENT, status="Available", remarks="First bunsen burner of equipment tracker!")
EQUIPMENT, CREATED = Equipment.objects.get_or_create(
name="Microscope", description="", category="Miscellaneous")
EQUIPMENT_INSTANCE, CREATED = EquipmentInstance.objects.get_or_create(
equipment=EQUIPMENT, status="Available", remarks="First microscope of equipment tracker!")
equipment_data = [
{
'name': 'Pyrex Beaker',
'description': '',
'category': 'Glassware',
'remarks': 'First beaker of equipment tracker!'
},
{
'name': 'Bunsen Burner',
'description': '',
'category': 'Miscellaneous',
'remarks': 'First bunsen burner of equipment tracker!'
},
{
'name': 'Microscope',
'description': '',
'category': 'Miscellaneous',
'remarks': 'First microscope of equipment tracker!'
}
]
for data in equipment_data:
EQUIPMENT, CREATED = Equipment.objects.get_or_create(
name=data['name'],
description=data['description'],
category=data['category']
)
if (CREATED):
print('Created Equipment: ' + data['name'])
# Generate 3 equipment instances per SKU
for x in range(3):
EQUIPMENT_INSTANCE = EquipmentInstance.objects.create(
equipment=EQUIPMENT,
status='Available',
remarks=data['remarks']
)
print('Created Equipment Instances: ' +
EQUIPMENT_INSTANCE.equipment.name)