Python

Psycopg是一种用于执行SQL语句的PythonAPI，可以为PostgreSQL、openGauss数据库提供统一访问接口，应用程序可基于它进行数据操作。Psycopg2对libpq进行封装，部分代码使用C语言实现，既高效又安全。它具有客户端游标和服务器端游标、异步通信和通知、支持“COPY TO/COPY FROM”功能。支持多种类型Python开箱即用，适配PostgreSQL数据类型；通过灵活的对象适配系统，可以扩展和定制适配。Psycopg2兼容Unicode和Python 3。

openGauss数据库提供了对Psycopg2特性的支持，并且支持psycopg2通过SSL模式链接。

表 1 Psycopg支持平台

操作系统

平台

EulerOS 2.5

x86_64位

EulerOS 2.8

ARM64位

加载驱动

  • 在使用驱动之前，需要做如下操作：

    1. 从openGauss官网上下载编译好的psycopg2压缩包。

      Python - 图1 说明： openGauss官网上下载的psycopg2适配的版本为Python3.6，如果使用其他版本的Python, 需要自行编译psycopg2，编译方法与在PostgreSQL数据库下基本相同。只是需要修改一下编译时的setup.py版本号校验部分的代码，可通过如下命令实现：

      1. sed -i "s/(pgmajor, pgminor, pgpatch)/(9, 2, 4)/g" setup.py

    2. 解压版本对应的驱动包，将psycopg2拷贝到python安装目录的第三方包文件夹（即site-packages目录）下。

    3. 确保psycopg2目录权限至少为755，以免调用时提示文件由于权限问题无法访问。

    4. 对于非数据库用户，需要将解压后的lib目录，配置在LD_LIBRARY_PATH环境变量中。

  • 在创建数据库连接之前，需要先加载如下数据库驱动程序：

    1. import  psycopg2

连接数据库

以下Python代码显示了如何连接到现有的数据库。 如果数据库不存在，那么它将自动创建，最后将返回一个数据库对象。

  1. #!/usr/bin/python
  2. import psycopg2
  3. conn = psycopg2.connect(database="testdb", user="openGauss", password="xxxxxxxx", host="127.0.0.1", port="26000")

上述代码中，加粗字体请根据具体情况替换。在这里指定使用testdb作为数据库名称。

创建表

以下Python程序将用于在先前创建的数据库（testdb）中创建一个表：

  1. #!/usr/bin/python
  3. import psycopg2
  5. conn = psycopg2.connect(database="testdb", user="openGauss", password="xxxxxxxx", host="127.0.0.1", port="26000")
  7. cur = conn.cursor()
  8. cur.execute('''CREATE TABLE COMPANY
  9.        (ID INT PRIMARY KEY     NOT NULL,
  10.        NAME           TEXT    NOT NULL,
  11.        AGE            INT     NOT NULL,
  12.        ADDRESS        CHAR(50),
  13.        SALARY         REAL);''')
  15. conn.commit()
  16. conn.close()

当执行上述程序时，它将在数据库testdb中创建COMPANY表。

插入操作

以下Python程序显示了如何在上述示例中创建的COMPANY表中创建记录：

  1. #!/usr/bin/python
  3. import psycopg2
  5. conn = psycopg2.connect(database="testdb", user="openGauss", password="xxxxxxxx", host="127.0.0.1", port="26000")
  7. cur = conn.cursor()
  9. cur.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
  10.       VALUES (1, 'Paul', 32, 'California', 20000.00 )");
  12. cur.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
  13.       VALUES (2, 'Allen', 25, 'Texas', 15000.00 )");
  15. cur.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
  16.       VALUES (3, 'Teddy', 23, 'Norway', 20000.00 )");
  18. cur.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
  19.       VALUES (4, 'Mark', 25, 'Rich-Mond ', 65000.00 )");
  21. conn.commit()
  22. conn.close()

当执行上述程序时，它将在COMPANY表中创建/插入给定的记录。

SELECT操作

