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

Colorize output depending on idle

This commit is contained in:
Oleksii Shevchuk 2017-03-16 18:07:51 +02:00
parent 8be925b74c
commit 398542404b
1 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
pupy/modules/w.py
View File

@ -25,22 +25,32 @@ class WModule(PupyModule):
def run(self, args):
try:
data = obtain(self.client.conn.modules.pupyps.users())
tablein = []
for user, hosts in reversed(sorted(data.iteritems())):
for host, sessions in hosts.iteritems():
for session in sessions:
color = ""
if 'idle' in session:
idle = session['idle']
color = "cyan" if idle < 10*60 else (
"grey" if idle > 60*60*24 else ""
)
object = {
'HOST': host,
'HOST': colorize(host, color),
'USER': colorize(
user,
"yellow" if user in ADMINS else (
"green" if session.get('me') else "")
),
'LOGIN': str(datetime.fromtimestamp(int(session['started']))),
'LOGIN': colorize(
str(datetime.fromtimestamp(int(session['started']))), color
),
}
if session.get('terminal'):
if session.get('name'):
what = '{} {}'.format(
@ -49,20 +59,16 @@ class WModule(PupyModule):
)
else:
what = ''
object.update({
'IDLE': str(timedelta(seconds=session['idle'])),
'PID': session.get('pid', ''),
'WHAT': what
'IDLE': colorize(str(timedelta(seconds=session['idle'])), color),
'PID': colorize(str(session.get('pid', '')), color),
'WHAT': colorize(what, color)
})
tablein.append(object)
self.stdout.write((PupyCmd.table_format(tablein)))
except Exception, e:
logging.exception(e)