views.py

# -*- coding: utf-8 -*-
from handlers import JsonHandler
import webapp2
from google.appengine.api import images
from webapp2_extras import sessions
import json
import csv
import copy
from models import data, in_bound, to_point, User, Message, Post, Store, Image
import random
import logging
from google.appengine.ext import blobstore
from google.appengine.ext.webapp import blobstore_handlers
from google.appengine.datastore.datastore_query import Cursor
static_map = "http://maps.googleapis.com/maps/api/staticmap?center={0},{1}&zoom=17&size=200x200&markers=color:red%7Ccolor:red%7Clabel:A%7C{0},{1}"
import urllib


dis = 500000 * 11 * 1e-6
class Explore(JsonHandler):
    def get(self, *args, **kwargs):
        tags = self.request.get('tags', '[]')
        tags = json.loads(tags)

        size = self.request.get_range('size') or 6

        latlng = self.request.get('latlng', '(25, 121)')
        latlng = to_point(latlng)
        ne = tuple([v+dis for v in latlng])
        sw = tuple([v-dis for v in latlng])
        result = filter(lambda d:in_bound(ne, sw, d['latlng']), data)
        random.shuffle(result)
        result = result[:size]
        result = copy.deepcopy(result)
        for r in result:
            r['img'] = random.choice([img for img in r['images'] if 'map' not in img])
            r['tags'] = r.pop('types')
            r['img_id'] = "img_"+r.pop('id')
            if r.get('images', False):
                del r['images']
                del r['photos']
                del r['place_id']
                del r['name']
                del r['address']
                del r['scope']

        self.JsonResponse(result)


class Recommend(JsonHandler):
    def get(self, *args, **kwargs):
        tags = self.request.get('tags', '[]')
        tags = json.loads(tags)

        size = self.request.get_range('size') or 6

        latlng = self.request.get('latlng', '(25, 121)')
        latlng = to_point(latlng)
        ne = tuple([v+dis for v in latlng])
        sw = tuple([v-dis for v in latlng])
        result = filter(lambda d:in_bound(ne, sw, d['latlng']), data)
        random.shuffle(result)
        result = result[:size]

        result = copy.deepcopy(result)
        for r in result:
            r['img'] = random.choice([img for img in r['images'] if 'map' not in img])
            r['tags'] = r.pop('types')
            if r.get('images', False):
                del r['images']
                del r['place_id']
                del r['scope']

        self.JsonResponse(result)


class Detail(JsonHandler):
    def get(self, *args, **kwargs):
        _id = self.request.get('id')
        logging.info(data[:10])
        result = filter(lambda d:d['id'] == _id, data)[0]
        result = copy.deepcopy(result)
        result['static_map'] = static_map.format(*result['latlng'])
        result['price_level'] = random.choice(range(1,6))
        if result.get('scope', False):
            del result['scope'] 
        if result.get('place_id', False):
            del result['place_id']
        result['logo'] = result['photos'] and result[0]
        self.JsonResponse(result)


class Login(JsonHandler):
    def post(self):
        email = self.request.get('email')
        password = self.request.get('password')

        user = User.get_by_id(email)
        if not user or password != user.password:
            self.JsonResponse({"STATUS": "ERROR", "MSG":"LOGIN ERROR"})
        else:
            self.session['user'] = user.to_json()
            self.JsonResponse({"STATUS": "SUCCESS"})


class Logout(JsonHandler):
    def get(self):
        self.session['user'] = None


class Signup(blobstore_handlers.BlobstoreUploadHandler, JsonHandler):
    def post(self):
        name = self.request.get('name')
        email = self.request.get('email')
        password = self.request.get('password')
        mug_shot = self.get_uploads('mug_shot')
        mug_shot = mug_shot and mug_shot[0].key()
        if User.get_by_id(email):
            self.JsonResponse({"STATUS": "ERROR", "MSG":"EMAIL REPEAT"})
            return            

        user = User(id=email)
        user.name = name
        user.email = email
        user.password = password
        if mug_shot:
            user.mug_shot = mug_shot
        user.put()
        self.JsonResponse({"STATUS": "SUCCESS"})


class SignupForm(JsonHandler):
    def get(self):
        template = '''
            <html>
                <body>
                    <form action="{}" method="POST" enctype="multipart/form-data">
                        Name: <input  name="name"><br> 
                        email: <input  name="email"><br> 
                        password: <input  name="password"><br> 
                        Upload File: <input type="file" name="mug_shot"><br> 
                        <input type="submit" name="submit" value="Submit"> 
                    </form>
                </body>
            </html>
        '''
        upload_url = blobstore.create_upload_url('/signup')
        self.response.out.write(template.format(upload_url))

