예제 #1
0
    def discover_schemas(self):
	if not self.get_self_db_conn():
	    return
	cur=self.prod_conn.cursor()
	try:
	    cur.execute("SELECT oid,nspname FROM pg_namespace WHERE nspname NOT IN ('pg_catalog', 'information_schema') AND nspname !~ '^pg_toast' AND nspname !~ '^pg_temp'")
	except Exception as e:
	    logger.error("Canot get schema info for database {0}".format(self.db_fields['db_name']))
	    cur.close()
	    return
	prod_schs=cur.fetchall()
	cur.close()
	self.cursor.execute("SELECT obj_oid,sch_name,id FROM schema_name WHERE dn_id={0} AND alive".format(self.id))
	local_schs=self.cursor.fetchall()
	for l_sch in local_schs:
	    for p_sch in prod_schs:
		if l_sch[0]==p_sch[0] and l_sch[1]==p_sch[1]:
		    break
	    else:
		old.sch=SchemaName(l_sch[2])
		old_sch.retire()
		logger.info("Retired schema {0} in database {1}".format(l_sch[1],self.db_fields['db_name']))
	for p_sch in prod_schs:
	    for l_sch in local_schs:
		if l_sch[0]==p_sch[0] and l_sch[1]==p_sch[1]:
		    break
	    else:
		new_sch=SchemaName()
		new_sch.set_fields(dn_id=self.id,obj_oid=p_sch[0],sch_name=p_sch[1])
		new_sch.create()
		new_sch.truncate()
		logger.info("Create new schema {0} in database {1}".format(p_sch[1],self.db_fields['db_name']))
예제 #2
0
    def discover_schemas(self):
	cur=self._get_p_cursor()
	if not cur:
	    logger.error("Return from DN discover_schemas. No p_cur obtained")
	    return False
	try:
	    cur.execute("SELECT oid,nspname FROM pg_namespace WHERE nspname NOT IN ('pg_catalog', 'information_schema') AND nspname !~ '^pg_toast' AND nspname !~ '^pg_temp'")
	except Exception as e:
	    logger.error("Canot execute schema discovery query on Prod {0}".format(e.pgerror))
	    cur.close()
	    return False
	prod_schs=cur.fetchall()
	cur.close()
	cur=self.db_conn.cursor()
	try:
	    cur.execute("SELECT obj_oid,sch_name,id FROM schema_name WHERE dn_id={0} AND alive".format(self.id))
	except Exception as e:
	    logger.error("Canot execute schema discovery query on Local {0}".format(e.pgerror))
	    cur.close()
	    return False
	local_schs=cur.fetchall()
	cur.close()
	for l_sch in local_schs:
	    for p_sch in prod_schs:
		if l_sch[0]==p_sch[0] and l_sch[1]==p_sch[1]:
		    old_sch=SchemaName(self.db_conn,self.prod_conn,l_sch[2])
		    if not old_sch.discover_tables():
			logger.error("DN.discover_schemas() False from discover_tables for old")
		    if not old_sch.discover_functions():
			logger.error("DN.discover_schemas() False from discover_functions for old")
		    break
	    else:
		old_sch=SchemaName(self.db_conn,self.prod_conn,l_sch[2])
		if old_sch.retire():
		    logger.info("Retired schema {0} in database {1}".format(l_sch[1],self.db_fields['db_name']))
		else:
		    logger.error("DN.discover_schemas() Cannot retire old")
	for p_sch in prod_schs:
	    for l_sch in local_schs:
		if l_sch[0]==p_sch[0] and l_sch[1]==p_sch[1]:
		    break
	    else:
		new_sch=SchemaName(self.db_conn,self.prod_conn)
		new_sch.set_fields(dn_id=self.id,obj_oid=p_sch[0],sch_name=p_sch[1])
		if new_sch._create():
		    if not new_sch.discover_tables():
			logger.error("DN.discover_schemas() False from discover_tables for new")
		    if not new_sch.discover_functions():
			logger.error("DN.discover_schemas() False from discover_functions for new")
		    logger.info("Create new schema {0} in database {1}".format(p_sch[1],self.db_fields['db_name']))
		else:
		    logger.error("DN.discover_schemas() Cannot create new")
	return True