Python 数据库操作 psycopg2

安装

psycopg 是 Python 语言中 PostpreSQL数据库接口

安装环境：

1. Python：v2.7, v3.4~3.8
2. PostGreSQL：7.4~12

pip install psycopg2

基本使用

import psycopg2

def connect_db(host: str,
               port: int,
               database: str,
               user: str,
               password: str,) -> tuple:
	conn = psycopg2.connect(host=host,
	                        port=port,
	                        database=database,
	                        user=user,
	                        password=password)
	return conn

def execute_sql(conn: object, sql: str, params: tuple = ()) -> list:
    cur = conn.cursor()
    cur.execute(sql, params)
    try:
        return cur.fetchall()
    except psycopg2.ProgrammingError:
        conn.commit()
        return