serve/main.py at master · Tagtoo/serve · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# -*- coding: utf8 -*-
import webapp2
from google.appengine.ext import blobstore
from google.appengine.api import images
from google.appengine.ext.webapp import blobstore_handlers
import urllib
import serve_static.edge_cache as static
from datetime import timedelta

class ServeHandler(blobstore_handlers.BlobstoreDownloadHandler):
    @static.expires(expire_interval=timedelta(days=30))
    def get(self, resource):
        resource = str(urllib.unquote(resource))
        blob_info = blobstore.BlobInfo.get(resource)
        self.send_blob(blob_info)


upload_form = u"""
<html>
    <body>
        <form action="%s" method="POST" enctype="multipart/form-data">
        上傳檔案: <input type="file" name="file"><br>
        <input type="submit" name="submit" value="上傳">
        </form>
    </body>
</html>"""


class AddImagePage(webapp2.RequestHandler):
    def get(self):
        upload_url = blobstore.create_upload_url('/serve/_ah/upload_image')
        self.response.out.write(upload_form % upload_url)


class AddFilePage(webapp2.RequestHandler):
    def get(self):
        upload_url = blobstore.create_upload_url('/serve/_ah/upload_file')
        self.response.out.write(upload_form % upload_url)


class UploadFileHandler(blobstore_handlers.BlobstoreUploadHandler):
    def post(self):
        upload_files = self.get_uploads('file')  # 'file' is file upload field in the form
        blob_info = upload_files[0]

        self.response.out.write(str('http://%s/serve/static/%s' % (self.request.host, blob_info.key())))


class UploadImageHandler(blobstore_handlers.BlobstoreUploadHandler):
    def post(self):
        upload_files = self.get_uploads('file')
        blob_info = upload_files[0]

        serve_url = images.get_serving_url(blob_info.key())

        self.response.out.write(serve_url)


app = webapp2.WSGIApplication([
    (r'/serve/add_image', AddImagePage),
    (r'/serve/add_file', AddFilePage),
    (r'/serve/_ah/upload_image', UploadImageHandler),
    (r'/serve/_ah/upload_file', UploadFileHandler),

    (r'/serve/static/([^/]+)?', ServeHandler),
])