hpcchris/sit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added option --excluldeusers ignoring users based on a given regular expression.

Browse source
This commit is contained in:
Christoph Niethammer 2014-02-13 09:21:05 +00:00
parent de52f45532
commit 64d5da9924
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
module_stats.py
View file

@ -27,6 +27,7 @@ parser.add_option("--startdate", metavar="YYYY-MM-DD", dest="startdate")
parser.add_option("--enddate", metavar="YYYY-MM-DD", dest="enddate") parser.add_option("--enddate", metavar="YYYY-MM-DD", dest="enddate")
parser.add_option("--nohpc", action="store_true", default=False, dest="nohpc", help="Exclude hpc* accounts from stats") parser.add_option("--nohpc", action="store_true", default=False, dest="nohpc", help="Exclude hpc* accounts from stats")
parser.add_option("--printusers", action="store_true", default=False, dest="printusers", help="Print detailed user lists for module usage.") parser.add_option("--printusers", action="store_true", default=False, dest="printusers", help="Print detailed user lists for module usage.")
parser.add_option("--excludeusers", metavar="REGEX", dest="excludeusers", help="Exclude users mathich given regular expression")
parser.add_option("--list-unused", action="store_true", default=False, dest="list_unused", help="List only unused modules.") parser.add_option("--list-unused", action="store_true", default=False, dest="list_unused", help="List only unused modules.")
(options, args) = parser.parse_args() (options, args) = parser.parse_args()
if options.logdir : if options.logdir :
@ -39,6 +40,8 @@ if options.enddate :
enddate = datetime.date(int(d[0:4]),int(d[5:7]),int(d[8:10])) enddate = datetime.date(int(d[0:4]),int(d[5:7]),int(d[8:10]))
if len(args) > 0 : if len(args) > 0 :
modulepattern = args[0] modulepattern = args[0]
if options.excludeusers :
excludeusers_re = re.compile(str(options.excludeusers))
for logfilename in os.listdir(logdir) : for logfilename in os.listdir(logdir) :
@ -56,6 +59,8 @@ for logfilename in os.listdir(logdir) :
user = line.split()[-3][5:] # remove user: in front user = line.split()[-3][5:] # remove user: in front
if options.nohpc and 'hpc' in user : if options.nohpc and 'hpc' in user :
continue continue
if options.excludeusers and excludeusers_re.match(user):
continue
total_users.add(user) total_users.add(user)
total_modules = total_modules + 1 total_modules = total_modules + 1
if module not in modules: if module not in modules: