Explorar o código

Import Django User model automatically when running nbshell

Jeremy Stretch %!s(int64=6) %!d(string=hai) anos
pai
achega
a02ded6b01
Modificáronse 1 ficheiros con 4 adicións e 0 borrados
  1. 4 0
      netbox/extras/management/commands/nbshell.py

+ 4 - 0
netbox/extras/management/commands/nbshell.py

@@ -5,6 +5,7 @@ import sys
 from django import get_version
 from django.apps import apps
 from django.conf import settings
+from django.contrib.auth.models import User
 from django.core.management.base import BaseCommand
 
 APPS = ['circuits', 'dcim', 'extras', 'ipam', 'secrets', 'tenancy', 'users', 'virtualization']
@@ -50,6 +51,9 @@ class Command(BaseCommand):
             except KeyError:
                 pass
 
+        # Additional objects to include
+        namespace['User'] = User
+
         # Load convenience commands
         namespace.update({
             'lsmodels': self._lsmodels,