📜  Python | PostgreSQL中的数据库管理

📅  最后修改于: 2020-04-15 07:43:45             🧑  作者: Mango

PostgreSQL是一个开源的对象关系数据库管理系统。以其可靠性,鲁棒性和性能而闻名。PostgreSQL有各种API(应用程序可编程接口)库,可用于各种流行的编程语言,例如Python。它为数据库管理提供了许多功能,例如视图,触发器,索引(使用B树)等。
有几个Python模块可以让我们使用PostgreSQL连接和操作数据库:

  • Psycopg2
  • 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;
  4. 配置数据库和密码后,退出psql服务器。
    conn = psycopg2.connect(
                database="test",
                user = "adith",
                password = "password",
                host = "localhost",
                port = "5432")

    建立连接后,我们可以使用Python操作数据库。
    光标对象用于执行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,名称VARCHAR(10),salary INT,dept INT)。创建表后,没有任何错误,我们将值插入表中。
    插入值后,我们可以查询表以选择所有行,并使用fetchall()函数将其显示给用户

    # 导入库
    import psycopg2
    # 连接数据库的功能.
    def connect():
        # 使用connect函数连接到名为test的数据库
        try:
            conn = psycopg2.connect(database ="test",
                                user = "adith",
                                password = "password",
                                host = "localhost",
                                port = "5432")
            # 创建光标对象
            cur = conn.cursor()
        except (Exception, psycopg2.DatabaseError) as error:
            print ("Error while creating PostgreSQL table", error)
        # 重载conn和cur对象以供以后使用
        return conn, cur
    # 创建emp表的函数.
    def create_table():
        # 连接到数据库.
        conn, cur = connect()
        try:
            # 测试数据库包含一个名为emp的表 : (id INTEGER PRIMARY KEY,
            # name VARCHAR(10), salary INT, dept INT)
            # 创建emp表
            cur.execute('CREATE TABLE emp (id INT PRIMARY KEY, name VARCHAR(10),
                                           salary INT, dept INT)')
            # commit函数将永久保存对数据库所做的更改,如果有任何不希望的更改,可以使用rollback()函数,并且只需撤销先前查询的更改即
        except:
            print('error')
        conn.commit()
    # 将数据插入emp表的函数
    def insert_data(id = 1, name = '', salary = 1000, dept = 1):
        conn, cur = connect()
        try:
            # 将值插入emp表
            cur.execute('INSERT INTO emp VALUES(%s, %s, %s, %s)',
                                        (id, name, salary, dept))
        except Exception as e:
            print('error', e)
        # 进行交易.
        conn.commit()
    # 从表中获取数据的功能
    def fetch_data():
        conn, cur = connect()
        # 从emp中选择所有行
        try:
            cur.execute('SELECT * FROM emp')
        except:
            print('error !')
        # 将结果存储在数据中
        data = cur.fetchall()
        # 返回结果
        return data
    # 打印数据的功能
    def print_data(data):
        print('Query result: ')
        print()
        # 遍历表中的所有行
        for row in data:
            # 打印列
            print('id: ', row[0])
            print('name: ', row[1])
            print('salary: ', row[2])
            print('dept: ', row[3])
            print('----------------------------------')
    # 删除表格的功能
    def delete_table():
        conn, cur = connect()
        # 删除表格
        try:
            cur.execute('删除表emp')
        except Exception as e:
            print('error', e)
        conn.commit()
    # 测试代码
    if __name__ == '__main__':
        # 创建表
        create_table()
        # 插入一些值
        insert_data(1, 'adith', 1000, 2)
        insert_data(2, 'tyrion', 100000, 2)
        insert_data(3, 'jon', 100, 3)
        insert_data(4, 'daenerys', 10000, 4)
        # 获取所有行
        data = fetch_data()
        # 打印行
        print_data(data)
        # 完成程序后,删除表
        delete_table()

    输出: