KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Restore sys.path after importing custom kitten

Browse Source
This commit is contained in:
Kovid Goyal 2018-07-25 09:51:04 +05:30
parent 21dbb0e820
commit 57871e1983
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
kittens/runner.py
View File

@ -17,18 +17,22 @@ def resolved_kitten(k):
def import_kitten_main_module(config_dir, kitten): def import_kitten_main_module(config_dir, kitten):
if kitten.endswith('.py'): if kitten.endswith('.py'):
path_modified = False
path = os.path.expanduser(kitten) path = os.path.expanduser(kitten)
if not os.path.isabs(path): if not os.path.isabs(path):
path = os.path.join(config_dir, path) path = os.path.join(config_dir, path)
path = os.path.abspath(path) path = os.path.abspath(path)
if os.path.dirname(path): if os.path.dirname(path):
sys.path.insert(0, os.path.dirname(path)) sys.path.insert(0, os.path.dirname(path))
path_modified = True
with open(path) as f: with open(path) as f:
src = f.read() src = f.read()
code = compile(src, path, 'exec') code = compile(src, path, 'exec')
g = {'__name__': 'kitten'} g = {'__name__': 'kitten'}
exec(code, g) exec(code, g)
hr = g.get('handle_result', lambda *a, **kw: None) hr = g.get('handle_result', lambda *a, **kw: None)
if path_modified:
del sys.path[0]
return {'start': g['main'], 'end': hr} return {'start': g['main'], 'end': hr}
kitten = resolved_kitten(kitten) kitten = resolved_kitten(kitten)