Add tile cookbook
[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 mapnik2
7 import os
8 import shutil
9 import sys
10 import tempfile
11 import resource
12
13 # Limit maximum CPU time
14 # The Postscript output format can sometimes take hours
15 resource.setrlimit(resource.RLIMIT_CPU,(180,180))
16
17 # Limit memory usage
18 # Some odd requests can cause extreme memory usage
19 resource.setrlimit(resource.RLIMIT_AS,(4000000000, 4000000000))
20
21 # Routine to output HTTP headers
22 def output_headers(content_type, filename = "", length = 0):
23   print "Content-Type: %s" % content_type
24   if filename:
25     print "Content-Disposition: attachment; filename=\"%s\"" % filename
26   if length:
27     print "Content-Length: %d" % length
28   print ""
29
30 # Routine to output the contents of a file
31 def output_file(file):
32   file.seek(0)
33   shutil.copyfileobj(file, sys.stdout)
34
35 # Routine to get the size of a file
36 def file_size(file):
37   return os.fstat(file.fileno()).st_size
38
39 # Routine to report an error
40 def output_error(message):
41   output_headers("text/html")
42   print "<html>"
43   print "<head>"
44   print "<title>Error</title>"
45   print "</head>"
46   print "<body>"
47   print "<h1>Error</h1>"
48   print "<p>%s</p>" % message
49   print "</body>"
50   print "</html>"
51
52 # Parse CGI parameters
53 form = cgi.FieldStorage()
54
55 # Make sure we have a user agent
56 if not os.environ.has_key('HTTP_USER_AGENT'):
57   os.environ['HTTP_USER_AGENT'] = 'NONE'
58
59 # Get the load average
60 loadavg = float(open("/proc/loadavg").readline().split(" ")[0])
61
62 # Process the request
63 if loadavg > 35.0:
64   # Abort if the load average on the machine is too high
65   print "Status: 503 Service Unavailable"
66   output_error("The load average on the server is too high at the moment. Please wait a few minutes before trying again.")
67 <% @blocks["user_agents"].each do |user_agent| -%>
68 elif os.environ['HTTP_USER_AGENT'] == '<%= user_agent %>':
69   # Block scraper
70   print "Status: 503 Service Unavailable"
71   output_error("The load average on the server is too high at the moment. Please wait a few minutes before trying again.")
72 <% end -%>
73 elif not form.has_key("bbox"):
74   # No bounding box specified
75   output_error("No bounding box specified")
76 elif not form.has_key("scale"):
77   # No scale specified
78   output_error("No scale specified")
79 elif not form.has_key("format"):
80   # No format specified
81   output_error("No format specified")
82 else:
83   # Create projection object
84   prj = mapnik2.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");
85
86   # Get the bounds of the area to render
87   bbox = [float(x) for x in form.getvalue("bbox").split(",")]
88
89   if bbox[0] >= bbox[2] or bbox[1] >= bbox[3]:
90     # Bogus bounding box
91     output_error("Invalid bounding box")
92   else:
93     # Project the bounds to the map projection
94     bbox = mapnik2.forward_(mapnik2.Box2d(*bbox), prj)
95
96     # Calculate the size of the final rendered image
97     scale = float(form.getvalue("scale"))
98     width = int(bbox.width() / scale / 0.00028)
99     height = int(bbox.height() / scale / 0.00028)
100
101     # Limit the size of map we are prepared to produce
102     if width * height > 4000000:
103       # Map is too large (limit is approximately A2 size)
104       output_error("Map too large")
105     else:
106       # Create map
107       map = mapnik2.Map(width, height)
108
109       # Load map configuration
110       mapnik2.load_map(map, "/home/jburgess/live/osm2.xml")
111
112       # Zoom the map to the bounding box
113       map.zoom_to_box(bbox)
114
115       # Render the map
116       if form.getvalue("format") == "png":
117         image = mapnik2.Image(map.width, map.height)
118         mapnik2.render(map, image)
119         png = image.tostring("png")
120         output_headers("image/png", "map.png", len(png))
121         sys.stdout.write(png)
122       elif form.getvalue("format") == "jpeg":
123         image = mapnik2.Image(map.width, map.height)
124         mapnik2.render(map, image)
125         jpeg = image.tostring("jpeg")
126         output_headers("image/jpeg", "map.jpg", len(jpeg))
127         sys.stdout.write(jpeg)
128       elif form.getvalue("format") == "svg":
129         file = tempfile.NamedTemporaryFile()
130         surface = cairo.SVGSurface(file.name, map.width, map.height)
131         mapnik2.render(map, surface)
132         surface.finish()
133         output_headers("image/svg+xml", "map.svg", file_size(file))
134         output_file(file)
135       elif form.getvalue("format") == "pdf":
136         file = tempfile.NamedTemporaryFile()
137         surface = cairo.PDFSurface(file.name, map.width, map.height)
138         mapnik2.render(map, surface)
139         surface.finish()
140         output_headers("application/pdf", "map.pdf", file_size(file))
141         output_file(file)
142       elif form.getvalue("format") == "ps":
143         file = tempfile.NamedTemporaryFile()
144         surface = cairo.PSSurface(file.name, map.width, map.height)
145         mapnik2.render(map, surface)
146         surface.finish()
147         output_headers("application/postscript", "map.ps", file_size(file))
148         output_file(file)
149       else:
150         output_error("Unknown format '%s'" % form.getvalue("format"))