PostgreSQL 学习之使用psycopg2 操作之数据库不存在才创建

Posted yungiu

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PostgreSQL 学习之使用psycopg2 操作之数据库不存在才创建相关的知识,希望对你有一定的参考价值。

大声的告诉我是不是被我标题中的两个“之”给带进来的??手动滑稽

需求

自己的一款软件 GitHub 地址,关于PostgreSQL 已经设置成运行后自动创建序列,表和函数,但是数据库还是要手动去创建,很不方便,想使用创建序列和表同样的方法,去自动创建数据库

过程

DDL 语句如下:

DB_NAME = """
CREATE DATABASE if not exists ;

ALTER DATABASE  OWNER TO postgres;
""".format(DATABASE_NAME, DATABASE_NAME)
发现会报一个DDL 语句语法错误
psycopg2.ProgrammingError: syntax error at or near "not"
LINE 2: CREATE DATABASE if not exists test_classs;

创建序列和表的时候都没问题啊(函数我使用的是先删除,后创建),这怎么报错了?本着先运行成功,再实现功能的原则,我把“if not exists”给去掉,结果还是报错:

psycopg2.InternalError: CREATE DATABASE cannot run inside a transaction block

不能在事务块中创建数据库,大概意思就是这样不安全吧,百度加谷歌,有两种方法:

1.在 psycopg2 extensions 里使用 ISOLATION_LEVEL_AUTOCOMMIT,原理就是让连接发出命令时不启动任何事务,看常量名字,字面意思也是自动提交,并且不需要commit()或rollback()如:

import psycopg2
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT # <-- ADD THIS LINE

con = psycopg2.connect(...)

con.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT) # <-- ADD THIS LINE

cur = con.cursor()
cur.execute("CREATE DATABASE %s  ;" % self.db_name)
cur.close()
con.close()

2.让连接处于自动提交模式,使用连接对象的autocommit 属性,如:

import psycopg2

con = psycopg2.connect(...)
con.autocommit = True

cur = con.cursor()
cur.execute(CREATE DATABASE ;.format(db_name))
cur.close()
# 如果连接不关闭,则需要改回连接对象的autocommit 属性
con.autocommit = False
# 如果连接关闭,则不需要执行上面这行代码
# con.close()

但是当我运行的时候,还是遇到了另外一个错误:

psycopg2.InternalError: CREATE DATABASE cannot be executed from a function or multi-command string

从字面意思上看,是不能在函数中创建数据库,且不能在多命令的字符串中创建数据库,函数我这里没用,只能是多行命令了,看了一下自己的DDL 语句,发现自己复制创建表的DDL 语句的部分有点多了。。从最简单的做起吧,毕竟上面的这两个例子都没有问题,将DDL 语句改为:

DB_NAME = """
CREATE DATABASE ;
""".format(DATABASE_NAME)

再次运行,虽然有一丢丢慢,但是还是创建成功了,但是在这里,我建议使用第二种方法,因为简单,不用再导入这个很长的常量,而且控制还很方便,比如下面这个需求,创建第一个数据库,创建第二个数据库(怎么会有这么奇葩的需求,哈哈,都是为了举例方便)

import psycopg2
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT,ISOLATION_LEVEL_DEFAULT

DATABASE_NAME = test_classs
DATABASE_NAME2 = test_classssssss

DB_NAME = """
CREATE DATABASE ;
""".format(DATABASE_NAME)

DB_NAME2 = """
CREATE DATABASE ;
""".format(DATABASE_NAME2)

conn = psycopg2.connect(...)
conn.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
cur = conn.cursor()
cur.execute(DB_NAME)
conn.set_isolation_level(ISOLATION_LEVEL_DEFAULT)
cur.execute(DB_NAME2)
cur.close()
conn.close()

看到了吧,你需要导入两个很长很长的变量名,当然,在IDE 的提示下,这样其实也不费事,但是我总感觉这样不怎么优雅,毕竟是Python,不优雅怎么行?所以我们可以使用另外一种方式:

import psycopg2

DATABASE_NAME = test_classs
DATABASE_NAME2 = test_classssssss

