mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-19 14:32:16 +00:00
Optimise generateSampleData by ~10x
Does remove reversion creation for now...
This commit is contained in:
36
RIGS/management/commands/deleteSampleData.py
Normal file
36
RIGS/management/commands/deleteSampleData.py
Normal file
@@ -0,0 +1,36 @@
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
|
||||
from django.contrib.auth.models import Group
|
||||
from assets import models
|
||||
from RIGS import models as rigsmodels
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'Deletes testing sample data'
|
||||
|
||||
def handle(self, *args, **kwargs):
|
||||
from django.conf import settings
|
||||
|
||||
if not (settings.DEBUG):
|
||||
raise CommandError('You cannot run this command in production')
|
||||
|
||||
self.delete_objects(models.AssetCategory)
|
||||
self.delete_objects(models.AssetStatus)
|
||||
self.delete_objects(models.Supplier)
|
||||
self.delete_objects(models.Connector)
|
||||
self.delete_objects(models.Asset)
|
||||
self.delete_objects(rigsmodels.VatRate)
|
||||
self.delete_objects(rigsmodels.Profile)
|
||||
self.delete_objects(rigsmodels.Person)
|
||||
self.delete_objects(rigsmodels.Organisation)
|
||||
self.delete_objects(rigsmodels.Venue)
|
||||
self.delete_objects(Group)
|
||||
self.delete_objects(rigsmodels.Event)
|
||||
self.delete_objects(rigsmodels.EventItem)
|
||||
self.delete_objects(rigsmodels.Invoice)
|
||||
self.delete_objects(rigsmodels.Payment)
|
||||
self.delete_objects(rigsmodels.RiskAssessment)
|
||||
self.delete_objects(rigsmodels.EventChecklist)
|
||||
|
||||
def delete_objects(self, model):
|
||||
for obj in model.objects.all():
|
||||
obj.delete()
|
||||
Reference in New Issue
Block a user