File tree Expand file tree Collapse file tree 1 file changed +14
-2
lines changed
graalpython/lib-graalpython/modules Expand file tree Collapse file tree 1 file changed +14
-2
lines changed Original file line number Diff line number Diff line change 48
48
import sys
49
49
50
50
51
+ def get_module_name (package_name ):
52
+ non_standard_packages = {
53
+ 'pyyaml' :'pyaml' ,
54
+ 'protobuf' :'google.protobuf' ,
55
+ 'python-dateutil' :'dateutil' ,
56
+ 'websocket-client' :'websocket' ,
57
+ }
58
+ module_name = non_standard_packages .get (package_name , package_name )
59
+ return module_name .replace ('-' , '_' )
60
+
61
+
51
62
def pip_package (name = None ):
52
63
def decorator (func ):
53
64
def wrapper (* args , ** kwargs ):
54
65
_name = name if name else func .__name__
55
66
try :
56
- importlib .import_module (_name )
57
- del sys .modules [_name ]
67
+ module_name = get_module_name (_name )
68
+ importlib .import_module (module_name )
69
+ del sys .modules [module_name ]
58
70
except ImportError :
59
71
print ("Installing required dependency: {}" .format (_name ))
60
72
func (* args , ** kwargs )
You can’t perform that action at this time.
0 commit comments