python使用psycopg2管理postgres
基于docker的postgres 部署见这篇 https://www.cnblogs.com/xuanmanstein/p/7742647.html
连接数据库
import psycopg2class MyTestCase(unittest.TestCase): @classmethod def setUpClass(cls): db = '172.19.0.12' # root用户 先连到默认数据库 connection_parameters = { 'host': db, 'database': 'postgres', 'user': 'postgres', 'password': 'example' } cls.conn = psycopg2.connect(**connection_parameters) cls.conn.autocommit = True
创建ROLE或USER
必须用连接其他数据库的其他用户,才能删除和创建别的用户,典型如默认用户postgres 连接到postgres,然后删除创建别的用户,和别的数据库
def test_create_user(self): #创建用户 name = 'stavka' pwd = '1111' SQL = f''' DROP ROLE IF EXISTS {name}; CREATE ROLE {name}; ALTER ROLE {name} ENCRYPTED PASSWORD '{pwd}'; ALTER ROLE {name} LOGIN CREATEDB CREATEROLE; ''' # 执行 with self.conn.cursor() as cursor: cursor.execute(SQL)
先删除(如果存在),再创建,再设置密码,最后设置各种权限
创建/删除数据库
为了使用postgis,在docker镜像建立的时候,已经1个template_postgis数据库,安装了各种extention,
然后就可以用它作为模板,来创建新数据库
self.conn.autocommit = True self.conn.set_isolation_level(0) with self.conn.cursor() as cur: cur.execute(f'''DROP DATABASE IF EXISTS {db_name};''') cur.execute(f'''CREATE DATABASE {db_name} OWNER {role_name} TEMPLATE template_postgis;''')
如果pgAdmin4里已经连接了 template 和删除的目标数据库中任意1个,则会报错
psycopg2.errors.ObjectInUse: database "XXX" is being accessed by other users
DETAIL: There is 1 other session using the database.
—————————————