luke
/
OpenRAM
mirror of https://github.com/VLSIDA/OpenRAM.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix path search in custom module finder

This commit is contained in:
Eren Dogan 2023-04-10 10:28:15 -07:00
parent 095e0baddd
commit 7af2badf31
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
__init__.py
View File

@ -62,9 +62,9 @@ class custom_module_finder(MetaPathFinder):
return None return None
# Check if this module can be custom # Check if this module can be custom
from openram import OPTS from openram import OPTS
if module_name in OPTS.__dict__.keys(): if hasattr(OPTS, module_name):
# Get custom name from OPTS # Get custom name from OPTS
custom_name = OPTS.__dict__[module_name] custom_name = getattr(OPTS, module_name)
# Skip if the module is default, it will be handled by Python # Skip if the module is default, it will be handled by Python
if custom_name == module_name: if custom_name == module_name:
return None return None
@ -72,6 +72,9 @@ class custom_module_finder(MetaPathFinder):
from importlib.util import spec_from_file_location from importlib.util import spec_from_file_location
# Try to find the module in sys.path # Try to find the module in sys.path
for path in sys.path: for path in sys.path:
# Skip this path if not directory
if not os.path.isdir(path):
continue
for file in os.listdir(path): for file in os.listdir(path):
# If there is a script matching the custom module name, # If there is a script matching the custom module name,
# import it with the default module name # import it with the default module name