DB_NAME = """
CREATE DATABASE ;
""".format(DATABASE_NAME)

DB_NAME2 = """
CREATE DATABASE ;
""".format(DATABASE_NAME2)

conn = psycopg2.connect(...)
conn.autocommit = True
cur = conn.cursor()
cur.execute(DB_NAME)
conn.autocommit = False
cur.execute(DB_NAME2)
cur.close()
conn.close()

效果是一样的,看着是不是舒服很多了?

好吧,现在我们的数据库是可以创建了,那么怎么样实现如果不存在就创建,存在就不处理呢?

既然不能像‘mysql’那样用‘if not exists’来创建数据库,那么只能换另外一种思路了,先查询看看这个数据库是否存在,如果不存在就设置自动提交然后创建,如果存在,就不做处理,excuse me?这是换思路了吗?咳咳。。好吧,思路没变,只是过程用代码实现了而已。。具体代码如下:

import psycopg2


DATABASE_NAME = test_classss
DB_NAME_EXIST = """
select * from pg_database where datname=‘‘;
""".format(DATABASE_NAME)
DB_NAME = """
CREATE DATABASE ;
""".format(DATABASE_NAME)
conn = psycopg2.connect(...)
cur = conn.cursor()
cur.execute(DB_NAME_EXIST)
result = cur.fetchall()
if result == []:
    conn.autocommit = True
    cur.execute(DB_NAME)
    conn.autocommit = False
else:
    print( is exist.format(DATABASE_NAME))
cur.close()
conn.close()

 

注意,ALL_DB_NAME 这条SQL 语句的条件后面占位符必须用引号引起来,否则会报 column "test_class" does not exist 的错误

测试了一下,貌似没有问题,那么改用一个不存在的数据库看看,果然,没有我想的那么简单。。

    conn.autocommit = True
psycopg2.ProgrammingError: set_session cannot be used inside a transaction

百度了一下,也没找到答案,不过根据字面意思,肯定是和事务有关,先尝试了一个最笨的方法,就是查询后记录查询结果,然后把游标和连接都关闭,然后再创建新的连接,新的游标,根据结果去创建数据库,或者不处理,发现没有问题,虽然感觉怪怪的,但是也蛮开心的,就在要往博客园上记下解决方法的时候,忽然想到了,把上次的查询后的给提交一下不就OK 了?快被自己蠢哭了,这也是为什么我打了那么多字,而不直接上代码的原因,这是经验的不足,也是数据库知识的不足,脸红,直接上代码吧,其实就加一行代码就解决了。

结果

 

import psycopg2


DATABASE_NAME = test_classss
DB_NAME_EXIST = """
select * from pg_database where datname=‘‘;
""".format(DATABASE_NAME)
DB_NAME = """
CREATE DATABASE ;
""".format(DATABASE_NAME)
conn = psycopg2.connect(...)
cur = conn.cursor()
cur.execute(DB_NAME_EXIST)
conn.commit()  # <-- ADD THIS LINE
result = cur.fetchall()
if result == []:
    conn.autocommit = True
    cur.execute(DB_NAME)
    conn.autocommit = False
else:
    print( is exist.format(DATABASE_NAME))
cur.close()
conn.close()

 

 

 

行吧,到此位置,功能实现~我们下次再见,如果你觉得对你有帮助,麻烦点个“推荐”,谢谢~

当然,如果你有更优雅的办法,请在下面留言,谢谢

 

参考

使用python创建Postgres数据库:https://www.e-learn.cn/content/wangluowenzhang/448331

 

 

以上是关于PostgreSQL 学习之使用psycopg2 操作之数据库不存在才创建的主要内容,如果未能解决你的问题,请参考以下文章

使用 SSH 隧道的 PostgreSQL/psycopg2 密码验证

psycopg2/python 将数据从 postgresql 复制到 Amazon RedShift(postgresql)

tcp-h数据&postgresql查询(数据库学习之二)

SQLAlchemy/psycopg2 到 PostgreSQL 数据库的连接是不是加密

psycopg2 Google Analytics数据到postgresql

使用 PostgreSQL (psycopg2) 和 Django 运行 Python 3.4 时出错