NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-23 04:22:01 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Closes #17482: Ignore Branch & StagedChange in nbshell

Browse Source
This commit is contained in:
Jeremy Stretch 2024-09-12 16:27:21 -04:00
parent fa1e89d3d2
commit 65f81ab81d
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
netbox/core/management/commands/nbshell.py
View File

@ -11,6 +11,10 @@ from core.models import ObjectType
from users.models import User from users.models import User
APPS = ('circuits', 'core', 'dcim', 'extras', 'ipam', 'tenancy', 'users', 'virtualization', 'vpn', 'wireless') APPS = ('circuits', 'core', 'dcim', 'extras', 'ipam', 'tenancy', 'users', 'virtualization', 'vpn', 'wireless')
EXCLUDE_MODELS = (
'extras.branch',
'extras.stagedchange',
)
BANNER_TEXT = """### NetBox interactive shell ({node}) BANNER_TEXT = """### NetBox interactive shell ({node})
### Python {python} | Django {django} | NetBox {netbox} ### Python {python} | Django {django} | NetBox {netbox}
@ -44,12 +48,16 @@ class Command(BaseCommand):
# Gather Django models and constants from each app # Gather Django models and constants from each app
for app in APPS: for app in APPS:
self.django_models[app] = [] models = []
# Load models from each app # Load models from each app
for model in apps.get_app_config(app).get_models(): for model in apps.get_app_config(app).get_models():
namespace[model.__name__] = model app_label = model._meta.app_label
self.django_models[app].append(model.__name__) model_name = model._meta.model_name
if f'{app_label}.{model_name}' not in EXCLUDE_MODELS:
namespace[model.__name__] = model
models.append(model.__name__)
self.django_models[app] = sorted(models)
# Constants # Constants
try: try: