pg_restore可以自动解压缩吗？

Question

我有一个程序可以备份PostgreSQL数据库，并给用户一个机会把这些备份压缩成zip文件：

if bkp_type == 'gz':
   command = 'pg_dumpall -U {} -h {} -p {} | gzip > {}'.format(user, server, port, file)
elif bkp_type == 'bz2':
   command = 'pg_dumpall -U {} -h {} -p {} | bzip2 > {}'.format(user, server, port, file)
elif bkp_type == 'zip':
   command = 'pg_dumpall -U {} -h {} -p {} | zip > {}'.format(user, server, port, file)
else:
   command = 'pg_dumpall -U {} -h {} -p {} > {}'.format(user, server, port, file)

result = subprocess.call(command, shell=True)

现在，我想用pg_restore来恢复这些备份，但如果文件是压缩的，我就无法直接恢复。请问有没有办法像pg_dump或pg_dumpall那样直接用pg_restore恢复压缩文件，还是说我得先检查压缩的类型，然后用Python解压缩呢？

Answer 1

最后，我找到了一个解决方案，可以用来恢复通过pg_dump生成的备份（目前还不支持pg_dumpall）：

if ext == 'gz':
   command = 'gunzip -c {} -k | pg_restore -U {} -h {} -p {}' \
             '-d {}'.format(file, user, server, port, new_dbname)
elif ext == 'bz2':
   command = 'bunzip2 -c {} -k | pg_restore -U {} -h {} -p {}' \
             '-d {}'.format(file, user, server, port, new_dbname)
elif ext == 'zip':
   command = 'unzip -p {} | pg_restore -U {} -h {} -p {} ' \
             '-d {}'.format(file, user, server, port, new_dbname)
else:
   command = 'pg_restore -U {} -h {} -p {} -d {} {}'.format(user,
                server, port, new_dbname, file)