insert.py

# -*- coding: utf-8 -*-
from models import Food, FoodDailyInfo
from google.appengine.ext import ndb, db

def process_item(item):
#    print item

    food = Food.get_or_insert(item['name'])
    for key, value in item.items():
        try:
            if key != 'date':
                setattr(food, key, value)
                print key, value
        except Exception as e:
            print e

    food.push_info(item['date'])
    food.aggregate()
    food.get_recommand()
    return food


def download(date):
    print 'start'
    from works import pork, checken, fish, vegetable
    
    results = []
    for item in vegetable.get(date):
        results.append(process_item(item))

    print len(results)
    for item in fish.get(date):
        results.append(process_item(item))

    print len(results)
    for item in checken.get(date):
        if u'其他' in item['name']:
            continue
        if u'待訂' in item['name']:
            continue
        results.append(process_item(item))

    print len(results)
    for item in pork.get(date):
        results.append(process_item(item))
    
    ndb.put_multi(results)
    print len(results)


from datetime import datetime, timedelta, date

def run():
    time = date(2014,6,1)
    today = datetime.now().date()
    while time <= today:
        print time
        download(time)
        time += timedelta(1)

def fix_date(food):
    pass

def sort():

    vegetables = Food.query(Food.type=="vegetable").fetch(1000)
    vegetables.sort(key=lambda x:x.get_point())
    vegetables.reverse()
    meats = Food.query(Food.type=="meat").fetch(1000)
    meats.sort(key=lambda x:x.get_point())
    meats.reverse()
    fishs = Food.query(Food.type=="fish").fetch(1000)
    fishs.sort(key=lambda x:x.get_point())
    fishs.reverse()

    c = 0
    for vegetable in vegetables:
        print 'vegetable', c, vegetable.name
        vegetable.rank = c
        print vegetable.point
        if c:
            vegetable.prev = vegetables[c-1].name
        if c < len(vegetables) - 1:
            vegetable.next_one = vegetables[c+1].name
        c += 1

    c=0
    for meat in meats:
        print 'meat', c, meat.name
        print meat.point
        meat.rank = c
        if c:
            meat.prev = meats[c-1].name
        if c < len(meats) - 1:
            meat.next_one = meats[c+1].name
        c += 1


    c=0
    for fish in fishs:
        print 'fish', c, fish.name
        print fish.point
        fish.rank = c
        if c:
            fish.prev = fishs[c-1].name
        if c < len(fishs) - 1:
            fish.next_one = fishs[c+1].name
        c += 1

    result = []
    result.extend(vegetables)
    result.extend(fishs)
    result.extend(meats)
    ndb.put_multi(result)


def set_image():
    import google_search_image
    foods = Food.query().fetch(1000)
    for food in foods:
        try:
            print food.name, food.image
            assert not food.image, 'had image'
            key = u'{}'.format(food.name)
            print key
            img = google_search_image.search(key.encode('utf-8')).next()
            food.image = img.get('url')
            print food.name, food.image
            food.put()
        except Exception as e:
            print e