1 #!/usr/bin/env python 2 3 "A WebStack application for a PEP 241 repository." 4 5 import WebStack.Generic 6 import XSLForms.Fields 7 import XSLForms.Output 8 import XSLOutput 9 import libxml2dom 10 import os 11 12 def add_elements(positions, element_name, element_parent_name=None): 13 if not positions: 14 return 15 for position in positions: 16 if element_parent_name: 17 parent_elements = position.xpath(element_parent_name) 18 if not parent_elements: 19 parent_element = position.ownerDocument.createElementNS(None, element_parent_name) 20 position.appendChild(parent_element) 21 else: 22 parent_element = parent_elements[0] 23 else: 24 parent_element = position 25 parent_element.appendChild(position.ownerDocument.createElementNS(None, element_name)) 26 27 def remove_elements(positions): 28 if not positions: 29 return 30 for position in positions: 31 position.parentNode.removeChild(position) 32 33 class PEP241Resource: 34 35 "A resource providing repository browsing." 36 37 resource_dir = os.path.join(os.path.split(__file__)[0], "Resources") 38 encoding = "utf-8" 39 40 def respond(self, trans): 41 42 # Produce stylesheet. 43 44 if trans.get_path_info().split("/")[-1] == "styles.css": 45 trans.set_content_type(WebStack.Generic.ContentType("text/css", self.encoding)) 46 out = trans.get_response_stream() 47 f = open(os.path.join(self.resource_dir, "styles.css"), "rb") 48 out.write(f.read()) 49 f.close() 50 raise WebStack.Generic.EndOfResponse 51 52 # Handle other requests. 53 54 global XSLForms # NOTE: Strangely required to avoid UnboundLocalError! 55 method = trans.get_request_method() 56 57 # Only obtain field information according to the stated method. 58 59 if method == "GET": 60 fields = XSLForms.Fields.Fields(encoding="iso-8859-1", values_are_lists=1) 61 parameters = trans.get_fields_from_path() 62 documents = fields.make_documents(parameters.items()) 63 elif method == "POST": 64 fields = XSLForms.Fields.Fields(encoding=self.encoding, values_are_lists=1) 65 parameters = trans.get_fields_from_body(self.encoding) 66 documents = fields.make_documents(parameters.items()) 67 else: 68 trans.set_response_code(405) 69 raise WebStack.Generic.EndOfResponse 70 71 # Ensure the presence of a document. 72 73 if documents.has_key("package"): 74 package = documents["package"] 75 else: 76 package = fields.new_instance("package") 77 78 # Add and remove elements according to the selectors found. 79 80 selectors = fields.get_selectors(parameters.items(), documents) 81 add_elements(selectors.get("add_platform"), "platform", "platforms") 82 remove_elements(selectors.get("remove_platform")) 83 add_elements(selectors.get("add_supported_platform"), "supported-platform", "supported-platforms") 84 remove_elements(selectors.get("remove_supported_platform")) 85 add_elements(selectors.get("add_keyword"), "keyword", "keywords") 86 remove_elements(selectors.get("remove_keyword")) 87 add_elements(selectors.get("add_author"), "author", "authors") 88 remove_elements(selectors.get("remove_author")) 89 add_elements(selectors.get("add_dependency"), "dependency", "dependencies") 90 remove_elements(selectors.get("remove_dependency")) 91 92 # Send a response according to certain parameters. 93 # When exported, an XML version of the data is returned. 94 95 if parameters.has_key("export"): 96 trans.set_content_type(WebStack.Generic.ContentType("text/xml", self.encoding)) 97 libxml2dom.toStream(package, trans.get_response_stream(), trans.get_response_stream_encoding()) 98 99 # When not exported, the data is transformed to produce a normal Web 100 # page. 101 102 else: 103 104 # Transform, adding enumerations/ranges. 105 106 categories_xsl = os.path.join(self.resource_dir, "pep241_categories.xsl") 107 categories_xml = os.path.join(self.resource_dir, "pep241_categories.xml") 108 proc = XSLOutput.Processor([categories_xsl], {"categories" : categories_xml}) 109 package = proc.get_result(package) 110 111 # Start the response. 112 113 trans.set_content_type(WebStack.Generic.ContentType("text/html", self.encoding)) 114 115 # Ensure that an output stylesheet exists. 116 117 trans_xsl = os.path.join(self.resource_dir, "pep241_output.xsl") 118 template_xml = os.path.join(self.resource_dir, "pep241_template.xhtml") 119 120 if not os.path.exists(trans_xsl) or \ 121 os.path.getmtime(trans_xsl) < os.path.getmtime(template_xml): 122 123 import XSLForms.Prepare 124 template_xml = os.path.join(self.resource_dir, "pep241_template.xhtml") 125 XSLForms.Prepare.make_stylesheet(template_xml, trans_xsl) 126 127 # Complete the response. 128 129 proc = XSLOutput.Processor([trans_xsl]) 130 proc.send_output(trans.get_response_stream(), trans.get_response_stream_encoding(), 131 package) 132 133 # vim: tabstop=4 expandtab shiftwidth=4