NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2026-01-23 12:08:43 -06:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Init storage at class level of BaseScript instead of in findsource function

Browse Source
This commit is contained in:
Brian Tiemann
2025-10-14 13:38:20 -04:00
parent 2edfde5753
commit 2580485faf
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
netbox/extras/scripts.py
View File

@@ -329,6 +329,9 @@ class BaseScript:
# Declare the placeholder for the current request # Declare the placeholder for the current request
self.request = None self.request = None
# Initiate the storage backend (local, S3, etc) as a class attr
self.storage = storages.create_storage(storages.backends["scripts"])
# Compile test methods and initialize results skeleton # Compile test methods and initialize results skeleton
for method in dir(self): for method in dir(self):
if method.startswith('test_') and callable(getattr(self, method)): if method.startswith('test_') and callable(getattr(self, method)):
@@ -394,8 +397,7 @@ class BaseScript:
return inspect.getfile(self.__class__) return inspect.getfile(self.__class__)
def findsource(self, object): def findsource(self, object):
storage = storages.create_storage(storages.backends["scripts"]) with self.storage.open(os.path.basename(self.filename), 'r') as f:
with storage.open(os.path.basename(self.filename), 'r') as f:
data = f.read() data = f.read()
# Break the source code into lines # Break the source code into lines