python postgresql选择用户输入

import psycopg2

# 连接到PostgreSQL数据库
conn = psycopg2.connect(database="your_database", user="your_user", password="your_password", host="your_host", port="your_port")

# 创建游标对象
cur = conn.cursor()

# 用户选择输入
user_input = input("请选择操作：1.查询数据 2.插入数据 3.更新数据 4.删除数据：")

# 根据用户选择执行不同的操作
if user_input == "1":
    # 查询数据
    cur.execute("SELECT * FROM your_table")
    rows = cur.fetchall()
    for row in rows:
        print(row)
elif user_input == "2":
    # 插入数据
    data = input("请输入要插入的数据：")
    cur.execute("INSERT INTO your_table (column_name) VALUES (%s)", (data,))
    conn.commit()
    print("数据插入成功！")
elif user_input == "3":
    # 更新数据
    data = input("请输入要更新的数据：")
    cur.execute("UPDATE your_table SET column_name = %s WHERE condition", (data,))
    conn.commit()
    print("数据更新成功！")
elif user_input == "4":
    # 删除数据
    condition = input("请输入要删除的条件：")
    cur.execute("DELETE FROM your_table WHERE condition", (condition,))
    conn.commit()
    print("数据删除成功！")
else:
    print("无效的选择！")

# 关闭游标和数据库连接
cur.close()
conn.close()