本文整理汇总了Python中regression.python_test_utils.test_utils.get_db_connection函数的典型用法代码示例。如果您正苦于以下问题:Python get_db_connection函数的具体用法?Python get_db_connection怎么用?Python get_db_connection使用的例子?那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。
在下文中一共展示了get_db_connection函数的15个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: verify_user_mapping
def verify_user_mapping(server, db_name, fsrv_name):
"""
This function will verify current foreign server.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param fsrv_name: FS name
:type fsrv_name: str
:return user_mapping: user mapping record
:rtype: tuple
"""
try:
connection = get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute(
"select umid from pg_user_mappings where srvname = '%s' order by"
" umid asc limit 1" % fsrv_name)
user_mapping = pg_cursor.fetchone()
connection.close()
return user_mapping
except Exception:
traceback.print_exc(file=sys.stderr)
示例2: verify_synonym
def verify_synonym(server, db_name, synonym_name):
"""
This function create the synonym on given schema node.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param synonym_name: synonym name
:type synonym_name: str
:return synonym: synonym record from database
:rtype: tuple
"""
try:
connection = utils.get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'])
pg_cursor = connection.cursor()
pg_cursor.execute("SELECT * FROM pg_synonym WHERE synname='%s'" %
synonym_name)
synonym = pg_cursor.fetchone()
connection.close()
return synonym
except Exception:
traceback.print_exc(file=sys.stderr)
示例3: delete_pgagent_job
def delete_pgagent_job(self):
"""
This function deletes the pgAgent job.
"""
try:
connection = utils.get_db_connection(
self.server['db'],
self.server['username'],
self.server['db_password'],
self.server['host'],
self.server['port'],
self.server['sslmode']
)
old_isolation_level = connection.isolation_level
connection.set_isolation_level(0)
pg_cursor = connection.cursor()
pg_cursor.execute(
"DELETE FROM pgagent.pga_job "
"WHERE jobid = '%s'::integer;" % self.job_id
)
connection.set_isolation_level(old_isolation_level)
connection.commit()
connection.close()
except Exception:
traceback.print_exc(file=sys.stderr)
示例4: create_role
def create_role(server, role_name):
"""
This function create the role.
:param server:
:param role_name:
:return:
"""
try:
connection = utils.get_db_connection(server['db'],
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute("CREATE ROLE %s LOGIN" % role_name)
connection.commit()
# Get 'oid' from newly created tablespace
pg_cursor.execute(
"SELECT pr.oid from pg_catalog.pg_roles pr WHERE pr.rolname='%s'" %
role_name)
oid = pg_cursor.fetchone()
role_id = ''
if oid:
role_id = oid[0]
connection.close()
return role_id
except Exception as exception:
exception = "Error while deleting role: %s: line:%s %s" % (
file_name, sys.exc_traceback.tb_lineno, exception)
print(exception, file=sys.stderr)
示例5: create_trigger_function_with_trigger
def create_trigger_function_with_trigger(server, db_name, schema_name,
func_name):
"""This function add the trigger function to schema"""
try:
connection = utils.get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
query = "CREATE FUNCTION " + schema_name + "." + func_name + \
"()" \
" RETURNS trigger LANGUAGE 'plpgsql' STABLE LEAKPROOF" \
" SECURITY DEFINER SET enable_sort=true AS $BODY$ BEGIN" \
" NULL; END; $BODY$"
pg_cursor.execute(query)
connection.commit()
# Get 'oid' from newly created function
pg_cursor.execute("SELECT pro.oid, pro.proname FROM"
" pg_proc pro WHERE pro.proname='%s'" %
func_name)
functions = pg_cursor.fetchone()
connection.close()
return functions
except Exception:
traceback.print_exc(file=sys.stderr)
示例6: runTest
def runTest(self):
""" This function will update added cast."""
self.server_id = self.database_info["server_id"]
self.db_id = self.database_info['db_id']
db_con = database_utils.connect_database(self,
utils.SERVER_GROUP,
self.server_id,
self.db_id)
if not db_con["info"] == "Database connected.":
raise Exception("Could not connect to database.")
connection = utils.get_db_connection(self.server['db'],
self.server['username'],
self.server['db_password'],
self.server['host'],
self.server['port'])
response = cast_utils.verify_cast(connection, self.source_type,
self.target_type)
if len(response) == 0:
raise Exception("Could not find cast.")
data = {
"description": "This is cast update comment",
"id": self.cast_id
}
put_response = self.tester.put(
self.url + str(utils.SERVER_GROUP) + '/' +
str(self.server_id) + '/' + str(
self.db_id) +
'/' + str(self.cast_id),
data=json.dumps(data),
follow_redirects=True)
self.assertEquals(put_response.status_code, 200)
示例7: drop_extension
def drop_extension(server, db_name, extension_name):
"""
This function used to drop the extension.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param extension_name: extension name
:type extension_name: str
:return: None
"""
try:
connection = get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute(
"SELECT * FROM pg_extension WHERE extname='%s'"
% extension_name)
if pg_cursor.fetchall():
pg_cursor.execute(
"DROP EXTENSION %s CASCADE" % extension_name)
connection.commit()
connection.close()
except Exception:
traceback.print_exc(file=sys.stderr)
示例8: create_resource_groups
def create_resource_groups(server, resource_group_name):
"""
This function create the resource groups into databases.
:param server: server details
:type server: dict
:param resource_group_name: resource group name
:type resource_group_name: str
:return resource_group_id: resource group id
:rtype: int
"""
try:
connection = utils.get_db_connection(server['db'],
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
old_isolation_level = connection.isolation_level
connection.set_isolation_level(0)
pg_cursor = connection.cursor()
pg_cursor.execute("CREATE RESOURCE GROUP %s" % resource_group_name)
connection.set_isolation_level(old_isolation_level)
connection.commit()
# Get oid of newly created resource group
pg_cursor.execute("SELECT oid FROM edb_resource_group WHERE "
"rgrpname='%s'" % resource_group_name)
resource_group = pg_cursor.fetchone()
resource_group_id = resource_group[0]
connection.close()
return resource_group_id
except Exception:
traceback.print_exc(file=sys.stderr)
示例9: verify_resource_group
def verify_resource_group(server, resource_group_name):
"""
This function verifies the resource group exist in database or not.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param resource_group_name: resource group name
:type resource_group_name: str
:return:
"""
try:
connection = utils.get_db_connection(server['db'],
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute("SELECT * FROM edb_resource_group WHERE "
"rgrpname='%s'" % resource_group_name)
resource_group = pg_cursor.fetchone()
connection.close()
return resource_group
except Exception:
traceback.print_exc(file=sys.stderr)
示例10: verify_event_trigger
def verify_event_trigger(server, db_name, trigger_name):
"""
This function verifies the event trigger is present in the database
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param trigger_name: trigger name to be verified
:type trigger_name: str
:return event_trigger: event trigger's details
:rtype event_trigger: tuple
"""
try:
connection = get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute(
"SELECT oid FROM pg_event_trigger WHERE evtname = '%s'"
% trigger_name)
event_trigger = pg_cursor.fetchone()
connection.close()
return event_trigger
except Exception:
traceback.print_exc(file=sys.stderr)
示例11: verify_foreignkey
def verify_foreignkey(server, db_name, local_table_name):
"""
This function verifies foreign key constraint exist or not.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param local_table_name: local table name
:type local_table_name: str
:return table: table record from database
:rtype: tuple
"""
try:
connection = utils.get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute(
"SELECT oid FROM pg_constraint where conname='%s_id_fkey'" %
local_table_name)
fk_record = pg_cursor.fetchone()
connection.close()
return fk_record
except Exception:
traceback.print_exc(file=sys.stderr)
示例12: verify_fdw
def verify_fdw(server, db_name, fdw_name):
"""
This function will verify current foreign data wrapper.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param fdw_name: FDW name
:type fdw_name: str
:return fdw: fdw details
:rtype: tuple
"""
try:
connection = get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute(
"SELECT oid FROM pg_foreign_data_wrapper WHERE fdwname = '%s'"
% fdw_name)
fdw = pg_cursor.fetchone()
connection.close()
return fdw
except Exception:
traceback.print_exc(file=sys.stderr)
示例13: verify_table
def verify_table(server, db_name, table_id):
"""
This function verifies table exist in database or not.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param table_id: schema name
:type table_id: int
:return table: table record from database
:rtype: tuple
"""
try:
connection = utils.get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute("SELECT * FROM pg_class tb WHERE tb.oid=%s" %
table_id)
table = pg_cursor.fetchone()
connection.close()
return table
except Exception:
traceback.print_exc(file=sys.stderr)
raise
示例14: verify_rule
def verify_rule(server, db_name, rule_name):
"""
This function verifies rule exist in database or not.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param rule_name: rule name
:type rule_name: str
:return rule: rule record from database
:rtype: tuple
"""
try:
connection = utils.get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute("select * from pg_rewrite where rulename='%s'" %
rule_name)
rule = pg_cursor.fetchone()
connection.close()
return rule
except Exception:
traceback.print_exc(file=sys.stderr)
raise
示例15: verify_fts_parser
def verify_fts_parser(server, db_name, fts_parser_name):
"""
This function will verify current FTS parser.
:param server: server details
:type server: dict
:param db_name: database name
:type db_name: str
:param fts_parser_name: FTS parser name to be added
:type fts_parser_name: str
:return fts_temp: FTS parser detail
:rtype: tuple
"""
try:
connection = get_db_connection(db_name,
server['username'],
server['db_password'],
server['host'],
server['port'],
server['sslmode'])
pg_cursor = connection.cursor()
pg_cursor.execute(
"select oid from pg_catalog.pg_ts_parser where "
"prsname = '%s' order by oid ASC limit 1"
% fts_parser_name)
fts_parser = pg_cursor.fetchone()
connection.close()
return fts_parser
except Exception:
traceback.print_exc(file=sys.stderr)