Debugging tainted namespace on profile modules.

This commit is contained in:
Anton Hvornum 2021-03-21 14:34:03 +01:00
parent e1e1a916fc
commit 1fb61bf8fc
No known key found for this signature in database
GPG Key ID: F1234C5BA67C59DF
1 changed files with 2 additions and 1 deletions

View File

@ -129,7 +129,7 @@ class Script():
if namespace:
self.namespace = namespace
if '.py' in self.namespace:
if '.py.py' in self.namespace:
raise KeyError("Debugging")
self.spec = importlib.util.spec_from_file_location(self.namespace, self.path)
@ -173,6 +173,7 @@ class Profile(Script):
# trigger a traditional:
# if __name__ == 'moduleName'
if '__name__' in source_data and '_prep_function' in source_data:
print(f"Checking if {self} has _prep_function by importing with namespace {self.namespace}.py")
with self.load_instructions(namespace=f"{self.namespace}.py") as imported:
if hasattr(imported, '_prep_function'):
return True