Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Ensure to reload terminal() patch

This commit is contained in:
Oleksii Shevchuk 2019-04-15 13:00:54 +03:00
parent 481edb4cdd
commit d3d917b0bc
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pupy/packages/all/pupyps.py
View File

@ -36,15 +36,14 @@ KNOWN_FIELDS = tuple(
if os.name == 'nt':
try:
import pupwinutils.security
if hasattr(pupwinutils.security, 'StationNameByPid'):
if 'terminal' not in psutil._as_dict_attrnames:
from pupwinutils import security
def terminal(self):
return pupwinutils.security.StationNameByPid(self.pid)
if hasattr(security, 'StationNameByPid'):
def terminal(self):
return security.StationNameByPid(self.pid)
setattr(psutil.Process, 'terminal', terminal)
psutil._as_dict_attrnames.add('terminal')
setattr(psutil.Process, 'terminal', terminal)
psutil._as_dict_attrnames.add('terminal')
except ImportError:
pass