changeset 25:e89967ca7f68

backup script for repositories
author Marcin Kuzminski
date Sun, 28 Feb 2010 17:03:17 +0100
parents 170af9dcfaed
children b3307ca60bfb
files pylons_app/lib/backup_manager.py
diffstat 1 files changed, 52 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pylons_app/lib/backup_manager.py	Sun Feb 28 17:03:17 2010 +0100
@@ -0,0 +1,52 @@
+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()
+
+