以下Python程序显示了如何从上述示例中创建的COMPANY表中获取和显示记录：

  1. #!/usr/bin/python
  3. import psycopg2
  5. conn = psycopg2.connect(database="testdb", user="openGauss", password="xxxxxxxx", host="127.0.0.1", port="26000")
  7. cur = conn.cursor()
  9. cur.execute("SELECT id, name, address, salary  from COMPANY")
  10. rows = cur.fetchall()
  11. for row in rows:
  12.    print("ID = ", row[0])
  13.    print("NAME = ", row[1])
  14.    print("ADDRESS = ", row[2])
  15.    print("SALARY = ", row[3])
  17. conn.close()

执行上述程序时，会返回以下结果：

  1. ID =,  1
  2. NAME =,  Paul
  3. ADDRESS =,  California
  4. SALARY =,  20000.0
  6. ID =,  2
  7. NAME =,  Allen
  8. ADDRESS =,  Texas
  9. SALARY =,  15000.0
  11. ID =,  3
  12. NAME =,  Teddy
  13. ADDRESS =,  Norway
  14. SALARY =,  20000.0
  16. ID =,  4
  17. NAME =,  Mark
  18. ADDRESS =,  Rich-Mond
  19. SALARY =,  65000.0

更新操作

以下Python代码显示了如何使用UPDATE语句来更新任何记录，然后从COMPANY表中获取并显示更新的记录：

  1. #!/usr/bin/python
  3. import psycopg2
  5. conn = psycopg2.connect(database="testdb", user="openGauss", password="xxxxxxxx", host="127.0.0.1", port="26000")
  7. cur = conn.cursor()
  9. cur.execute("UPDATE COMPANY set SALARY = 25000.00 where ID=1")
  10. conn.commit
  12. cur.execute("SELECT id, name, address, salary  from COMPANY")
  13. rows = cur.fetchall()
  14. for row in rows:
  15.    print("ID = ", row[0])
  16.    print("NAME = ", row[1])
  17.    print("ADDRESS = ", row[2])
  18.    print("SALARY = ", row[3])
  20. conn.close()

执行上述程序时，会产生以下结果：

  1. ID =,  1
  2. NAME =,  Paul
  3. ADDRESS =,  California
  4. SALARY =,  25000.0
  6. ID =,  2
  7. NAME =,  Allen
  8. ADDRESS =,  Texas
  9. SALARY =,  15000.0
  11. ID =,  3
  12. NAME =,  Teddy
  13. ADDRESS =,  Norway
  14. SALARY =,  20000.0
  16. ID =,  4
  17. NAME =,  Mark
  18. ADDRESS =,  Rich-Mond
  19. SALARY =,  65000.0

删除操作

以下Python代码显示了如何使用DELETE语句来删除记录，然后从COMPANY表中获取并显示剩余的记录：

  1. #!/usr/bin/python
  3. import psycopg2
  5. conn = psycopg2.connect(database="testdb", user="openGauss", password="xxxxxxxx", host="127.0.0.1", port="26000")
  7. cur = conn.cursor()
  9. cur.execute("DELETE from COMPANY where ID=2;")
  10. conn.commit
  12. cur.execute("SELECT id, name, address, salary  from COMPANY")
  13. rows = cur.fetchall()
  14. for row in rows:
  15.    print("ID = ", row[0])
  16.    print("NAME = ", row[1])
  17.    print("ADDRESS = ", row[2])
  18.    print("SALARY = ", row[3])
  20. conn.close()

执行上述程序时，会产生以下结果：

  1. ID =,  1
  2. NAME =,  Paul
  3. ADDRESS =,  California
  4. SALARY =,  20000.0
  6. ID =,  3
  7. NAME =,  Teddy
  8. ADDRESS =,  Norway
  9. SALARY =,  20000.0
  11. ID =,  4
  12. NAME =,  Mark
  13. ADDRESS =,  Rich-Mond
  14. SALARY =,  65000.0