mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 13:32:15 +00:00
24 lines
696 B
Python
24 lines
696 B
Python
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
from assets import models
|
|
|
|
|
|
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)
|
|
|
|
def delete_objects(self, model):
|
|
for object in model.objects.all():
|
|
object.delete()
|