33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
'''
|
|
Instead of adding every program's directory to my system PATH, I prefer having
|
|
a single directory on my PATH which contains Windows shortcuts (lnk files)
|
|
pointing to each program.
|
|
Windows lnks are different from softlinks because they maintain the program's
|
|
real directory necessary for loading nearby dlls etc.
|
|
However, this breaks `shutil.which` --> `subprocess.run` because subprocess.run
|
|
with shell=False does not interpret lnks, and needs a direct path to the exe.
|
|
|
|
So, this module provides a function `which` that if it finds an lnk file, will
|
|
return the exe path, otherwise behaves the same as normal shutil which.
|
|
'''
|
|
import os
|
|
import shutil
|
|
try:
|
|
import winshell
|
|
# This fixes some "CoInitialize has not been called" errors which I found
|
|
# were occuring intermittently on some subprocess calls.
|
|
winshell.pythoncom.CoInitialize()
|
|
except ImportError:
|
|
winshell = None
|
|
|
|
def which(cmd, *args, **kwargs):
|
|
path = shutil.which(cmd, *args, **kwargs)
|
|
|
|
if path is None:
|
|
return None
|
|
|
|
if os.name == 'nt' and winshell and path.lower().endswith('.lnk'):
|
|
path = winshell.Shortcut(path).path
|
|
|
|
return path
|