chatgpt赋能python：Python连接PostgreSQL实践指南

import psycopg2
 
# 连接PostgreSQL数据库
def connect_to_postgresql(dbname, user, password, host, port):
    try:
        conn = psycopg2.connect(
            dbname=dbname,
            user=user,
            password=password,
            host=host,
            port=port
        )
        print("连接数据库成功")
        return conn
    except Exception as e:
        print(f"连接数据库失败: {e}")
        return None
 
# 使用连接对象创建游标
def create_cursor(conn):
    if conn is not None:
        cursor = conn.cursor()
        print("创建游标成功")
        return cursor
    else:
        print("无法创建游标，因为没有数据库连接")
        return None
 
# 使用游标执行SQL语句
def execute_sql(cursor, sql):
    if cursor is not None:
        cursor.execute(sql)
        print("SQL语句执行成功")
    else:
        print("无法执行SQL语句，因为没有创建游标")
 
# 关闭游标和连接
def close_connection(conn, cursor):
    if cursor is not None:
        cursor.close()
        print("关闭游标成功")
    if conn is not None:
        conn.close()
        print("关闭数据库连接成功")
 
# 示例使用
if __name__ == "__main__":
    dbname = "your_dbname"
    user = "your_user"
    password = "your_password"
    host = "your_host"
    port = "your_port"
 
    conn = connect_to_postgresql(dbname, user, password, host, port)
    cursor = create_cursor(conn)
 
    sql = "SELECT version();"
    execute_sql(cursor, sql)
 
    # 关闭游标和连接
    close_connection(conn, cursor)

这段代码展示了如何使用Python连接PostgreSQL数据库，创建游标，执行SQL语句，以及关闭连接。这是一个简洁的实现，适合作为初学者的教学示例。在实际应用中，你需要替换your_dbname, your_user, your_password, your_host, your_port为你的数据库信息。