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

Make display module compatible to all systems which supports .environ()

This commit is contained in:
Oleksii Shevchuk 2017-05-21 22:55:31 +03:00 committed by Oleksii Shevchuk
parent 0ab67d0a9c
commit c157039b50
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pupy/modules/display.py
View File

@ -4,7 +4,7 @@ from pupylib.utils.rpyc_utils import obtain
__class_name__="Display" __class_name__="Display"
@config(compat="linux", cat="admin") @config(compat="posix", cat="admin")
class Display(PupyModule): class Display(PupyModule):
""" Set display variable """ """ Set display variable """

6
pupy/packages/linux/all/display.py → pupy/packages/posix/all/display.py
View File

@ -93,7 +93,11 @@ def guess_displays():
userinfos = {} userinfos = {}
for process in psutil.process_iter(): for process in psutil.process_iter():
info = process.as_dict(['username', 'environ']) try:
info = process.as_dict(['username', 'environ'])
except:
continue
if info['username'] and info['environ']: if info['username'] and info['environ']:
if not 'DISPLAY' in info['environ']: if not 'DISPLAY' in info['environ']:
continue continue