backup_manager.py
107 lines
| 3.5 KiB
| text/x-python
|
PythonLexer
r252 | #!/usr/bin/env python | |||
# encoding: utf-8 | ||||
# mercurial repository backup manager | ||||
# Copyright (C) 2009-2010 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 | ||||
""" | ||||
|
r25 | import logging | ||
import tarfile | ||||
import os | ||||
import datetime | ||||
import sys | ||||
|
r28 | import subprocess | ||
|
r33 | logging.basicConfig(level=logging.DEBUG, | ||
format="%(asctime)s %(levelname)-5.5s %(message)s") | ||||
|
r25 | |||
class BackupManager(object): | ||||
r434 | 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' | ||||
|
r25 | logging.info('starting backup for %s', self.repos_path) | ||
logging.info('backup target %s', self.backup_file_path) | ||||
r434 | 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) | ||||
|
r28 | sys.exit() | ||
r434 | 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 | ||||
|
r25 | |||
def backup_repos(self): | ||||
|
r28 | bckp_file = os.path.join(self.backup_file_path, self.backup_file_name) | ||
|
r25 | tar = tarfile.open(bckp_file, "w:gz") | ||
|
r34 | 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) | ||||
|
r25 | tar.close() | ||
|
r26 | logging.info('finished backup of mercurial repositories') | ||
|
r25 | |||
|
r28 | |||
def transfer_files(self): | ||||
params = { | ||||
'id_rsa_key': self.id_rsa_path, | ||||
r434 | 'backup_file':os.path.join(self.backup_file_path, | |||
self.backup_file_name), | ||||
'backup_server':self.backup_server | ||||
|
r28 | } | ||
r39 | cmd = ['scp', '-l', '40000', '-i', '%(id_rsa_key)s' % params, | |||
r434 | '%(backup_file)s' % params, | |||
'%(backup_server)s' % params] | ||||
|
r28 | |||
|
r35 | subprocess.call(cmd) | ||
|
r28 | logging.info('Transfered file %s to %s', self.backup_file_name, cmd[4]) | ||
|
r33 | |||
def rm_file(self): | ||||
|
r36 | logging.info('Removing file %s', self.backup_file_name) | ||
|
r34 | os.remove(os.path.join(self.backup_file_path, self.backup_file_name)) | ||
|
r33 | |||
|
r28 | |||
|
r25 | if __name__ == "__main__": | ||
r434 | ||||
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) | ||||
|
r34 | B_MANAGER.backup_repos() | ||
B_MANAGER.transfer_files() | ||||
B_MANAGER.rm_file() | ||||
|
r25 | |||