Python| PostgreSQL 中的数据库管理

PostgreSQL是一个开源的对象关系数据库管理系统。它以其可靠性、稳健性和性能而闻名。 PostgreSQL 有各种 API（应用程序可编程接口）库，可用于各种流行的编程语言，例如Python。它为数据库管理提供了许多功能，例如视图、触发器、索引（使用 B-Trees）等。

有几个Python模块允许我们使用 PostgreSQL 连接和操作数据库：

• 心理战2
• PG8000
• py-postgresql
• PyGreSQL

Psycopg2是 PostgreSQL 最流行的Python驱动程序之一。它被积极维护并为不同版本的Python提供支持。它还提供对线程的支持，并可用于多线程应用程序。由于这些原因，它是开发人员的热门选择。

在本文中，我们将通过在Python中构建一个简单的数据库管理系统来探索使用 psycopg2 的 PostgreSQl 的特性。

安装：

sudo pip3 install psycopg2 

注意：如果您使用的是 Python2，请使用 pip install 而不是 pip3

在您的系统中安装psycopg后，我们可以连接到数据库并在Python中执行查询。

创建数据库

在我们可以在Python中访问数据库之前，我们需要在 postgresql 中创建数据库。要创建数据库，请按照以下步骤操作：

1. 登录到 PostgreSQL：

   sudo -u postgres psql

   

2. 配置密码：

   \password

   然后将提示您输入密码。请记住这一点，因为我们将使用它连接到Python中的数据库。
   

3. 创建一个名为“test”的数据库。我们将连接到这个数据库。

   CREATE DATABASE test; 

   配置好数据库和密码后，退出 psql 服务器。

   

   连接到数据库

   connect()方法用于建立与数据库的连接。它需要5个参数：

   1. 数据库：您要连接的数据库的名称
   2. user：本地系统的用户名
   3. password : 登录 psql 的密码
   4. host ：主机，默认设置为 localhost
   5. port ：端口号，默认为 5432

   conn = psycopg2.connect(
               database="test", 
               user = "adith", 
               password = "password", 
               host = "localhost", 
               port = "5432")

   建立连接后，我们可以在Python中操作数据库。

   Cursor对象用于执行 sql 查询。我们可以使用连接对象（conn）创建一个游标对象

   cur = conn.cursor()  

   使用这个对象，我们可以对我们连接的数据库进行更改。

   执行完所有查询后，我们需要断开连接。不断开连接不会导致任何错误，但通常认为断开连接是一种好习惯。

   conn.close() 

   执行查询

   execute()方法接受一个参数，即要执行的 SQL 查询。 SQL 查询采用包含 SQL 语句的字符串的形式。

   cur.execute("SELECT * FROM emp") 

   获取数据

   执行查询后，可以使用fetchall()方法获取查询结果。此方法不接受任何参数并返回选择查询的结果。

   res = cur.fetchall() 

   查询的结果存储在 res 变量中。

   把它们放在一起

   一旦我们在 PostgreSQL 中创建了数据库，我们就可以在Python中访问该数据库。我们首先在数据库中创建一个名为 test 的 emp 表，其模式为：(id INTEGER PRIMARY KEY, name VARCHAR(10), Salary INT, dept INT)。一旦创建表没有任何错误，我们将值插入表中。
   插入值后，我们可以查询表以选择所有行并使用fetchall()函数将它们显示给用户。

   # importing libraries
   import psycopg2
     
   # a function to connect to
   # the database.
   def connect():
     
       # connecting to the database called test
       # using the connect function
       try:
     
           conn = psycopg2.connect(database ="test", 
                               user = "adith", 
                               password = "password", 
                               host = "localhost", 
                               port = "5432")
     
           # creating the cursor object
           cur = conn.cursor()
         
       except (Exception, psycopg2.DatabaseError) as error:
             
           print ("Error while creating PostgreSQL table", error)
         
     
       # returing the conn and cur
       # objects to be used later
       return conn, cur
     
     
   # a function to create the 
   # emp table.
   def create_table():
     
       # connect to the database.
       conn, cur = connect()
     
       try:
           # the test database contains a table called emp  
           # the schema : (id INTEGER PRIMARY KEY,  
           # name VARCHAR(10), salary INT, dept INT) 
           # create the emp table 
     
           cur.execute('CREATE TABLE emp (id INT PRIMARY KEY, name VARCHAR(10),
                                          salary INT, dept INT)')
     
           # the commit function permanently
           # saves the changes made to the database
           # the rollback() function can be used if
           # there are any undesirable changes and
           # it simply undoes the changes of the
           # previous query
         
       except:
     
           print('error')
     
       conn.commit() 
     
      
   # a function to insert data
   # into the emp table
   def insert_data(id = 1, name = '', salary = 1000, dept = 1):
     
       conn, cur = connect()
     
       try:
           # inserting values into the emp table
           cur.execute('INSERT INTO emp VALUES(%s, %s, %s, %s)',
                                       (id, name, salary, dept))
         
       except Exception as e:
     
           print('error', e)
       # commiting the transaction.
       conn.commit()
     
     
   # a function to fetch the data 
   # from the table
   def fetch_data():
     
       conn, cur = connect()
     
       # select all the rows from emp
       try:
           cur.execute('SELECT * FROM emp')
         
       except:
           print('error !')
     
       # store the result in data
       data = cur.fetchall()
     
       # return the result
       return data
     
   # a function to print the data
   def print_data(data):
     
       print('Query result: ')
       print()
     
       # iterating over all the 
       # rows in the table
       for row in data:
     
           # printing the columns
           print('id: ', row[0])
           print('name: ', row[1])
           print('salary: ', row[2])
           print('dept: ', row[3])
           print('----------------------------------')
     
   # function to delete the table
   def delete_table():
     
       conn, cur = connect()
     
       # delete the table
       try:
     
           cur.execute('DROP TABLE emp')
     
       except Exception as e:
           print('error', e)
     
       conn.commit()
     
     
   # driver function
   if __name__ == '__main__':
     
       # create the table
     
       create_table()
     
       # inserting some values
       insert_data(1, 'adith', 1000, 2)
       insert_data(2, 'tyrion', 100000, 2)
       insert_data(3, 'jon', 100, 3)
       insert_data(4, 'daenerys', 10000, 4)
     
       # getting all the rows
       data = fetch_data()
     
       # printing the rows
       print_data(data)
     
       # deleting the table
       # once we are done with
       # the program
       delete_table()
   

   输出 ：