summaryrefslogtreecommitdiff
path: root/puppet/modules/couchdb/templates/couchdb-backup.py.erb
blob: c49df65b3f7945db69fbef54aabfba75b754d118 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
#!/usr/bin/env python
# file manage by puppet 

import os
import gzip
import tarfile
import datetime
import urllib2
import simplejson
import couchdb.tools.dump
from os.path import join

DB_URL="http://127.0.0.1:5984"
DUMP_DIR="<%= backupdir %>"
TODAY=datetime.datetime.today().strftime("%A").lower()

ftar = os.path.join(DUMP_DIR,"%s.tar" % TODAY)
tmp_ftar = os.path.join(DUMP_DIR,"_%s.tar" % TODAY)
tar = tarfile.open(tmp_ftar, "w")

databases = simplejson.load(urllib2.urlopen("%s/_all_dbs" % DB_URL))

for db in databases:
  db_file = os.path.join(DUMP_DIR,"%s.gz" % db)
  f = gzip.open(db_file, 'wb')
  couchdb.tools.dump.dump_db(os.path.join(DB_URL,db), output=f)
  f.close()
  tar.add(db_file,"%s.gz" % db)
  os.remove(db_file)

tar.close()
os.rename(tmp_ftar,ftar)