| OLD | NEW |
| (Empty) |
| 1 | |
| 2 # Copyright (c) 2001-2004 Twisted Matrix Laboratories. | |
| 3 # See LICENSE for details. | |
| 4 | |
| 5 | |
| 6 """I contain PythonScript, which is a very simple python script resource. | |
| 7 """ | |
| 8 | |
| 9 import server | |
| 10 import resource | |
| 11 import html | |
| 12 import error | |
| 13 | |
| 14 try: | |
| 15 import cStringIO as StringIO | |
| 16 except ImportError: | |
| 17 import StringIO | |
| 18 | |
| 19 from twisted.web import http | |
| 20 from twisted import copyright | |
| 21 import traceback | |
| 22 import os | |
| 23 from twisted.web import resource | |
| 24 from twisted.web import static | |
| 25 | |
| 26 rpyNoResource = """<p>You forgot to assign to the variable "resource" in your sc
ript. For example:</p> | |
| 27 <pre> | |
| 28 # MyCoolWebApp.rpy | |
| 29 | |
| 30 import mygreatresource | |
| 31 | |
| 32 resource = mygreatresource.MyGreatResource() | |
| 33 </pre> | |
| 34 """ | |
| 35 | |
| 36 class AlreadyCached(Exception): | |
| 37 """This exception is raised when a path has already been cached. | |
| 38 """ | |
| 39 | |
| 40 class CacheScanner: | |
| 41 def __init__(self, path, registry): | |
| 42 self.path = path | |
| 43 self.registry = registry | |
| 44 self.doCache = 0 | |
| 45 | |
| 46 def cache(self): | |
| 47 c = self.registry.getCachedPath(self.path) | |
| 48 if c is not None: | |
| 49 raise AlreadyCached(c) | |
| 50 self.recache() | |
| 51 | |
| 52 def recache(self): | |
| 53 self.doCache = 1 | |
| 54 | |
| 55 noRsrc = error.ErrorPage(500, "Whoops! Internal Error", rpyNoResource) | |
| 56 | |
| 57 def ResourceScript(path, registry): | |
| 58 """ | |
| 59 I am a normal py file which must define a 'resource' global, which should | |
| 60 be an instance of (a subclass of) web.resource.Resource; it will be | |
| 61 renderred. | |
| 62 """ | |
| 63 cs = CacheScanner(path, registry) | |
| 64 glob = {'__file__': path, | |
| 65 'resource': noRsrc, | |
| 66 'registry': registry, | |
| 67 'cache': cs.cache, | |
| 68 'recache': cs.recache} | |
| 69 try: | |
| 70 execfile(path, glob, glob) | |
| 71 except AlreadyCached, ac: | |
| 72 return ac.args[0] | |
| 73 rsrc = glob['resource'] | |
| 74 if cs.doCache and rsrc is not noRsrc: | |
| 75 registry.cachePath(path, rsrc) | |
| 76 return rsrc | |
| 77 | |
| 78 def ResourceTemplate(path, registry): | |
| 79 from quixote import ptl_compile | |
| 80 | |
| 81 glob = {'__file__': path, | |
| 82 'resource': error.ErrorPage(500, "Whoops! Internal Error", | |
| 83 rpyNoResource), | |
| 84 'registry': registry} | |
| 85 | |
| 86 e = ptl_compile.compile_template(open(path), path) | |
| 87 exec e in glob | |
| 88 return glob['resource'] | |
| 89 | |
| 90 | |
| 91 class ResourceScriptWrapper(resource.Resource): | |
| 92 | |
| 93 def __init__(self, path, registry=None): | |
| 94 resource.Resource.__init__(self) | |
| 95 self.path = path | |
| 96 self.registry = registry or static.Registry() | |
| 97 | |
| 98 def render(self, request): | |
| 99 res = ResourceScript(self.path, self.registry) | |
| 100 return res.render(request) | |
| 101 | |
| 102 def getChildWithDefault(self, path, request): | |
| 103 res = ResourceScript(self.path, self.registry) | |
| 104 return res.getChildWithDefault(path, request) | |
| 105 | |
| 106 | |
| 107 | |
| 108 class ResourceScriptDirectory(resource.Resource): | |
| 109 def __init__(self, pathname, registry=None): | |
| 110 resource.Resource.__init__(self) | |
| 111 self.path = pathname | |
| 112 self.registry = registry or static.Registry() | |
| 113 | |
| 114 def getChild(self, path, request): | |
| 115 fn = os.path.join(self.path, path) | |
| 116 | |
| 117 if os.path.isdir(fn): | |
| 118 return ResourceScriptDirectory(fn, self.registry) | |
| 119 if os.path.exists(fn): | |
| 120 return ResourceScript(fn, self.registry) | |
| 121 return error.NoResource() | |
| 122 | |
| 123 def render(self, request): | |
| 124 return error.NoResource().render(request) | |
| 125 | |
| 126 | |
| 127 class PythonScript(resource.Resource): | |
| 128 """I am an extremely simple dynamic resource; an embedded python script. | |
| 129 | |
| 130 This will execute a file (usually of the extension '.epy') as Python code, | |
| 131 internal to the webserver. | |
| 132 """ | |
| 133 isLeaf = 1 | |
| 134 def __init__(self, filename, registry): | |
| 135 """Initialize me with a script name. | |
| 136 """ | |
| 137 self.filename = filename | |
| 138 self.registry = registry | |
| 139 | |
| 140 def render(self, request): | |
| 141 """Render me to a web client. | |
| 142 | |
| 143 Load my file, execute it in a special namespace (with 'request' and | |
| 144 '__file__' global vars) and finish the request. Output to the web-page | |
| 145 will NOT be handled with print - standard output goes to the log - but | |
| 146 with request.write. | |
| 147 """ | |
| 148 request.setHeader("x-powered-by","Twisted/%s" % copyright.version) | |
| 149 namespace = {'request': request, | |
| 150 '__file__': self.filename, | |
| 151 'registry': self.registry} | |
| 152 try: | |
| 153 execfile(self.filename, namespace, namespace) | |
| 154 except IOError, e: | |
| 155 if e.errno == 2: #file not found | |
| 156 request.setResponseCode(http.NOT_FOUND) | |
| 157 request.write(error.NoResource("File not found.").render(request
)) | |
| 158 except: | |
| 159 io = StringIO.StringIO() | |
| 160 traceback.print_exc(file=io) | |
| 161 request.write(html.PRE(io.getvalue())) | |
| 162 request.finish() | |
| 163 return server.NOT_DONE_YET | |
| OLD | NEW |