166 lines
4.5 KiB
Python
Executable File
166 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys,os
|
|
from datetime import datetime
|
|
import re,signal
|
|
import subprocess
|
|
|
|
VERSION=1
|
|
|
|
W= '30'
|
|
R= '31'
|
|
G= '32'
|
|
Y= '33'
|
|
B= '34'
|
|
M= '35'
|
|
C= '36'
|
|
S= '1'
|
|
E= '0'
|
|
BR= '41'
|
|
CLR = '\033[2J'
|
|
SAVE = '\033[s'
|
|
LOAD = '\033[u'
|
|
CLRLN = '\033[K'
|
|
CLRBLN = '\033[1K'
|
|
DOWN = '\033[1B'
|
|
|
|
def setup_options():
|
|
''' Setup the command line options '''
|
|
from argparse import ArgumentParser
|
|
import argparse
|
|
|
|
parser=ArgumentParser(description='''
|
|
Tool to clean up and colorize the output of Anduril.
|
|
Example: anduril run yourscript.and | %(prog)s
|
|
You can tap in to an existing log with:
|
|
tail -f -n +0 log/_global | %(prog)s''',formatter_class=argparse.RawTextHelpFormatter)
|
|
parser.add_argument("--time-stamp",'-t',action="store_true",dest="timestamp",default=False,
|
|
help="Print a timestamp on each line")
|
|
parser.add_argument("--no-colors",'--nc',action="store_false",dest="colors",default=True,
|
|
help="Disable colored output")
|
|
parser.add_argument("logfile",type=str,action="store",default="",nargs="?",
|
|
help="Log file to read, uses stdin automatically")
|
|
parser.add_argument("--version",action='version', version=VERSION)
|
|
options=parser.parse_args()
|
|
|
|
return options
|
|
def c(attribs):
|
|
''' ANSI colorizer '''
|
|
return '\033['+';'.join(attribs)+'m'
|
|
|
|
def pos(y,x):
|
|
''' ANSI absolute position set '''
|
|
return "\033["+str(y)+";"+str(x)+"H"
|
|
|
|
color_match={#'line_ends':(re.compile('$'),c.END),
|
|
'err':(re.compile('(Failed)'),c([R,S])+'\\1'+c([E])),
|
|
'done':(re.compile('(Done)'),c([G,S])+'\\1'+c([E])),
|
|
}
|
|
|
|
|
|
def colorize(string):
|
|
''' colorizes a string based on color_match '''
|
|
if not options.colors:
|
|
return string
|
|
for c in color_match:
|
|
string=color_match[c][0].sub(color_match[c][1],string)
|
|
return string
|
|
|
|
|
|
def count_running(string, stats):
|
|
''' Counts the running executions '''
|
|
|
|
spl=[i.strip() for i in string.split('|')]
|
|
if spl[3] in stats['files']:
|
|
index=stats['files'].index(spl[3])
|
|
stats['running'][index]=spl
|
|
else:
|
|
stats['running'].append(spl)
|
|
stats['running'].sort(key=lambda x: x[3])
|
|
stats['files']=[i[3] for i in stats['running']]
|
|
return stats
|
|
|
|
class EndProgram( Exception ):
|
|
''' Nice way of exiting the program '''
|
|
pass
|
|
|
|
def remove_running(stats):
|
|
''' Remove list item matching to string '''
|
|
if not stats['running']:
|
|
return stats
|
|
if len(stats['running'])>10:
|
|
for e in enumerate(stats['running']):
|
|
if e[1][2]=='Done':
|
|
stats['running'].pop(e[0])
|
|
stats['files'].pop(e[0])
|
|
return stats
|
|
|
|
|
|
def print_stats(stats):
|
|
''' Prints logged errors, and the status line '''
|
|
sys.stdout.write(SAVE)
|
|
|
|
for e in range(2):
|
|
sys.stdout.write(pos(e+1,0)+CLRLN)
|
|
sys.stdout.write(pos(e+1,0)+"="*20+" "+human_time()+CLRLN)
|
|
if (stats['running']):
|
|
sys.stdout.write(pos(e+2,0)+"Last update: "+stats['running'][0][0]+CLRLN)
|
|
else:
|
|
return
|
|
|
|
for ex in enumerate(stats['running']):
|
|
sys.stdout.write(pos(e+3+ex[0],0)+'('+str(ex[0]+1)+') '+' '.join([(ex[1][1]),colorize(ex[1][2]),ex[1][3]])+CLRLN)
|
|
for i in range(3):
|
|
sys.stdout.write(pos(e+4+ex[0],0)+CLRLN)
|
|
|
|
sys.stdout.write(DOWN+CLRBLN+CLRLN)
|
|
sys.stdout.write(LOAD)
|
|
|
|
def human_time():
|
|
t=datetime.now().strftime("%I:%M:%S %p")
|
|
return t
|
|
|
|
def readinput(lf):
|
|
try:
|
|
line = lf.stdout.readline()
|
|
#line=lf.readline()
|
|
return line
|
|
except:
|
|
return "CleanerTimeout"
|
|
|
|
options=setup_options()
|
|
|
|
stats={'time':datetime.now(),
|
|
'running':[],
|
|
'files':[]}
|
|
|
|
proc = subprocess.Popen(['aerofs-sh','transfers'],stdout=subprocess.PIPE)
|
|
|
|
sys.stdout.write(CLR)
|
|
while 1:
|
|
|
|
try:
|
|
sys.stdout.flush()
|
|
# set a 3 second timeout for the line read.
|
|
signal.signal(signal.SIGALRM, readinput)
|
|
signal.alarm(3)
|
|
line=readinput(proc)
|
|
if not line:
|
|
raise EndProgram
|
|
# timeout returns a special string, in this case we re-read
|
|
if line=="CleanerTimeout":
|
|
print_stats(stats)
|
|
continue
|
|
stats=count_running(line,stats)
|
|
# store only maximum number of error lines
|
|
# if line empty, read next
|
|
if line.strip()=="":
|
|
continue
|
|
print_stats(stats)
|
|
stats=remove_running(stats)
|
|
except EndProgram,KeyboardInterrupt:
|
|
sys.stdout.flush()
|
|
sys.exit(0)
|
|
|
|
|