Mercurial > kallithea
view rhodecode/lib/backup_manager.py @ 1039:51b70569c330 beta
extended admin rescan to show what repositories was added and what removed
fixed found bugs, added expunge to cached getter since it introduce some more problems.
Added some relationship to properly clean related objects from database.
author | Marcin Kuzminski <marcin@python-works.com> |
---|---|
date | Sun, 13 Feb 2011 01:38:39 +0100 |
parents | 07a6e8c65526 |
children | 6832ef664673 |
line wrap: on
line source
#!/usr/bin/env python # encoding: utf-8 # mercurial repository backup manager # Copyright (C) 2009-2011 Marcin Kuzminski <marcin@python-works.com> # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; version 2 # of the License or (at your opinion) any later version of the license. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, # MA 02110-1301, USA. """ Created on Feb 28, 2010 Mercurial repositories backup manager @author: marcink """ import logging import tarfile import os import datetime import sys import subprocess logging.basicConfig(level=logging.DEBUG, format="%(asctime)s %(levelname)-5.5s %(message)s") class BackupManager(object): def __init__(self, repos_location, rsa_key, backup_server): today = datetime.datetime.now().weekday() + 1 self.backup_file_name = "mercurial_repos.%s.tar.gz" % today self.id_rsa_path = self.get_id_rsa(rsa_key) self.repos_path = self.get_repos_path(repos_location) self.backup_server = backup_server self.backup_file_path = '/tmp' logging.info('starting backup for %s', self.repos_path) logging.info('backup target %s', self.backup_file_path) def get_id_rsa(self, rsa_key): if not os.path.isfile(rsa_key): logging.error('Could not load id_rsa key file in %s', rsa_key) sys.exit() return rsa_key def get_repos_path(self, path): if not os.path.isdir(path): logging.error('Wrong location for repositories in %s', path) sys.exit() return path def backup_repos(self): bckp_file = os.path.join(self.backup_file_path, self.backup_file_name) tar = tarfile.open(bckp_file, "w:gz") for dir_name in os.listdir(self.repos_path): logging.info('backing up %s', dir_name) tar.add(os.path.join(self.repos_path, dir_name), dir_name) tar.close() logging.info('finished backup of mercurial repositories') def transfer_files(self): params = { 'id_rsa_key': self.id_rsa_path, 'backup_file':os.path.join(self.backup_file_path, self.backup_file_name), 'backup_server':self.backup_server } cmd = ['scp', '-l', '40000', '-i', '%(id_rsa_key)s' % params, '%(backup_file)s' % params, '%(backup_server)s' % params] subprocess.call(cmd) logging.info('Transfered file %s to %s', self.backup_file_name, cmd[4]) def rm_file(self): logging.info('Removing file %s', self.backup_file_name) os.remove(os.path.join(self.backup_file_path, self.backup_file_name)) if __name__ == "__main__": repo_location = '/home/repo_path' backup_server = 'root@192.168.1.100:/backups/mercurial' rsa_key = '/home/id_rsa' B_MANAGER = BackupManager(repo_location, rsa_key, backup_server) B_MANAGER.backup_repos() B_MANAGER.transfer_files() B_MANAGER.rm_file()