class PostForm(JsonHandler):
    def get(self):
        template = '''
            <html>
                <body>
                    <form action="{}" method="POST" enctype="multipart/form-data">
                        commit: <input  name="commit"><br> 
                        Upload File: <input type="file" name="image"><br> 
                        <input type="submit" name="submit" value="Submit"> 
                    </form>
                </body>
            </html>
        '''
        upload_url = blobstore.create_upload_url('/post')
        self.response.out.write(template.format(upload_url))

    def post(self):
        return self.response.out.write(blobstore.create_upload_url('/post'))


class Post(JsonHandler):
    
    def get(self, _id=None):
        size = self.request.get_range('size') or 10
        cursor = self.request.get('cursor') or None

        if not _id:
            posts, next_curs, more = Post.query(share=True).fetch_page(size, start_cursor=cursor)
        elif _id == 'me':
            posts, next_curs, more = Post.query(author=User.get_by_id(self.session['user'].email)).fetch_page(size, start_cursor=cursor)
        else:
            posts = [Post.get_by_id(_id)]
            next_curs = None
            more = False
        
        posts = [post.to_json() for post in posts]
        
        self.JsonResponse({"posts":posts, "more":more, "next_cursor": next_cursor})
            
        
    def post(self):
        post = Post()
        post.author = User.get_by_id(self.session['user'].email)
        assert author
        post.commit = self.request.get('commit')
        post.image = self.get_uploads('image')[0].key()
        post.share = bool(self.get('share', False))
        post.put()
        
        self.JsonResponse({"STATUS": "SUCCESS"})

class Message(JsonHandler):
    def post(self):
        post_id = self.request.get('post_id')
        commit = self.request.get('commit')
        user = User.get_by_id(self.session['user'].email)
        assert user

        post = Post.get_by_id(post_id)
        msg = Message()
        msg.author = user
        msg.commit = commit
        
        msg.put()
        post.msgs.append(msg)
        post.put()


class Preview(webapp2.RequestHandler):
    def get(self):
        cursor = self.request.get('cursor', "")
        page = self.request.get_range('page')
        
        if cursor:
            cursor = Cursor(urlsafe=cursor)
            ss, next_curs, more = Store.query().fetch_page(100, start_cursor=cursor)
        else:
            ss, next_curs, more = Store.query().fetch_page(100)

        base_template = u'''
            <div>
                {}
            </div>
            <form>
                <input name='cursor' value='{}'>
                <input type='submit' value="下一頁">
            </form>
        '''

        store_template = u'''
            <div>
                <h2>{0}{1.name}</h2>
                <a href="/store/{0}?action=delete"  target="_blank">刪除store</a>
                <ul>{2}</ul>
            </div>
        
        '''

        img_template = u'''
            <li>
                <img src="{1}">
                <a href="/images/{0}?action=delete&{2}" target="_blank" >刪除圖片</a>
                <a href="/images/{0}?action=set&tag=Parity&{2}" target="_blank" >平價</a>
                <a href="/images/{0}?action=set&tag=Meals&{2}" target="_blank" >簡餐</a>
                <a href="/images/{0}?action=set&tag=Restaurant&{2}" target="_blank" >餐廳</a>
                <a href="/images/{0}?action=set&tag=Space&{2}" target="_blank" >空間</a>
                <a href="/images/{0}?action=set&tag=drink&{2}" target="_blank" >飲料</a>
                <div>{3}</div>
            </li>
        '''
        stores = []
        for s in ss:
            imgs = []
            for img in s.imgs:
                try:
                    code = urllib.urlencode({'url':img.image})
                    imgs.append(img_template.format(s.key.id(), img.image, code, img.tag))
                except Exception as e:
                    pass

            imgs = "\n".join(imgs)
            try:
                stores.append(store_template.format(s.key.id(), s, imgs))
            except Exception as e:
                print e
        html = base_template.format("\n".join(stores), next_curs.urlsafe())
        self.response.out.write(html)


class ImageHandler(webapp2.RequestHandler):
    def get(self, _id):
        _id = int(_id)
        action = self.request.get('action')
        

        url = self.request.get('url')
        s = Store.get_by_id(int(_id))
        imgs = s.imgs


        f = True
        tmp = []
        if action == 'delete':

            for img in imgs:
                if f and img.image == url:
                    f = False
                else:
                    tmp.append(img)

            s.imgs = tmp
            s.put()


        tag = self.request.get('tag')
        if action == 'set':
            for img in imgs:
                if img.image == url:
                    img.tag = tag
                tmp.append(img)

            s.imgs = tmp
            s.put()
        html = '''
        <script>
        window.close();
        </script>
        '''
        self.response.out.write(html)


class StoreHandler(webapp2.RequestHandler):
    def get(self, _id):
        store = Store.get_by_id(int(_id))       
        store.key.delete()

        html = '''
        <script>
        window.close();
        </script>
        '''
        self.response.out.write(html)