mirror of
https://github.com/zulip/zulip.git
synced 2025-10-31 20:13:46 +00:00
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
#!/usr/bin/env python2.7
|
|
|
|
from __future__ import print_function
|
|
import subprocess
|
|
import sys
|
|
import logging
|
|
import dateutil.parser
|
|
import pytz
|
|
from datetime import datetime, timedelta
|
|
|
|
logging.basicConfig(format="%(asctime)s %(levelname)s: %(message)s")
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def run(args, dry_run=False):
|
|
if dry_run:
|
|
print("Would have run: " + " ".join(args))
|
|
return ""
|
|
|
|
p = subprocess.Popen(args, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
stdout, stderr = p.communicate()
|
|
if p.returncode:
|
|
logger.error("Could not invoke %s\nstdout: %s\nstderror: %s"
|
|
% (args[0], stdout, stderr))
|
|
sys.exit(1)
|
|
return stdout
|
|
|
|
# Only run if we're the master
|
|
if run(['psql', '-t', '-c', 'select pg_is_in_recovery()']).strip() != 'f':
|
|
sys.exit(0)
|
|
|
|
run(['env-wal-e', 'backup-push', '/var/lib/postgresql/9.1/main'])
|
|
|
|
now = datetime.now(tz=pytz.utc)
|
|
with open('/var/lib/nagios_state/last_postgres_backup', 'w') as f:
|
|
f.write(now.isoformat())
|
|
f.write("\n")
|
|
|
|
backups = {}
|
|
lines = run(['env-wal-e', 'backup-list']).split("\n")
|
|
for line in lines[1:]:
|
|
if line:
|
|
backup_name, date, _, _ = line.split()
|
|
backups[dateutil.parser.parse(date)] = backup_name
|
|
|
|
one_month_ago = now - timedelta(days=30)
|
|
for date in sorted(backups.keys(), reverse=True):
|
|
if date < one_month_ago:
|
|
run(['env-wal-e', 'delete', '--confirm', 'before', backups[date]])
|
|
# Because we're going from most recent to least recent, we
|
|
# only have to do one delete operation
|
|
break
|