include opstates in ad rspec. include ssh-user in manifest rspec

[sfa.git] / sfa / rspecs / elements / v3 / services.py

diff --git a/sfa/rspecs/elements/v3/services.py b/sfa/rspecs/elements/v3/services.py
new file mode 100644 (file)
index 0000000..b4ac1ca
--- /dev/null
+++ b/sfa/rspecs/elements/v3/services.py
@@ -0,0 +1,60 @@
+from sfa.rspecs.elements.element import Element  
+from sfa.rspecs.elements.execute import Execute  
+from sfa.rspecs.elements.install import Install  
+from sfa.rspecs.elements.services import Services  
+from sfa.rspecs.elements.login import Login
+
+class Services:
+    @staticmethod
+    def add_services(xml, services):
+        if not services:
+            return 
+        for service in services:
+            service_elem = xml.add_element('services')
+            child_elements = {'install': Install.fields,
+                              'execute': Execute.fields,
+                              'login': Login.fields}
+            for (name, fields) in child_elements.items():
+                child = service.get(name)
+                if not child: 
+                    continue
+                if isinstance(child, dict):
+                    service_elem.add_instance(name, child, fields)
+                elif isinstance(child, list):
+                    for obj in child:
+                        service_elem.add_instance(name, obj, fields)
+
+            # add ssh_users
+            if service['services_user']:
+                for ssh_user in service['services_user']: 
+                    ssh_user_elem = service_elem.add_element('{%s}services_user' % xml.namespaces['ssh-user'],
+                                                             login=ssh_user['login'],
+                                                             user_urn=ssh_user['user_urn'])
+                    for key in ssh_user['keys']:
+                        pkey_elem = ssh_user_elem.add_element('{%s}public_key' % xml.namespaces['ssh-user'])
+                        pkey_elem.element.text=key
+                          
+              
+    @staticmethod
+    def get_services(xml):
+        services = []
+        for services_elem in xml.xpath('./default:services | ./services'):
+            service = Services(services_elem.attrib, services_elem)
+            # get install 
+            install_elems = xml.xpath('./default:install | ./install')
+            service['install'] = [install_elem.get_instance(Install) for install_elem in install_elems]
+            # get execute
+            execute_elems = xml.xpath('./default:execute | ./execute')
+            service['execute'] = [execute_elem.get_instance(Execute) for execute_elem in execute_elems]
+            # get login
+            login_elems = xml.xpath('./default:login | ./login')
+            service['login'] = [login_elem.get_instance(Login) for login_elem in login_elems]
+            
+            ssh_user_elems = xml.xpath('./ssh-user:service_user | ./service_user')
+            service_users = []    
+            for ssh_user_elem in ssh_user_elems:
+                service_user = ssh_user_elem.get_instance(None, fields=['login', 'user_urn'])
+            service['services_user'] = service_user
+            services.append(service)  
+        return services
+