From dafa48723db14f32a6ae07aa9a21df53bd945650 Mon Sep 17 00:00:00 2001
From: Jason Kincl <kincljc@ornl.gov>
Date: Wed, 22 May 2013 11:05:18 -0400
Subject: 1.3 Upgrade Tools: Added script to facilitate upgrading from Probe
 XML to database backend

---
 tools/upgrade/1.3/migrate_probe_groups_to_db.py | 63 +++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100755 tools/upgrade/1.3/migrate_probe_groups_to_db.py

(limited to 'tools')

diff --git a/tools/upgrade/1.3/migrate_probe_groups_to_db.py b/tools/upgrade/1.3/migrate_probe_groups_to_db.py
new file mode 100755
index 000000000..540d10505
--- /dev/null
+++ b/tools/upgrade/1.3/migrate_probe_groups_to_db.py
@@ -0,0 +1,63 @@
+#!/bin/env python
+
+import os
+os.environ['DJANGO_SETTINGS_MODULE'] = 'Bcfg2.settings'
+
+import lxml.etree
+import sys
+import Bcfg2.Options
+
+from Bcfg2.Server.Plugins.Metadata import MetadataClientModel
+from Bcfg2.Server.Plugins.Probes import ProbesGroupsModel
+
+def migrate(xclient):
+    client_name = xclient.get('name')
+    try:
+        try:
+            client = MetadataClientModel.objects.get(hostname=client_name)
+        except MetadataClientModel.DoesNotExist:
+            client = MetadataClientModel(hostname=client_name)
+            client.save()
+    except:
+        print("Failed to migrate client %s" % (client))
+        return False
+
+    try:
+        cgroups = []
+        for xgroup in xclient.findall('Group'):
+            group_name = xgroup.get('name')
+            cgroups.append(group_name)
+            try:
+                group = ProbesGroupsModel.objects.get(hostname=client_name, group=group_name)
+            except ProbesGroupsModel.DoesNotExist:
+                group = ProbesGroupsModel(hostname=client_name, group=group_name)
+                group.save()
+
+        ProbesGroupsModel.objects.filter(
+            hostname=client.hostname).exclude(
+            group__in=cgroups).delete()
+
+    except:
+        print("Failed to migrate groups")
+        return False
+    return True
+
+def main():
+    opts = dict(repo=Bcfg2.Options.SERVER_REPOSITORY)
+    setup = Bcfg2.Options.OptionParser(opts)
+    setup.parse(sys.argv[1:])
+
+    probefile = os.path.join(setup['repo'], 'Probes', "probed.xml")
+
+    try:
+        xdata = lxml.etree.parse(probefile)
+    except lxml.etree.XMLSyntaxError:
+        err = sys.exc_info()[1]
+        print("Could not parse %s, skipping: %s" % (probefile, err))
+    
+    for xclient in xdata.findall('Client'):
+        print "Migrating Metadata and Probe groups for %s" % xclient.get('name')
+        migrate(xclient)
+
+if __name__ == '__main__':
+    sys.exit(main())
-- 
cgit v1.2.3-1-g7c22


From bc925f4059496894c951b0ddb86cb5ae774edd1e Mon Sep 17 00:00:00 2001
From: Jason Kincl <kincljc@ornl.gov>
Date: Wed, 22 May 2013 11:59:16 -0400
Subject: adding doc

---
 tools/upgrade/1.3/README                        | 5 +++++
 tools/upgrade/1.3/migrate_probe_groups_to_db.py | 5 +++++
 2 files changed, 10 insertions(+)

(limited to 'tools')

diff --git a/tools/upgrade/1.3/README b/tools/upgrade/1.3/README
index 2831d8f00..1a919f869 100644
--- a/tools/upgrade/1.3/README
+++ b/tools/upgrade/1.3/README
@@ -19,3 +19,8 @@ migrate_dbstats.py
 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)
+
+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. 
diff --git a/tools/upgrade/1.3/migrate_probe_groups_to_db.py b/tools/upgrade/1.3/migrate_probe_groups_to_db.py
index 540d10505..73339e787 100755
--- a/tools/upgrade/1.3/migrate_probe_groups_to_db.py
+++ b/tools/upgrade/1.3/migrate_probe_groups_to_db.py
@@ -1,4 +1,7 @@
 #!/bin/env python
+""" 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. """
 
 import os
 os.environ['DJANGO_SETTINGS_MODULE'] = 'Bcfg2.settings'
@@ -11,6 +14,7 @@ from Bcfg2.Server.Plugins.Metadata import MetadataClientModel
 from Bcfg2.Server.Plugins.Probes import ProbesGroupsModel
 
 def migrate(xclient):
+    """ Helper to do the migration given a <Client/> XML element """
     client_name = xclient.get('name')
     try:
         try:
@@ -43,6 +47,7 @@ def migrate(xclient):
     return True
 
 def main():
+    """ Main """
     opts = dict(repo=Bcfg2.Options.SERVER_REPOSITORY)
     setup = Bcfg2.Options.OptionParser(opts)
     setup.parse(sys.argv[1:])
-- 
cgit v1.2.3-1-g7c22