7 from monitor import database
8 from monitor.database.info.model import *
12 print "blacklist.py --delete=<i>"
19 longopts = ["delete=", "loginbase", "help"]
20 (opts, argv) = getopt.getopt(sys.argv[1:], "d:lh", longopts)
21 except getopt.GetoptError, err:
22 print "Error: " + err.msg
25 hostnames_q = BlacklistRecord.getHostnameBlacklist()
26 loginbases_q = BlacklistRecord.getLoginbaseBlacklist()
27 hostnames = [ h.hostname for h in hostnames_q ]
28 loginbases = [ h.loginbase for h in loginbases_q ]
30 for (opt, optval) in opts:
31 if opt in ["-d", "--delete"]:
33 bl = BlacklistRecord.get_by(hostname=i)
35 elif opt in ["-l", "--loginbase"]:
54 line = sys.stdin.readline()
58 if line not in hostnames and line not in loginbases:
60 bl = BlacklistRecord(loginbase=line)
62 bl = BlacklistRecord(hostname=line)
68 print "Total %d loginbases in blacklist" % (i_cnt)
70 print "Total %d nodes in blacklist" % (i_cnt)
72 if __name__ == '__main__':
76 #except Exception, error:
77 # print "Exception %s" % error