5e4b97414b9c843390047bc85e06eccf4d56074e
[chef.git] / cookbooks / tile / templates / default / export.erb
1 #!/usr/bin/python -u
2 # -*- coding: utf-8 -*-
3
4 import cairo
5 import cgi
6 import mapnik
7 import os
8 import shutil
9 import sys
10 import tempfile
11 import resource
12 import signal
13
14 # Limit maximum CPU time
15 # The Postscript output format can sometimes take hours
16 resource.setrlimit(resource.RLIMIT_CPU,(180,180))
17
18 # Limit memory usage
19 # Some odd requests can cause extreme memory usage
20 resource.setrlimit(resource.RLIMIT_AS,(4000000000, 4000000000))
21
22 # Routine to output HTTP headers
23 def output_headers(content_type, filename = "", length = 0):
24   print "Content-Type: %s" % content_type
25   if filename:
26     print "Content-Disposition: attachment; filename=\"%s\"" % filename
27   if length:
28     print "Content-Length: %d" % length
29   print ""
30
31 # Routine to output the contents of a file
32 def output_file(file):
33   file.seek(0)
34   shutil.copyfileobj(file, sys.stdout)
35
36 # Routine to get the size of a file
37 def file_size(file):
38   return os.fstat(file.fileno()).st_size
39
40 # Routine to report an error
41 def output_error(message, status = "400 Bad Request"):
42   print "Status: %s" % status
43   output_headers("text/html")
44   print "<html>"
45   print "<head>"
46   print "<title>Error</title>"
47   print "</head>"
48   print "<body>"
49   print "<h1>Error</h1>"
50   print "<p>%s</p>" % message
51   print "</body>"
52   print "</html>"
53
54 # Parse CGI parameters
55 form = cgi.FieldStorage()
56
57 # Make sure we have a user agent
58 if not os.environ.has_key('HTTP_USER_AGENT'):
59   os.environ['HTTP_USER_AGENT'] = 'NONE'
60
61 # Make sure we have a referer
62 if not os.environ.has_key('HTTP_REFERER'):
63   os.environ['HTTP_REFERER'] = 'NONE'
64
65 # Get the load average
66 cputimes = [float(n) for n in open("/proc/stat").readline().rstrip().split()[1:-1]]
67 idletime = cputimes[3] / sum(cputimes)
68
69 # Process the request
70 if idletime < 0.2:
71   # Abort if the CPU idle time on the machine is too low
72   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
73 elif form.getvalue("scale") == "3600":
74   # Block scraper
75   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
76 <% @blocks["user_agents"].each do |user_agent| -%>
77 elif os.environ['HTTP_USER_AGENT'] == '<%= user_agent %>':
78   # Block scraper
79   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
80 <% end -%>
81 <% @blocks["referers"].each do |referer| -%>
82 elif os.environ['HTTP_REFERER'] == '<%= referer %>':
83   # Block scraper
84   output_error("The server is too busy at the moment. Please wait a few minutes before trying again.", "503 Service Unavailable")
85 <% end -%>
86 elif not form.has_key("bbox"):
87   # No bounding box specified
88   output_error("No bounding box specified")
89 elif not form.has_key("scale"):
90   # No scale specified
91   output_error("No scale specified")
92 elif not form.has_key("format"):
93   # No format specified
94   output_error("No format specified")
95 else:
96   # Create projection object
97   prj = mapnik.Projection("+proj=merc +a=6378137 +b=6378137 +lat_ts=0.0 +lon_0=0.0 +x_0=0.0 +y_0=0 +k=1.0 +units=m +nadgrids=@null +no_defs +over");
98
99   # Get the bounds of the area to render
100   bbox = [float(x) for x in form.getvalue("bbox").split(",")]
101
102   if bbox[0] >= bbox[2] or bbox[1] >= bbox[3]:
103     # Bogus bounding box
104     output_error("Invalid bounding box")
105   else:
106     # Project the bounds to the map projection
107     bbox = mapnik.forward_(mapnik.Box2d(*bbox), prj)
108
109     # Get the style to use
110     style = form.getvalue("style", "default")
111
112     # Calculate the size of the final rendered image
113     scale = float(form.getvalue("scale"))
114     width = int(bbox.width() / scale / 0.00028)
115     height = int(bbox.height() / scale / 0.00028)
116
117     # Limit the size of map we are prepared to produce
118     if width * height > 4000000:
119       # Map is too large (limit is approximately A2 size)
120       output_error("Map too large")
121     else:
122       # Create map
123       map = mapnik.Map(width, height)
124
125       # Load map configuration
126       mapnik.load_map(map, "/srv/tile.openstreetmap.org/styles/%s/project.xml" % style)
127
128       # Zoom the map to the bounding box
129       map.zoom_to_box(bbox)
130
131       # Fork so that we can handle crashes rendering the map
132       pid = os.fork()
133
134       # Render the map
135       if pid == 0:
136         if form.getvalue("format") == "png":
137           image = mapnik.Image(map.width, map.height)
138           mapnik.render(map, image)
139           png = image.tostring("png")
140           output_headers("image/png", "map.png", len(png))
141           sys.stdout.write(png)
142         elif form.getvalue("format") == "jpeg":
143           image = mapnik.Image(map.width, map.height)
144           mapnik.render(map, image)
145           jpeg = image.tostring("jpeg")
146           output_headers("image/jpeg", "map.jpg", len(jpeg))
147           sys.stdout.write(jpeg)
148         elif form.getvalue("format") == "svg":
149           file = tempfile.NamedTemporaryFile(prefix = "export")
150           surface = cairo.SVGSurface(file.name, map.width, map.height)
151           mapnik.render(map, surface)
152           surface.finish()
153           output_headers("image/svg+xml", "map.svg", file_size(file))
154           output_file(file)
155         elif form.getvalue("format") == "pdf":
156           file = tempfile.NamedTemporaryFile(prefix = "export")
157           surface = cairo.PDFSurface(file.name, map.width, map.height)
158           mapnik.render(map, surface)
159           surface.finish()
160           output_headers("application/pdf", "map.pdf", file_size(file))
161           output_file(file)
162         elif form.getvalue("format") == "ps":
163           file = tempfile.NamedTemporaryFile(prefix = "export")
164           surface = cairo.PSSurface(file.name, map.width, map.height)
165           mapnik.render(map, surface)
166           surface.finish()
167           output_headers("application/postscript", "map.ps", file_size(file))
168           output_file(file)
169         else:
170           output_error("Unknown format '%s'" % form.getvalue("format"))
171       else:
172         pid, status = os.waitpid(pid, 0)
173         if status & 0xff == signal.SIGXCPU:
174           output_error("CPU time limit exceeded", "509 Resource Limit Exceeded")
175         elif status & 0xff == signal.SIGSEGV:
176           output_error("Memory limit exceeded", "509 Resource Limit Exceeded")
177         elif status != 0:
178           output_error("Internal server error", "500 Internal Server Error")