Backend - python 连接 & 操作 PostgreSQL DB(数据库)

目录

一、基础环境

(一)建立虚拟环境

(二)安装依赖

二、连接数据库

三、操作数据库

(一)查询

(二)单笔更新/插入

(三)批量更新/插入

(四)遇到的问题

[1. SQL语句 - 栏位名报错](#1. SQL语句 - 栏位名报错)

(1)具体问题

(2)解决

[2. SQL语句 - 表不存在](#2. SQL语句 - 表不存在)

(1)具体问题

(2)解决

[3. SQL语句 - 插入的数据格式不正确](#3. SQL语句 - 插入的数据格式不正确)

(1)具体问题

(2)解决


一、基础环境

(一)建立虚拟环境

Python 3.10.7

(二)安装依赖

psycopg2==2.9.9 (在虚拟环境中执行安装)

二、连接数据库

python 复制代码
import psycopg2
from psycopg2.extras import execute_values
class ExcuteDB:
	# --------------------------------------------------------------------------------------------------------------------------------------- 连接数据库
    def __init__(self, host, dbname, user, port, password):
        self.host = host
        self.dbname = dbname
        self.user = user
        self.port = port
        self.password = password
          # self.conn = self.__postgres  # 连接数据库(放在各种功能方法中,比如exc_query方法)
  
    @property
    def __postgres(self):
        conn = psycopg2.connect(host=self.host,  # 主机地址
                                dbname=self.dbname,  # 数据库名称
                                user=self.user,  # 用户名
                                password=self.password,  # 密码
                                port=self.port)  # 端口号
        return conn
	# --------------------------------------------------------------------------------------------------------------------------------------- 查询
    def exc_query(self, sql):
        try:
            conn = self.__postgres  # 连接数据库
            cur = conn.cursor()  # 创建游标对象
            cur.execute(sql)  # 执行SQL语句
            res_list = None
            res_list = cur.fetchall()  # 获取记录值,结果是列表包元组:[('1F002', '数据结构怎么学'), ('1F001', '数据库原理')]
            cols_list = [row[0] for row in cur.description]  # 获取栏位名称['card', 'name']
            data_list = [dict(map(lambda key, value: [key, value], cols_list, val_list)) for val_list in res_list]  # 组装列表包字典:data_list [{'card': '1F002', 'name': '数据结构怎么学'}, {'card': '1F001', 'name': '数据库原理'}]
        except Exception as e:
            print(str(e))
        finally:
            cur.close()  # 关闭游标
            conn.close()  # 关闭数据库连接
		       return {'栏位名+栏位值': data_list, '只有栏位值': res_list}
	# --------------------------------------------------------------------------------------------------------------------------------------- 单笔更新/插入
	def exc_edit(self, sql):
		        try:
		            conn = self.__postgres  # 连接数据库
		            cur = conn.cursor()  # 创建游标对象
		            cur.execute(sql)  # 执行SQL语句
		            conn.commit()  # 提交事务
		        except Exception as e:
		            conn.rollback()  # 回滚commit事务
		            print(str(e))
		        finally:
		            cur.close()  # 关闭游标
		            conn.close()  # 关闭数据库连接
	# --------------------------------------------------------------------------------------------------------------------------------------- 批量更新/插入
    def exc_bulk_edit(self, sql_list):
        try:
            conn = self.__postgres  # 连接数据库
            cur = conn.cursor()  # 创建游标对象
            for l in sql_list:  # 循环不同的sql
                sql = l['sql']  # sql语句:'insert into auth_group (card, name, author, btype, price, num) values %s '
                args_list = l['args_list']  # 批量插入的值:[('2F0001', '高等数学', '小星星', '教育', 35, 10), ('2F0002', '线性代数', '小星星', '教育', 40, 12)]
                execute_values(cur, sql, args_list)  # 批量执行
            conn.commit()  # 提交事务
        except Exception as e:
            conn.rollback()  # 回滚commit事务
            print(str(e))
        finally:
            cur.close()  # 关闭游标
            conn.close()  # 关闭数据库连接

三、操作数据库

(一)查询

python 复制代码
exc_db = ExcuteDB('XX.XX.XX.XX', 'bookDB', 'admin', '5432', '123')   # 连接数据库
res = exc_db.exc_query('select card, name from "myApp_book" limit 10')  # 执行查询

(二)单笔更新/插入

python 复制代码
exc_db = ExcuteDB('XX.XX.XX.XX', 'bookDB', 'admin', '5432', '123')   # 连接数据库
res2 = exc_db.exc_edit('update "myApp_book" set name=\'数据结构\' where card=\'1F002\' ')  # 执行更新

(三)批量更新/插入

python 复制代码
exc_db = ExcuteDB('XX.XX.XX.XX', 'bookDB', 'admin', '5432', '123')   # 连接数据库
res2 = exc_db.exc_bulk_edit([{'sql': 'insert into auth_group (name) values %s ', 
                             'args_list': [('2F0001',), ('2F0002',)]},  # 批量插入auth_group表
                            {'sql': 'insert into "myApp_book" (card, name, author, btype, price, num) values %s ', 
                             'args_list': [('2F0001', '高等数学', '小星星', '教育', 35, 10), 
                                           ('2F0002', '线性代数', '小星星', '教育', 40, 12)]}  # 批量插入myApp_book表
                            ])

(四)遇到的问题

1. SQL语句 - 栏位名报错

(1)具体问题
bash 复制代码
column "数据结构" does not exist
LINE 1: update "myApp_book" set name="数据结构" where card='1F002'
(2)解决

数据值不能用双引号括起来,而是用 \'。

错误写法:

python 复制代码
exc_db.exc_query('update "myApp_book" set name="数据结构" where card=\'1F002\' ') 

正确写法:

python 复制代码
exc_db.exc_query('update "myApp_book" set name=\'数据结构\' where card=\'1F002\' ') 

2. SQL语句 - 表不存在

(1)具体问题
bash 复制代码
relation "myapp_book" does not exist
(2)解决

表名要用双引号包裹。

正确写法:

python 复制代码
insert into "myApp_book"  ...

错误写法:

python 复制代码
insert into myApp_book ...

3. SQL语句 - 插入的数据格式不正确

(1)具体问题
bash 复制代码
INSERT has more expressions than target columns
LINE 1: insert into auth_group (name) values ('2','F','0','0','0','1...
(2)解决

元组里只有一个数据时,得用逗号结尾。

正确写法:

python 复制代码
[('2F0001',), ('2F0002',)]  # 列表包元组

错误写法:

python 复制代码
[('2F0001'), ('2F0002')]  # 会自动转变为:['2F0001','2F0002']
相关推荐
凛铄linshuo20 分钟前
爬虫简单实操2——以贴吧为例爬取“某吧”前10页的网页代码
爬虫·python·学习
牛客企业服务23 分钟前
2025年AI面试推荐榜单,数字化招聘转型优选
人工智能·python·算法·面试·职场和发展·金融·求职招聘
胡斌附体35 分钟前
linux测试端口是否可被外部访问
linux·运维·服务器·python·测试·端口测试·临时服务器
IT项目管理1 小时前
达梦数据库DMHS介绍及安装部署
linux·数据库
你都会上树?1 小时前
MySQL MVCC 详解
数据库·mysql
大春儿的试验田1 小时前
高并发收藏功能设计:Redis异步同步与定时补偿机制详解
java·数据库·redis·学习·缓存
likeGhee1 小时前
python缓存装饰器实现方案
开发语言·python·缓存
项目題供诗1 小时前
黑马python(二十五)
开发语言·python
读书点滴2 小时前
笨方法学python -练习14
java·前端·python
Ein hübscher Kerl.2 小时前
虚拟机上安装 MariaDB 及依赖包
数据库·mariadb