Initial bcfg2-admin backup part

2 files changed, 48 insertions, 1 deletions

diff --git a/src/lib/Server/Admin/Backup.py b/src/lib/Server/Admin/Backup.py
new file mode 100644
index 000000000..583f0329c
--- /dev/null
+++ b/src/lib/Server/Admin/Backup.py
@@ -0,0 +1,47 @@
+import glob
+import sys
+import time
+import tarfile
+import Bcfg2.Server.Admin
+import Bcfg2.Options
+
+class Backup(Bcfg2.Server.Admin.MetadataCore):
+    __shorthelp__ = "Make a backup of the Bcfg2 repository."
+    __longhelp__ = (__shorthelp__ + "\n\nbcfg2-admin backup start"
+                                    "\n\nbcfg2-admin backup restore")
+    __usage__ = ("bcfg2-admin backup [start|restore]")
+
+    def __init__(self, configfile):
+        Bcfg2.Server.Admin.MetadataCore.__init__(self, configfile,
+                                                 self.__usage__)
+
+    def __call__(self, args):
+        Bcfg2.Server.Admin.MetadataCore.__call__(self, args)
+        # Get Bcfg2 repo directory
+        opts = {'repo': Bcfg2.Options.SERVER_REPOSITORY}
+        setup = Bcfg2.Options.OptionParser(opts)
+        setup.parse(sys.argv[1:])
+        repo = setup['repo']
+        
+	if len(args) == 0:
+            self.errExit("No argument specified.\n"
+                         "Please see bcfg2-admin backup help for usage.")
+        if args[0] == 'start':
+            timestamp = time.strftime('%Y%m%d%H%M%S')
+            format = 'gz'
+            mode = 'w:' + format
+            filename = timestamp + '.tar' + '.' + format
+            out = tarfile.open(filename, mode=mode)
+            content = os.listdir(os.getcwd())           
+            for item in content:
+                out.add(item)
+            out.close()
+            print "Archive %s was stored.\nLocation: %s" % (filename, datastore)
+
+        elif args[0] == 'restore':
+            print 'Not implemented yet'
+        
+        else:
+            print "No command specified"
+            raise SystemExit(1)
+
diff --git a/src/lib/Server/Admin/__init__.py b/src/lib/Server/Admin/__init__.py
index 3a088b2fb..bb5c41895 100644
--- a/src/lib/Server/Admin/__init__.py
+++ b/src/lib/Server/Admin/__init__.py
@@ -2,7 +2,7 @@ __revision__ = '$Revision$'
 
 __all__ = ['Mode', 'Client', 'Compare', 'Init', 'Minestruct', 'Perf',
            'Pull', 'Query', 'Reports', 'Snapshots', 'Tidy', 'Viz', 
-           'Xcmd', 'Group']
+           'Xcmd', 'Group', 'Backup']
 
 import ConfigParser
 import logging