首先,你需要安装对应的数据库驱动。对于 MySQL 和 MariaDB,你可以使用 PyMySQL 驱动;对于 PostgreSQL,你可以使用 psycopg2 驱动;对于 Oracle,你可以使用 cx_Oracle 驱动;对于 Kingbase,你可以使用 pyodbc 驱动;对于 Hive,你可以使用 PyHive 驱动。
具体的安装方法可以参考各自的文档,这里不再赘述。
然后,你可以使用以下代码来获取数据库连接:
import pymysql
import psycopg2
import cx_Oracle
import pyodbc
import PyHive
def get_mysql_connection(host, port, user, password, database): connection = pymysql.connect(
host=host,
port=port,
user=user,
password=password,
database=database
)
return connection
def get_pgsql_connection(host, port, user, password, database):
connection = psycopg2.connect(
host=host,
port=port,
user=user,
password=password,
database=database
)
return connection
def get_oracle_connection(host, port, user, password, database):
dsn = cx_Oracle.makedsn(host, port, database)
connection = cx_Oracle.connect(
user=user,
password=password,
dsn=dsn
)
return connection
def get_kingbase_connection(host, port, user, password, database):
connection_string = 'DRIVER={Kingbase ODBC Driver};SERVER=%s;PORT=%s;DATABASE=%s;UID=%s;PWD=%s' % (host, port, database, user, password)
connection = pyodbc.connect(connection_string)
return connection
def get_mariadb_connection(host, port, user, password, database):
connection = pymysql.connect(
host=host,
port=port,
user=user,
password=password,
database=database
)
return connection
def get_hive_connection(host, port, user, password, database):
connection = PyHive.connect(
host=host,
port=port,
user=user,
password=password,
database=database
)
return connection
这里的代码假