backup script for repositories

pylons_app/lib/backup_manager.py

@@ -0,0 +1,52 b''
+import logging
+from mercurial import config
+import tarfile
+import os
+import datetime
+import sys
+logging.basicConfig(level = logging.DEBUG,
+                    format = "%(asctime)s %(levelname)-5.5s %(message)s")
+
+class BackupManager(object):
+    def __init__(self):
+
+        dn = os.path.dirname
+        self.backup_file_path = os.path.join(dn(dn(dn(__file__))), 'data')
+        cfg = config.config()
+        try:
+            cfg.read(os.path.join(dn(dn(dn(__file__))), 'hgwebdir.config'))
+        except IOError:
+            logging.error('Could not read hgwebdir.config')
+            sys.exit()
+        self.set_repos_path(cfg.items('paths'))
+        logging.info('starting backup for %s', self.repos_path)
+        logging.info('backup target %s', self.backup_file_path)
+
+        if not os.path.isdir(self.repos_path):
+            raise Exception('Not a valid directory in %s' % self.repos_path)
+
+    def set_repos_path(self, paths):
+        repos_path = paths[0][1].split('/')
+        if repos_path[-1] in ['*', '**']:
+            repos_path = repos_path[:-1]
+        if repos_path[0] != '/':
+            repos_path[0] = '/'
+        self.repos_path = os.path.join(*repos_path)
+
+    def backup_repos(self):
+        today = datetime.datetime.now().weekday() + 1
+        bckp_file = os.path.join(self.backup_file_path,
+                                 "mercurial_repos.%s.tar.gz" % today)
+        tar = tarfile.open(bckp_file, "w:gz")
+
+        for dir in os.listdir(self.repos_path):
+            logging.info('backing up %s', dir)
+            tar.add(os.path.join(self.repos_path, dir), dir)
+        tar.close()
+
+
+if __name__ == "__main__":
+    bm = BackupManager()
+    bm.backup_repos()
+
+