1 #!/usr/bin/env python 2 3 "A WebStack application for a system configurator." 4 5 import WebStack.Generic 6 import XSLForms.Fields 7 import XSLForms.Output 8 import XSLOutput 9 import libxml2dom 10 import os 11 12 # NOTE: Move these functions into a common utility library. 13 14 def add_elements(positions, element_name, element_parent_name=None): 15 if not positions: 16 return 17 for position in positions: 18 if element_parent_name: 19 parent_elements = position.xpath(element_parent_name) 20 if not parent_elements: 21 parent_element = position.ownerDocument.createElementNS(None, element_parent_name) 22 position.appendChild(parent_element) 23 else: 24 parent_element = parent_elements[0] 25 else: 26 parent_element = position 27 parent_element.appendChild(position.ownerDocument.createElementNS(None, element_name)) 28 29 def remove_elements(positions): 30 if not positions: 31 return 32 for position in positions: 33 position.parentNode.removeChild(position) 34 35 class ConfiguratorResource: 36 37 "A resource providing a system configurator." 38 39 resource_dir = os.path.join(os.path.split(__file__)[0], "Resources") 40 encoding = "utf-8" 41 42 def respond(self, trans): 43 44 global XSLForms # NOTE: Strangely required to avoid UnboundLocalError! 45 46 # Only obtain field information according to the stated method. 47 48 method = trans.get_request_method() 49 50 if method == "GET": 51 fields = XSLForms.Fields.Fields(encoding="iso-8859-1", values_are_lists=1) 52 parameters = trans.get_fields_from_path() 53 documents = fields.make_documents(parameters.items()) 54 elif method == "POST": 55 fields = XSLForms.Fields.Fields(encoding=self.encoding, values_are_lists=1) 56 parameters = trans.get_fields_from_body(self.encoding) 57 documents = fields.make_documents(parameters.items()) 58 else: 59 trans.set_response_code(405) 60 raise WebStack.Generic.EndOfResponse 61 62 # Ensure the presence of a document. 63 64 if documents.has_key("configuration"): 65 configuration = documents["configuration"] 66 else: 67 configuration = fields.new_instance("configuration") 68 69 # Add and remove elements according to the selectors found. 70 71 selectors = fields.get_selectors(parameters.items(), documents) 72 add_elements(selectors.get("add-memory-unit"), "memory-unit") 73 remove_elements(selectors.get("remove-memory-unit")) 74 add_elements(selectors.get("add-storage-unit"), "storage-unit") 75 remove_elements(selectors.get("remove-storage-unit")) 76 add_elements(selectors.get("add-hard-disk"), "hard-disk") 77 remove_elements(selectors.get("remove-hard-disk")) 78 79 # Send a response according to certain parameters. 80 # When exported, an XML version of the data is returned. 81 82 if parameters.has_key("export"): 83 trans.set_content_type(WebStack.Generic.ContentType("text/xml", self.encoding)) 84 libxml2dom.toStream(configuration, trans.get_response_stream(), trans.get_response_stream_encoding()) 85 86 # When not exported, the data is transformed to produce a normal Web 87 # page. 88 89 else: 90 91 # Transform, adding enumerations/ranges. 92 93 database_xsl = os.path.join(self.resource_dir, "config_database.xsl") 94 database_xml = os.path.join(self.resource_dir, "config_database.xml") 95 proc = XSLOutput.Processor([database_xsl], references={"database" : database_xml}) 96 configuration = proc.get_result(configuration) 97 98 # Start the response. 99 100 trans.set_content_type(WebStack.Generic.ContentType("text/html", self.encoding)) 101 102 # Ensure that an output stylesheet exists. 103 104 trans_xsl = os.path.join(self.resource_dir, "config_output.xsl") 105 template_xml = os.path.join(self.resource_dir, "config_template.xhtml") 106 107 if not os.path.exists(trans_xsl) or \ 108 os.path.getmtime(trans_xsl) < os.path.getmtime(template_xml): 109 110 import XSLForms.Prepare 111 template_xml = os.path.join(self.resource_dir, "config_template.xhtml") 112 XSLForms.Prepare.make_stylesheet(template_xml, trans_xsl) 113 114 # Complete the response. 115 116 proc = XSLOutput.Processor([trans_xsl]) 117 proc.send_output(trans.get_response_stream(), trans.get_response_stream_encoding(), 118 configuration) 119 120 # vim: tabstop=4 expandtab shiftwidth=4