mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2025-01-31 05:06:41 +00:00
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
|
import subprocess
|
||
|
|
|
||
|
|
|
||
|
|
class PostgresDumper:
|
||
|
|
"""--dbname=postgresql://postgres@%2Frun%2Fpostgresql/pleroma"""
|
||
|
|
|
||
|
|
def __init__(self, db_name: str):
|
||
|
|
self.db_name = db_name
|
||
|
|
self.user = "postgres"
|
||
|
|
self.socket_dir = r"%2Frun%2Fpostgresql"
|
||
|
|
|
||
|
|
def backup_database(self, backup_file: str):
|
||
|
|
# Create the database dump and pipe it to gzip
|
||
|
|
dump_command = [
|
||
|
|
"pg_dump",
|
||
|
|
f"--dbname=postgresql://{self.user}@{self.socket_dir}/{self.db_name}",
|
||
|
|
]
|
||
|
|
gzip_command = ["gzip", "--rsyncable"]
|
||
|
|
|
||
|
|
with open(backup_file, "wb") as f_out:
|
||
|
|
dump_process = subprocess.Popen(dump_command, stdout=subprocess.PIPE)
|
||
|
|
gzip_process = subprocess.Popen(
|
||
|
|
gzip_command, stdin=dump_process.stdout, stdout=f_out
|
||
|
|
)
|
||
|
|
dump_process.stdout.close() # Allow dump_process to receive a SIGPIPE if gzip_process exits
|
||
|
|
gzip_process.communicate()
|
||
|
|
|
||
|
|
return backup_file
|
||
|
|
|
||
|
|
def restore_database(self, backup_file: str):
|
||
|
|
# Decompress the backup file
|
||
|
|
gunzip_command = ["gunzip", backup_file]
|
||
|
|
subprocess.run(gunzip_command, check=True)
|
||
|
|
|
||
|
|
# Restore the database from the decompressed file
|
||
|
|
dump_file = backup_file.replace(".gz", "")
|
||
|
|
restore_command = [
|
||
|
|
"pg_restore",
|
||
|
|
"--dbname=postgresql://{}@{}/{}".format(
|
||
|
|
self.user, self.socket_dir, self.db_name
|
||
|
|
),
|
||
|
|
"--clean",
|
||
|
|
"--create",
|
||
|
|
dump_file,
|
||
|
|
]
|
||
|
|
subprocess.run(restore_command, check=True)
|