# 需要导入模块: import cx_Oracle [as 别名]

# 或者: from cx_Oracle import SYSDBA [as 别名]

def db_connect(args):

if args.type == "mysql" or args.type == "mariadb":

import mysql.connector

try:

connection = mysql.connector.connect(

user=args.user,

password=args.password,

database=args.db)

except mysql.connector.Error as err:

print(colorize("red", "[ERROR] {}".format(err)))

return None

elif args.type == "mssql":

import pymssql

try:

connection = pymssql.connect(server="localhost", database=args.db)

except pymssql.Error as err:

print(colorize("red", "[ERROR] {}".format(err)))

return None

elif args.type == "pgsql":

import psycopg2

try:

connection = psycopg2.connect(

"dbname='{}' user='{}' password='{}'".format(

args.db, args.user, args.password))

except psycopg2.Error as err:

print(colorize("red", "[ERROR] {}".format(err)))

return None

elif args.type == "oracle":

import cx_Oracle

try:

connection = cx_Oracle.connect(

args.user, args.password, cx_Oracle.makedsn(

'127.0.0.1', 1521, args.db), mode=cx_Oracle.SYSDBA)

except cx_Oracle.Error as err:

print(colorize("red", "[ERROR] {}".format(err)))

return None

return connection