Parcourir la source

Import Django User model automatically when running nbshell

Jeremy Stretch il y a 6 ans
Parent
commit
a02ded6b01
1 fichiers modifiés avec 4 ajouts et 0 suppressions
  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 import get_version
 from django.apps import apps
 from django.apps import apps
 from django.conf import settings
 from django.conf import settings
+from django.contrib.auth.models import User
 from django.core.management.base import BaseCommand
 from django.core.management.base import BaseCommand
 
 
 APPS = ['circuits', 'dcim', 'extras', 'ipam', 'secrets', 'tenancy', 'users', 'virtualization']
 APPS = ['circuits', 'dcim', 'extras', 'ipam', 'secrets', 'tenancy', 'users', 'virtualization']
@@ -50,6 +51,9 @@ class Command(BaseCommand):
             except KeyError:
             except KeyError:
                 pass
                 pass
 
 
+        # Additional objects to include
+        namespace['User'] = User
+
         # Load convenience commands
         # Load convenience commands
         namespace.update({
         namespace.update({
             'lsmodels': self._lsmodels,
             'lsmodels': self._lsmodels,