Added some additional policy methods

Author: Mierdin

functions/newfunctions_ucs.py

@@ -255,9 +255,9 @@ def createWwpnPool(self, fabric, blockstart, blockend):
     def removeWwpnPool(self, mo):
         pass
 
-    ##############
-    #  POLICIES  #
-    ##############
+    #####################
+    #  GLOBAL POLICIES  #
+    #####################
 
     def setPowerPolicy(self, redundancy):
         try:
@@ -286,6 +286,10 @@ def setChassisDiscoveryPolicy(self, links):
         except UcsException:
             print "Error setting Chassis Discovery policy"
 
+    ##################
+    #  QOS POLICIES  #
+    ##################
+
     def setGlobalQosPolicy(self, qosconfig):
         try:
 
@@ -395,3 +399,102 @@ def createQosPolicy(self, classname, hostcontrol):
                 }, True)
         except UcsException:
             print "QoS Policy already exists"     
+
+    ###########################
+    #  ORG-SPECIFIC POLICIES  #
+    ###########################
+
+    def getLocalDiskPolicy(self, name):
+        pass
+
+    def createLocalDiskPolicy(self, name, mode):
+        try:
+            self.handle.AddManagedObject(self.org, "storageLocalDiskConfigPolicy", 
+                {       
+                    "Name":name, 
+                    "Descr":name, 
+                    "PolicyOwner":"local", 
+                    "ProtectConfig":"yes", 
+                    "Dn":self.orgNameDN + "local-disk-config-" + name, 
+                    "Mode":mode, 
+                    "FlexFlashState":"disable", 
+                    "FlexFlashRAIDReportingState":"disable"
+                })
+        except UcsException:
+            print "Local Disk Configuration Policy already exists"
+
+    def removeLocalDiskPolicy(self, name):
+        pass
+
+    def getHostFWPackage(self, name):
+        pass
+
+    def createHostFWPackage(self, name):
+        try:
+            self.handle.AddManagedObject(self.org, "firmwareComputeHostPack", 
+                {
+                    "Name":name, 
+                    "BladeBundleVersion":"", 
+                    "RackBundleVersion":"", 
+                    "PolicyOwner":"local", 
+                    "Dn":self.orgNameDN + "fw-host-pack-" + name, 
+                    "Mode":"staged", 
+                    "IgnoreCompCheck":"yes", 
+                    "StageSize":"0", 
+                    "Descr":"Host Firmware Package", 
+                    "UpdateTrigger":"immediate"
+                })
+        except UcsException:
+            print "Host Firmware Package already exists"
+
+    def removeHostFWPackage(self, name):
+        pass
+
+    def getMaintPolicy(self, name):
+        pass
+
+    def createMaintPolicy(self, name, policy):
+        try:
+            self.handle.AddManagedObject(self.org, "lsmaintMaintPolicy", 
+                {
+                    "Descr":name, 
+                    "SchedName":"", 
+                    "Name":name, 
+                    "Dn":self.orgNameDN + "maint-" + name, 
+                    "PolicyOwner":"local", 
+                    "UptimeDisr":policy
+                })
+        except UcsException:
+            print "Maintenance Policy already exists"
+
+    def removeMaintPolicy(self, name):
+        pass
+
+    def createNetControlPolicy(self, name):
+        try:
+            mo = self.handle.AddManagedObject(self.org, "nwctrlDefinition", 
+                {
+                    "Name":name, 
+                    "Cdp":"enabled", 
+                    "Dn":self.orgNameDN + "nwctrl-" + name, 
+                    "PolicyOwner":"local", 
+                    "MacRegisterMode":"only-native-vlan", 
+                    "UplinkFailAction":"link-down", 
+                    "Descr":name
+                })
+            mo_1 = self.handle.AddManagedObject(mo, "dpsecMac", 
+                {
+                    "Name":"", 
+                    "Dn":self.orgNameDN + "nwctrl-" + name + "/mac-sec", 
+                    "Forge":"allow", 
+                    "PolicyOwner":"local", 
+                    "Descr":""
+                }, True)
+        except UcsException:
+            print "Network Control Policy already exists"
+
+    #TODO: Tabling this for now. This is a complicated thing to do.
+    def createBootPolicy(name):
+        pass
+
+    

workers/ucs.py

@@ -1,6 +1,10 @@
 #!/usr/bin/env python
 """ UCS worker class for pyflex
 
+    In the future, this will really be the core renderer. All of the 
+    work of deleting config that shouldn't be there, or creating it
+    when it should, will be handled here.
+
 """
 from worker import FlexWorker
 #from functions.functions_ucs import UcsFunctions
@@ -92,11 +96,21 @@ def startworker(self):
         for fabric in wwpnpools: #TODO: This loop is here for the future, but obviously since the name is statically set, this only works with a single pool per fabric, currently.
             newfxns.createWwpnPool(fabric, wwpnpools[fabric]['blockbegin'], wwpnpools[fabric]['blockend'])
 
-
+        """ GLOBAL POLICIES """
 
         newfxns.setPowerPolicy("grid")
         newfxns.setChassisDiscoveryPolicy(str(self.config['ucs']['links']))
-        newfxns.setGlobalQosPolicy(self.config['qos'])
 
+        """ QOS """
+
+        newfxns.setGlobalQosPolicy(self.config['qos'])
         for classname, hostcontrol in self.config['qos']['classes'].iteritems():
-            newfxns.createQosPolicy(classname, hostcontrol)
+            newfxns.createQosPolicy(classname, hostcontrol)
+
+        """ ORG-SPECIFIC POLICIES """
+
+        newfxns.createLocalDiskPolicy("NO_LOCAL", "no-local-storage")
+        newfxns.createLocalDiskPolicy("RAID1", "raid-mirrored")
+        newfxns.createHostFWPackage("HOST_FW_PKG")
+        newfxns.createMaintPolicy("MAINT_USERACK", "user-ack")
+        newfxns.createNetControlPolicy("NTKCTRL-CDP")