瀏覽代碼

Import Django User model automatically when running nbshell

Jeremy Stretch 6 年之前
父節點
當前提交
a02ded6b01
共有 1 個文件被更改,包括 4 次插入0 次删除
  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,