tools: Add migrate_perms_to_mode tool

Helps users to migrate their repositories from the perms attribute to
mode.

Signed-off-by: Sol Jerome <sol.jerome@gmail.com>

2 files changed, 72 insertions, 0 deletions

diff --git a/tools/upgrade/1.3/README b/tools/upgrade/1.3/README
index e72650729..2831d8f00 100644
--- a/tools/upgrade/1.3/README
+++ b/tools/upgrade/1.3/README
@@ -15,3 +15,7 @@ service_modes.py
 
 migrate_dbstats.py
     - Convert old DBStats entries to the new Reporting system
+
+migrate_perms_to_mode.py
+    - Convert perms attribute to mode (note that if you have info/:info
+      files, you should run migrate_info.py first)
diff --git a/tools/upgrade/1.3/migrate_perms_to_mode.py b/tools/upgrade/1.3/migrate_perms_to_mode.py
new file mode 100644
index 000000000..0aa9c574c
--- /dev/null
+++ b/tools/upgrade/1.3/migrate_perms_to_mode.py
@@ -0,0 +1,68 @@
+#!/usr/bin/env python
+
+import lxml.etree
+import os
+import sys
+
+import Bcfg2.Options
+
+
+def setmodeattr(elem):
+    """Set the mode attribute for a given element."""
+    if elem.attrib.has_key('perms'):
+        elem.set('mode', elem.get('perms'))
+        del elem.attrib['perms']
+        return True
+
+
+def writefile(f, xdata):
+    """Write xml data to a file"""
+    newfile = open(f, 'w')
+    newfile.write(lxml.etree.tostring(xdata, pretty_print=True))
+    newfile.close()
+
+
+def convertinfo(ifile):
+    """Do perms -> mode conversion for info.xml files."""
+    xdata = lxml.etree.parse(ifile)
+    found = False
+    for i in xdata.findall('//Info'):
+        found = setmodeattr(i)
+    if found:
+        writefile(ifile, xdata)
+
+
+def convertstructure(structfile):
+    """Do perms -> mode conversion for structure files."""
+    xdata = lxml.etree.parse(structfile)
+    found = False
+    for path in xdata.findall('//BoundPath'):
+        found = setmodeattr(path)
+    for path in xdata.findall('//Path'):
+        found = setmodeattr(path)
+    if found:
+        writefile(structfile, xdata)
+
+
+def main():
+    opts = dict(repo=Bcfg2.Options.SERVER_REPOSITORY,
+                configfile=Bcfg2.Options.CFILE,
+                plugins=Bcfg2.Options.SERVER_PLUGINS)
+    setup = Bcfg2.Options.OptionParser(opts)
+    setup.parse(sys.argv[1:])
+    repo = setup['repo']
+
+    for plugin in setup['plugins']:
+        if plugin in ['Base', 'Bundler', 'Rules']:
+            for root, dirs, files in os.walk(os.path.join(repo, plugin)):
+                for fname in files:
+                    convertstructure(os.path.join(root, fname))
+        if plugin not in ['Cfg', 'TGenshi', 'TCheetah']:
+            continue
+        for root, dirs, files in os.walk(os.path.join(repo, plugin)):
+            for fname in files:
+                if fname == 'info.xml':
+                    convertinfo(os.path.join(root, fname))
+
+if __name__ == '__main__':
+    sys.exit(main())