### 使用 Python 和 psql 对 PostgreSQL 数据库进行基本操作 #### 连接 PostgreSQL 数据库 对于使用 Python 来连接 PostgreSQL 数据库，`psycopg2` 是一个常用的库。安装 `psycopg2` 后可以通过它建立到数据库的连接[^3]。 ```python import psycopg2 connection = psycopg2.connect( host="localhost", database="your_database_name", user="your_username", password="your_password" ) cursor = connection.cursor() ``` 一旦建立了连接，就可以利用这个连接来执行 SQL 命令以及获取返回的结果集。 #### 执行查询 要从数据库中检索数据，可以构建 SELECT 查询语句并通过游标的 execute 方法发送给服务器： ```python query = "SELECT * FROM your_table;" cursor.execute(query) rows = cursor.fetchall() for row in rows: print(row) ``` 这段代码会打印出表中的每一行记录。如果只需要部分列的数据，则可以在 SELECT 之后指定所需的字段名称而不是星号(*)。 #### 插入、更新和删除数据 除了读取之外，还可以修改存储在数据库里的信息。下面的例子展示了如何向表格插入新纪录，并且展示了一个简单的 UPDATE 和 DELETE 操作： ```python insert_query = """ INSERT INTO your_table (column1, column2) VALUES (%s, %s); """ data_to_insert = ('value1', 'value2') cursor.execute(insert_query, data_to_insert) update_query = """ UPDATE your_table SET column1=%s WHERE some_column=%s; """ new_value = ("updated_value", "condition") cursor.execute(update_query, new_value) delete_query = """ DELETE FROM your_table WHERE condition_column=%s; """ deletion_condition = ("specific_value", ) cursor.execute(delete_query, deletion_condition) ``` 每次更改都需要调用 commit() 函数以保存所做的变更；如果不希望保留这些变化，在关闭程序之前可以选择回滚(rollback())它们。 #### 提交事务和关闭连接 当所有的数据库操作完成后，应该提交任何未决的变化并断开与数据库之间的链接: ```python connection.commit() cursor.close() connection.close() ``` 这一步骤非常重要，因为它确保了所有已做的工作都被永久化到了磁盘上，并释放了占用资源。 #### 错误处理 考虑到可能出现的各种错误情况（比如网络中断），应当适当地捕获异常并采取相应的措施： ```python try: # Database operations here... except Exception as e: print(f"An error occurred: {e}") finally: if connection is not None and not connection.closed: connection.rollback() cursor.close() connection.close() ``` 通过这种方式能够更加稳健地应对潜在的问题，同时也保持良好的编程习惯。