From ad438f9e17decda2efca0e8974995a6d028f7788 Mon Sep 17 00:00:00 2001
From: Michael Fenn <fennm@deshawresearch.com>
Date: Fri, 7 Nov 2014 10:00:53 -0500
Subject: SYSV: add a migration tool to change simplename attributes to
 simplefile

---
 tools/upgrade/1.3/README                     |  4 +++
 tools/upgrade/1.3/migrate_sysv_simplename.py | 51 ++++++++++++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100755 tools/upgrade/1.3/migrate_sysv_simplename.py

(limited to 'tools')

diff --git a/tools/upgrade/1.3/README b/tools/upgrade/1.3/README
index 1a919f869..29fd9886b 100644
--- a/tools/upgrade/1.3/README
+++ b/tools/upgrade/1.3/README
@@ -24,3 +24,7 @@ migrate_probe_groups_to_db.py
     - Migrate Probe host and group data from XML to DB backend for Metadata 
       and Probe plugins. Does not migrate individual probe return data. Assumes 
       migration to BOTH Metadata and Probe to database backends. 
+
+migrate_sysv_simplename.py
+    - Migrate any Pkgmgr entries which may have been using the simplename 
+      attribute introduced in 1.3.5 to the simplefile attribute 
diff --git a/tools/upgrade/1.3/migrate_sysv_simplename.py b/tools/upgrade/1.3/migrate_sysv_simplename.py
new file mode 100755
index 000000000..f6599756b
--- /dev/null
+++ b/tools/upgrade/1.3/migrate_sysv_simplename.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+
+import os
+import sys
+import glob
+import lxml.etree
+import Bcfg2.Options
+
+def main():
+    opts = dict(repo=Bcfg2.Options.SERVER_REPOSITORY)
+    setup = Bcfg2.Options.OptionParser(opts)
+    setup.parse(sys.argv[1:])
+
+    files = []
+    for plugin in ['Pkgmgr']:
+        files.extend(glob.glob(os.path.join(setup['repo'], plugin, "*")))
+
+    for bfile in files:
+        bdata = lxml.etree.parse(bfile)
+        changed = False
+
+        if not bdata.xpath("//@type='sysv'"):
+            print("%s doesn't contain any sysv packages, skipping" % bfile)
+            continue
+
+        pkglist = bdata.getroot()
+        if pkglist.tag != "PackageList":
+            print("%s doesn't look like a PackageList, skipping" % bfile)
+            continue
+
+        for pkg in bdata.xpath("//Package"):
+            if "simplename" in pkg.attrib:
+                pkg.set("simplefile", pkg.get("simplename"))
+                del pkg.attrib["simplename"]
+                changed = True
+
+        # if we switched to simplefile, we also need to switch to uri
+        if changed and "url" in pkglist.attrib:
+            pkglist.set("uri", pkglist.get("url"))
+            del pkglist.attrib["url"]
+
+        if changed:
+            print("Writing %s" % bfile)
+            try:
+                open(bfile, "w").write(lxml.etree.tostring(bdata))
+            except IOError:
+                err = sys.exc_info()[1]
+                print("Could not write %s: %s" % (bfile, err))
+
+if __name__ == '__main__':
+    sys.exit(main())
-- 
cgit v1.2.3-1-g7c22