X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=findbadpcu.py;fp=findbadpcu.py;h=ca653442ebc50fb329a9be4afd945ddc23303f47;hb=ee740a3ff286a9720cd1656cd60a3c85f0f14b29;hp=e3d160ddb3455b2828dc08a9081ffc4945fca426;hpb=fcbf7923a944e57c789a608b31fbb9b75ff712b1;p=monitor.git

diff --git a/findbadpcu.py b/findbadpcu.py
index e3d160d..ca65344 100755
--- a/findbadpcu.py
+++ b/findbadpcu.py
@@ -12,6 +12,7 @@ import sets
     
 import signal
 import traceback
+from nodequery import pcu_select
 
 #old_handler = signal.getsignal(signal.SIGCHLD)
 
@@ -329,7 +330,7 @@ def checkAndRecordState(l_pcus, cohash):
 	global count
 	global_round = externalState['round']
 
-	tp = threadpool.ThreadPool(20)
+	tp = threadpool.ThreadPool(10)
 
 	# CREATE all the work requests
 	for pcuname in l_pcus:
@@ -390,6 +391,11 @@ def main():
 		pcus = []
 		for node in l_nodes:
 			pcus += node['pcu_ids']
+		# clear out dups.
+		l_pcus = [pcu for pcu in sets.Set(pcus)]
+	elif config.pcuselect is not None:
+		n, pcus = pcu_select(config.pcuselect)
+		# clear out dups.
 		l_pcus = [pcu for pcu in sets.Set(pcus)]
 
 	elif config.nodelist == None and config.pcuid == None:
@@ -421,6 +427,7 @@ if __name__ == '__main__':
 	parser.set_defaults(nodelist=None, 
 						increment=False, 
 						pcuid=None,
+						pcuselect=None,
 						site=None,
 						dbname="findbadpcus", 
 						cachenodes=False,
@@ -430,6 +437,8 @@ if __name__ == '__main__':
 						help="Provide the input file for the node list")
 	parser.add_option("", "--site", dest="site", metavar="FILE", 
 						help="Get all pcus associated with the given site's nodes")
+	parser.add_option("", "--pcuselect", dest="pcuselect", metavar="FILE", 
+						help="Query string to apply to the findbad pcus")
 	parser.add_option("", "--pcuid", dest="pcuid", metavar="id", 
 						help="Provide the id for a single pcu")