def _set_pg_mon_conn(self):
try:
self.pg_mon_conn=psycopg2.connect(settings.custom_dsn('pg_mon'))
except Exception as e:
logger.critical("Cannot connect to pg_mon DB: {0}".format(e))
return False
logger.debug("Successfully connected to pg_mon database")
return True
args=parser.parse_args()
ip_regexp = r"^(([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])\.){3}([0-9]|[1-9][0-9]|1[0-9]{2}|2[0-4][0-9]|25[0-5])$";
hostname_regexp = r"^(([a-zA-Z]|[a-zA-Z][a-zA-Z0-9\-]*[a-zA-Z0-9])\.)*([A-Za-z]|[A-Za-z][A-Za-z0-9\-]*[A-Za-z0-9])$";
ip_address_pattern=re.compile(ip_regexp)
hostname_pattern=re.compile(hostname_regexp)
if not re.match(ip_address_pattern,args.ip_address):
print "Wrong IP address format"
sys.exit(1)
if not re.match(hostname_pattern,args.hostname):
print "Wrong hostname format"
conn=psycopg2.connect(settings.custom_dsn('pg_mon'))
hc=hostcluster.HostCluster(conn)
string="hc.add('"+args.ip_address+"','"+args.hostname+"'"
del(args.hostname)
del(args.ip_address)
for a,v in vars(args).iteritems():
if v:
string += ","+a+"="
if type(v) is str:
string += "'"+str(v)+"'"
elif type(v) is int or bool:
string+=str(v)
string+=")"
parser.add_argument('-s',nargs=1,choices=['file','SQLite'],default='file',dest='source',
help="Source for updates. SQLite is not supported currently")
parser.add_argument('-v','--verbose',action='store_true',default=False,help="Show additional info on terminal")
parser.add_argument('-l','--last',dest='last_applied',action='store_true',default=False,
help="Show last applied update number and exit")
args=parser.parse_args()
if args.install:
args.dry_run=True
try:
conn=psycopg2.connect(settings.custom_dsn('db_handler_1'))
except Exception as e:
print "ERROR! Cannot connect to database {0}".format(e)
sys.exit(1)
db_st=DbState(conn)
installed=db_st.get_installed()
#last_applied=db_st.get_last_applied()
#if last_applied == -1:
# conn.close()
# sys.exit()
