[英]Postgresql Database Backup Using Python
我想使用 Python 代码备份数据库。 我想备份一些相关数据的表。 如何使用“SELECT”语句备份以及如何选择所需的表?
例如
我想获取一些表的 2014-05-01 到 2014-05-10 的数据,并将这个结果输出为 .sql 扩展文件
如何使用 python 代码获取这种格式? 如果你不介意,请解释一下。 谢谢。
使用 psycopg2 建立数据连接。 文档中有不少例子:
配置数据源后,通过将结果集打印到文件来遍历“SELECT”语句的结果,构建“INSERT INTO”语句。 基本上是一些反向逻辑。
这样,如果时间到了并且您需要使用备份文件,您只需运行 SQL 文件,将数据插入回...
例子:
import psycopg2
import sys
con = None
try:
con = psycopg2.connect(database='local', user='local', password='local',port='1970')
cur = con.cursor()
cur.execute('SELECT x FROM t')
f = open('test.sql', 'w')
for row in cur:
f.write("insert into t values (" + str(row) + ");")
except psycopg2.DatabaseError, e:
print 'Error %s' % e
sys.exit(1)
finally:
if con:
con.close()
然后恢复:
psql <dbname> <username> < test.sql
干杯,
如果您的操作系统是 Linux,您可以使用下面的代码。 首先,您应该运行apt-get install postgresql
。
def create_essentials():
yaml_file = open("settings.yaml", 'r')
settings = yaml.load(yaml_file)
db_name = settings["db_name"]
db_user = settings["db_user"]
db_password = settings["db_password"]
db_host = settings["db_host"]
db_port = settings["db_port"]
backup_path = settings["backup_path"]
filename = settings["filename"]
filename = filename + "-" + time.strftime("%Y%m%d") + ".backup"
command_str = str(db_host)+" -p "+str(db_port)+" -d "+db_name+" -U "+db_user
return command_str, backup_path, filename
def backup_database(table_names=None):
command_str,backup_path,filename = create_essentials()
command_str = "pg_dump -h "+command_str
if table_names is not None:
for x in table_names:
command_str = command_str +" -t "+x
command_str = command_str + " -F c -b -v -f '"+backup_path+"/"+filename+"'"
try:
os.system(command_str)
print "Backup completed"
except Exception as e:
print "!!Problem occured!!"
print e
def restore_database(table_names=None):
command_str,backup_path,filename = create_essentials()
command_str = "pg_restore -h "+command_str
if table_names is not None:
for x in table_names:
command_str = command_str +" -t "+x
command_str = command_str + " -v '"+backup_path+"/"+filename+"'"
try:
os.system(command_str)
print "Restore completed"
except Exception as e:
print "!!Problem occured!!"
print e
我想到的第一个想法是调用pg_dump命令转储您的表,类似于此处介绍的方法(但 google 有很多替代方法)。
但是,由于您的备份策略要求您选择精确的日期而不仅仅是表格,您可能不得不依赖于一系列查询,然后我的建议是使用像Psycopg这样的库。
编辑:
我无法提供完整的示例,因为我不知道:
以下示例生成一个文件,用于存储单个查询的结果。
import psycopg
conn = psycopg2.connect("dbname=test user=postgres") # change this according to your RDBMS configuration
cursor = conn.cursor()
table_name='YOUR_TABLE_HERE' # place your table name here
with open("table_dump.sql") as f:
cursor.execute("SELECT * FROM %s" % (table_name)) # change the query according to your needs
column_names = []
columns_descr = cursor.description
for c in columns_descr:
column_names.append(c[0])
insert_prefix = 'INSERT INTO %s (%s) VALUES ' % (table_name, ', '.join(column_names))
rows = cursor.fetchall()
for row in rows:
row_data = []
for rd in row:
if rd is None:
row_data.append('NULL')
elif isinstance(rd, datetime.datetime):
row_data.append("'%s'" % (rd.strftime('%Y-%m-%d %H:%M:%S') ))
else:
row_data.append(repr(rd))
f.write('%s (%s);\n' % (insert_prefix, ', '.join(row_data))) # this is the text that will be put in the SQL file. You can change it if you wish.
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.