truncate_2.py 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. def handle(self, *args, **options):
  2. if options['apps'] == None or options['apps'] == [] or options['apps'] == ['']:
  3. print('No apps was entered')
  4. else:
  5. print('Fetching apps...')
  6. for someApp in options['apps']:
  7. try:
  8. self.appsList.append(apps.get_app_config(someApp))
  9. except LookupError:
  10. print("The app " + someApp + " does not exist")
  11. print('Finished apps fetching')
  12. if options['models'] == None or options['models'] == [] or options['models'] == ['']:
  13. print("No models were called")
  14. elif self.appsList:
  15. print("Fetching models...")
  16. for appConf in self.appsList:
  17. for someModel in options['models']:
  18. try:
  19. self.modelsList.append(appConf.get_model(someModel))
  20. except LookupError:
  21. print("The model " + someModel + " does not exist in this app")
  22. if not self.modelsList:
  23. self.appsList = []
  24. print('Finished models fetching')
  25. if not self.modelsList and not self.appsList:
  26. print('Nothing was called')
  27. elif self.appsList and not self.modelsList:
  28. print('Starting apps truncate..')
  29. for _app in self.appsList:
  30. app_models = _app.get_models()
  31. for _model in app_models:
  32. _model.objects.all().delete()
  33. print("The " + _app.name + " app is now empty")
  34. print('Finished apps truncate!')
  35. else:
  36. for _model in self.modelsList:
  37. _model.objects.all().delete()
  38. print('Finished models truncate!')