python操作PostgreSQL數據庫

import psycopg2 as pg2

# 返回數據庫PostgreSQL連接
def get_db_conn():
    # 創建連接
    return pg2.connect(database='test', user='admin', password='admin', host='127.0.0.1', port='10001')


# 操作數據庫PostgreSQL，返回一條結果
def db_fetchone(sql):
    try:
        conn = get_db_conn()
        cur = conn.cursor()
        cur.execute(sql)
        rows = cur.fetchone()  # 返回一條結果，返回多條結果使用rows = cur.fetchall()

        return rows[0]
    except pg2.DatabaseError as e:
        print('Error $s' % e)

    finally:
        conn.commit()
        cur.close()
        conn.close()