sample-apps/python/images-api-in-action/main.py at selenium · AppScale/sample-apps · 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import re
import wsgiref.handlers
from google.appengine.api import users
from google.appengine.ext import db
from google.appengine.ext import webapp
from google.appengine.ext.webapp import template
from google.appengine.api import urlfetch
from google.appengine.api import images

class Uploaded(db.Model):
  author = db.UserProperty()
  image = db.BlobProperty()

class UploadImage(webapp.RequestHandler):
  def get(self):
    user = users.get_current_user()
    if user is not None:
      response = "You are logged in as " + user.nickname() + ".<br />"
      response += "Is that <a href='" + users.create_logout_url("/upload") + "'>not you?</a><br /><br />"
    else:
      response = "You are not logged in yet.<br />"
      response += "Would you <a href='" + users.create_login_url("/upload") + "'>like to?</a><br /><br />"

    response += "Please upload an image:<br /><br />"
    response += "<form action='upload' enctype='multipart/form-data' method='post'>"
    response += "<div><label>From your Computer:</label><input type='file' name='hdd'/></div><br />"
    response += "<div><label>From the Interwebs:</label><input type='text' name='net'/></div><br />"
    response += "<div><input type='submit' value='Upload'></div>"
    response += "</form>"
    self.response.out.write(template.render('index.html',
                                            {'tool': 'Upload an image',
                                            'result': response}))
  def post(self):
    hdd = self.request.get("hdd")
    net = self.request.get("net")

    user = users.get_current_user()
    uploaded = Uploaded(key_name=str(user))
    response = "Uploaded an image anonymously.<br /><br />"
    if user:
      uploaded.author = user
      response = "Uploaded an image as " + user.nickname() + "<br /><br />"

    if hdd != "":
      uploaded.image = db.Blob(hdd)
      uploaded.put()
    elif net != "":
      data = urlfetch.fetch(net).content
      uploaded.image = db.Blob(data)
      uploaded.put()

    self.redirect("/")

class ManipulateImage(webapp.RequestHandler):
  def get(self):
    user = users.get_current_user()
    username = "anonymous"
    if user:
      username = user.nickname()

    uploaded = Uploaded.get_by_key_name(str(user))
    if uploaded is None:
      response = "No image uploaded yet."
    else:
      response = "Image for " + username + ":"
      response += "<div><img src='img?img_id=" + str(uploaded.key()) + "'></img></div><br /><br />"
      response += "Actions (choose one):<br /><br />"
      response += "<form action='/' method='post'>"
      response += "<div><input type='submit' name='rotater' value='Rotate Right'/></div><br />"
      response += "<div><input type='submit' name='rotatel' value='Rotate Left' /></div><br />"
      response += "<div><input type='submit' name='hflip' value='Horizontal Flip'/></div><br />"
      response += "<div><input type='submit' name='vflip' value='Vertical Flip' /></div><br />"
      response += "</form>"
    self.response.out.write(template.render('index.html',
                                            {'tool': 'Your Image',
                                             'result': response}))
  def post(self):
    user = users.get_current_user()
    uploaded = Uploaded.get_by_key_name(str(user))
    response = "baz"
    if uploaded is None:
      response = "No image uploaded yet."
    else:
      image = uploaded.image
      if self.request.get('hflip'):
        newimage = images.horizontal_flip(image)
      elif self.request.get('vflip'):
        newimage = images.vertical_flip(image)
      elif self.request.get('rotater'):
        newimage = images.rotate(image, 90)
      elif self.request.get('rotatel'):
        newimage = images.rotate(image, 270)

      uploaded.image = newimage
      uploaded.put()

    self.redirect("/")

class Image(webapp.RequestHandler):
    def get(self):
      uploaded = db.get(self.request.get("img_id"))
      if uploaded.image:
          self.response.headers['Content-Type'] = "image/png"
          self.response.out.write(uploaded.image)
      else:
          self.error(404)

def main():
  wsgiref.handlers.CGIHandler().run(webapp.WSGIApplication([
    ('/', ManipulateImage),
    ('/img', Image),
    ('/upload', UploadImage)
  ]))

if __name__ == '__main__':
  main()