Changeset - fd0c6fa2df82
[Not reviewed]
default
0 1 0
Marcin Kuzminski - 16 years ago 2010-03-01 12:29:27

backup script update
1 file changed with 27 insertions and 4 deletions:
0 comments (0 inline, 0 general)
pylons_app/lib/backup_manager.py
Show inline comments
 
@@ -4,12 +4,14 @@ 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):
 

	
 
        self.id_rsa_path = '/home/pylons/id_rsa'
 
        self.check_id_rsa()
 
        cur_dir = os.path.realpath(__file__)
 
        dn = os.path.dirname
 
        self.backup_file_path = os.path.join(dn(dn(dn(cur_dir))), 'data')
 
@@ -26,6 +28,11 @@ class BackupManager(object):
 
        if not os.path.isdir(self.repos_path):
 
            raise Exception('Not a valid directory in %s' % self.repos_path)
 

	
 
    def check_id_rsa(self):
 
        if not os.path.isfile(self.id_rsa_path):
 
            logging.error('Could not load id_rsa key file in %s', self.id_rsa_path)
 
            sys.exit()
 

	
 
    def set_repos_path(self, paths):
 
        repos_path = paths[0][1].split('/')
 
        if repos_path[-1] in ['*', '**']:
 
@@ -36,19 +43,35 @@ class BackupManager(object):
 

	
 
    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)
 
        self.backup_file_name = "mercurial_repos.%s.tar.gz" % today
 
        bckp_file = os.path.join(self.backup_file_path, self.backup_file_name)
 
        tar = tarfile.open(bckp_file, "w:gz")
 

	
 
        for dir in os.listdir(self.repos_path):
 
        for dir in os.listdir(self.repos_path)[:2]:
 
            logging.info('backing up %s', dir)
 
            tar.add(os.path.join(self.repos_path, dir), dir)
 
        tar.close()
 
        logging.info('finished backup of mercurial repositories')
 

	
 

	
 

	
 
    def transfer_files(self):
 
        params = {
 
                  'id_rsa_key': self.id_rsa_path,
 
                  'backup_file_path':self.backup_file_path,
 
                  'backup_file_name':self.backup_file_name,
 
                  }
 
        cmd = ['scp', '-i', '%(id_rsa_key)s' % params,
 
               '%(backup_file_path)s/%(backup_file_name)s' % params,
 
               'root@192.168.2.102:/backups/mercurial' % params]
 

	
 
        subprocess.Popen(cmd)
 
        logging.info('Transfered file %s to %s', self.backup_file_name, cmd[4])
 

	
 

	
 
if __name__ == "__main__":
 
    bm = BackupManager()
 
    bm.backup_repos()
 
    bm.transfer_files()
 

	
 

	
0 comments (0 inline, 0 general)