Mercurial > archived > louis > tools > lappback

comparison backup-lapp-app @ 0:b4d371b24a5e default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Initial Import
author Louis Opter <kalessin@kalessin.fr>
date Thu, 01 Apr 2010 19:18:49 +0000
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:b4d371b24a5e
1 #!/usr/bin/env python
2
3 # The MIT License
4 #
5 # Copyright (c) 2010 Louis Opter <kalessin@kalessin.fr>
6 #
7 # Permission is hereby granted, free of charge, to any person obtaining a copy
8 # of this software and associated documentation files (the "Software"), to deal
9 # in the Software without restriction, including without limitation the rights
10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 # copies of the Software, and to permit persons to whom the Software is
12 # furnished to do so, subject to the following conditions:
13 #
14 # The above copyright notice and this permission notice shall be included in
15 # all copies or substantial portions of the Software.
16 #
17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23 # THE SOFTWARE.
24
25 import errno
26 import os
27 import shutil
28 import subprocess
29 import sys
30 import tarfile
31 import tempfile
32
33 class TempDir(object):
34 """Wrap tempfile.mkdtemp to use the with statement"""
35
36 def __init__(self):
37 pass
38
39 def __enter__(self):
40 self.__path = tempfile.mkdtemp()
41 return self.__path
42
43 def __exit__(self, exc_type, exc_value, traceback):
44 shutil.rmtree(self.__path)
45
46 def status(message):
47 print '%s: %s.' % (os.path.basename(sys.argv[0]), message)
48
49 def die(message):
50 status(message)
51 status('a log has been recorded in %s' % log_path)
52 sys.exit(1)
53
54 def dump_db():
55 """Use pg_dump to backup the databse into `root'"""
56
57 dump_path = os.path.join(root, db_name + '.sql')
58 with open(dump_path, 'w') as dump:
59 status('Dumping database to %s' % dump_path)
60 if subprocess.call(['sudo', '-u', 'postgres', 'pg_dump', db_name],
61 stdout=dump, stderr=log) != 0:
62 die('error while dumping database %s to a temporary directory' % db_name)
63
64 status('Database %s dumped' % db_name)
65
66 def archive():
67 """Create a tar bzipped archive of the db dump and the www-dir"""
68
69 try:
70 archive = tarfile.open(os.path.join(root, archive_name), 'w:bz2')
71 archive.add(os.path.join(root, db_name + '.sql'), db_name + '.sql')
72 archive.add(www_dir, os.path.basename(www_dir))
73 archive.close()
74 except tarfile.TarError, ex:
75 die('Can\'t create the tar archive %s: %s' % (archive_name, str(ex)))
76
77 status('Archive %s created' % archive_name)
78
79 def send_archive():
80 """Send the tarball using scp or s3cmd"""
81
82 archive = os.path.join(root, archive_name)
83 if dest[0:5] == 's3://':
84 ret = subprocess.call(['s3cmd', '--acl-private', 'put', archive, dest], stdout=log, stderr=log)
85 else:
86 ret = subprocess.call(['scp', archive, dest], stdout=log, stderr=log)
87
88 if ret != 0:
89 die('error while cpying %s to %s' % (archive_name, dest))
90
91 status('Archive %s copied to "%s"' % (archive_name, dest))
92
93 if __name__ == '__main__':
94 if len(sys.argv) != 4 or sys.argv[1] == '--help':
95 print 'Usage: %s db-name www-dir [[user@]host:]path|s3://bucket/[path]' % os.path.basename(sys.argv[0])
96 sys.exit(0)
97
98 # globals
99 db_name = sys.argv[1]
100 # python basename return '' for a path with an ending slash...
101 www_dir = sys.argv[2] != '/' and os.path.join('/var/www', sys.argv[2]) or '/var/www'
102 dest = sys.argv[3]
103 archive_name = 'backup-lapp-app-%s.tar.bz2' % db_name
104 log_path = os.path.join(os.getenv('TMP') or '/tmp/', db_name + '.log')
105
106 if not os.path.exists(www_dir):
107 status('%s: %s' % (www_dir, os.strerror(errno.ENOENT)))
108 sys.exit(1)
109
110 with open(log_path, 'w') as log:
111 with TempDir() as root:
112 status('root=%s' % root)
113 dump_db()
114 archive()
115 send_archive()
116 os.remove(log_path)
117
118 sys.exit(0)