changeset 34:b4b25ece7797

Code refactoring, and changed proper way of removing files
author Marcin Kuzminski <marcin@python-blog.com>
date Tue, 09 Mar 2010 00:22:29 +0100
parents 22b2234e51cd
children c92f827280a3
files pylons_app/lib/backup_manager.py
diffstat 1 files changed, 13 insertions(+), 9 deletions(-) [+]
line wrap: on
line diff
--- a/pylons_app/lib/backup_manager.py	Mon Mar 08 10:25:13 2010 +0100
+++ b/pylons_app/lib/backup_manager.py	Tue Mar 09 00:22:29 2010 +0100
@@ -1,3 +1,4 @@
+'''BACKUP MANAGER'''
 import logging
 from mercurial import config
 import tarfile
@@ -10,6 +11,8 @@
 
 class BackupManager(object):
     def __init__(self):
+        self.repos_path = None
+        self.backup_file_name = None
         self.id_rsa_path = '/home/pylons/id_rsa'
         self.check_id_rsa()
         cur_dir = os.path.realpath(__file__)
@@ -30,7 +33,8 @@
 
     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)
+            logging.error('Could not load id_rsa key file in %s',
+                          self.id_rsa_path)
             sys.exit()
 
     def set_repos_path(self, paths):
@@ -47,9 +51,9 @@
         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):
-            logging.info('backing up %s', dir)
-            tar.add(os.path.join(self.repos_path, dir), dir)
+        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')
 
@@ -70,14 +74,14 @@
         
     
     def rm_file(self):
-        os.remove(self.backup_file_path)
+        os.remove(os.path.join(self.backup_file_path, self.backup_file_name))
     
 
 
 if __name__ == "__main__":
-    bm = BackupManager()
-    bm.backup_repos()
-    bm.transfer_files()
-    bm.rm_file()
+    B_MANAGER = BackupManager()
+    B_MANAGER.backup_repos()
+    B_MANAGER.transfer_files()
+    B_MANAGER.rm_file()