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

FIX: Fix asset sample data command when run alone

Browse Source
This commit is contained in:
FreneticScribbler
2020-02-17 16:29:25 +00:00
parent 68891dccd2
commit 6e15f12fbf
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
assets/management/commands/generateSampleAssetsData.py
View File

@@ -17,6 +17,7 @@ class Command(BaseCommand):
random.seed('Some object to see the random number generator') random.seed('Some object to see the random number generator')
self.create_profile()
self.create_categories() self.create_categories()
self.create_statuses() self.create_statuses()
self.create_suppliers() self.create_suppliers()
@@ -24,6 +25,13 @@ class Command(BaseCommand):
self.create_connectors() self.create_connectors()
self.create_cables() self.create_cables()
# Make sure that there's at least one profile if this command is run standalone
def create_profile(self):
name = "Fred Johnson"
models.Profile.objects.create(username=name.replace(" ", ""), first_name=name.split(" ")[0], last_name=name.split(" ")[-1],
email=name.replace(" ", "") + "@example.com",
initials="".join([j[0].upper() for j in name.split()]))
def create_categories(self): def create_categories(self):
categories = ['Case', 'Video', 'General', 'Sound', 'Lighting', 'Rigging'] categories = ['Case', 'Video', 'General', 'Sound', 'Lighting', 'Rigging']