mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Import constants from each app

Browse Source
This commit is contained in:
Jeremy Stretch
2017-07-24 14:22:07 -04:00
parent 05aaafc1cf
commit 091cf390d2
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
netbox/extras/management/commands/nbshell.py
View File

@ -37,9 +37,11 @@ class Command(BaseCommand):
def get_namespace(self): def get_namespace(self):
namespace = {} namespace = {}
# Gather Django models from each app # Gather Django models and constants from each app
for app in APPS: for app in APPS:
self.django_models[app] = [] self.django_models[app] = []
# Models
app_models = sys.modules['{}.models'.format(app)] app_models = sys.modules['{}.models'.format(app)]
for name in dir(app_models): for name in dir(app_models):
model = getattr(app_models, name) model = getattr(app_models, name)
@ -50,6 +52,15 @@ class Command(BaseCommand):
except TypeError: except TypeError:
pass pass
# Constants
try:
app_constants = sys.modules['{}.constants'.format(app)]
for name in dir(app_constants):
namespace[name] = getattr(app_constants, name)
except KeyError:
pass
# Load convenience commands # Load convenience commands
namespace.update({ namespace.update({
'lsmodels': self._lsmodels, 'lsmodels': self._lsmodels,