profile.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python


import webapp2, jinja2, os, db, base
from webapp2_extras import i18n
from google.appengine.api import images

JINJA_ENVIRONMENT = jinja2.Environment(
    loader=jinja2.FileSystemLoader(os.path.dirname(__file__)),
    extensions=['jinja2.ext.i18n', 'jinja2.ext.autoescape'],
    autoescape=True)

JINJA_ENVIRONMENT.install_gettext_translations(i18n)

class ProfileHandler(base.BaseHandler):
    def get(self, teachId):
        if not self.loggedIn():
            self.redirect("/")
            return
        values = self.getValues()
        i18n.get_i18n().set_locale(self.getLanguage())

        teacher = db.Teacher.get_by_id(long(teachId))
        lang = self.request.get("lang", None)
        if lang is not None:
            teacher.setLanguage(lang)
            referrer = self.request.headers.get('referer')
            if referrer:
                return self.redirect(referrer)
            return self.redirect("/")

        if not teachId or not teacher:
            self.redirect("/")
        else:

            own = teachId == str(self.getUserId())
            values["own"] = own
            values["teacher"] = teacher
            values["username"] = self.getUserName()
            values["subjects"] = ", ".join([s.name for s in db.Subject.getSubjectsByTeacher(teacher.key)])

            template = JINJA_ENVIRONMENT.get_template('/view/profile/index.html')
            self.response.write(template.render(values))

    def post(self):
        if not self.loggedIn():
            self.redirect("/")
            return

        name = self.request.get("name", None)
        if name is not None:
            # Modify teacher
            db.Teacher.addOrUpdate(self.getEmail(), self.request.get("name", None))

            self.redirect("/profile/%s" % self.getUserId())

        try:
            avatar = self.request.get("img", None)
            if avatar is not None:
                #avatar = images.resize(avatar, 32, 32)
                teacher = db.Teacher.getByEmail(self.getEmail())
                teacher.setAvatar(avatar)
                return
        except:
            pass

class ProfileImageHandler(base.BaseHandler):
    def get(self):
        imgId = self.request.get('id')

        if imgId != "":
            teacher = db.Teacher.get_by_id(long(imgId))

            if teacher is not None and teacher.avatar:
                self.response.headers['Content-Type'] = 'image/png'
                self.response.out.write(teacher.avatar)
                return

        self.redirect("/img/user.png")

app = webapp2.WSGIApplication([
    ('/profile/?([0-9]*)', ProfileHandler),
    ('/profile/modify', ProfileHandler),
    ('/profile/img', ProfileImageHandler)
], debug=True)