Exemplo n.º 1
0
 def test_expr(self):
     key = "expr"
     self.assertEquals(None, cache.get(key))
     cache.set(key, u"ABC", 1)
     self.assertEquals(u"ABC", cache.get(key))
     time.sleep(1.2)
     self.assertEquals(None, cache.get(key))
Exemplo n.º 2
0
 def test_incr(self):
     key = "inc"
     self.assertEquals(None, cache.get(key))
     cache.set(key, 123)
     self.assertEquals(123, cache.get(key))
     cache.incr(key)
     self.assertEquals(124, cache.get(key))
     cache.incr(key, 10)
     self.assertEquals(134, cache.get(key))
Exemplo n.º 3
0
 def save_request_token(self, token, secret):
     # cache them, but first calculate expires time:
     now = datetime.now()
     t = datetime(now.year, now.month, now.day)
     delta = now - t
     exp = 86400 - delta.seconds # makes it expires in midnight (24:00)
     if exp > 3600:
         exp = 3600
     logging.info('fetched and cache for %d seconds.' % exp)
     cache.set(token, secret, exp)
Exemplo n.º 4
0
 def test_cache(self):
     key1 = "key1"
     key2 = "key2"
     self.assertEquals(None, cache.get(key1))
     self.assertEquals(None, cache.get(key2))
     cache.set(key1, "abc")
     cache.set(key2, ["x", "y", "z"])
     self.assertEquals("abc", cache.get(key1))
     self.assertEquals(["x", "y", "z"], cache.get(key2))
     cache.delete(key1)
     cache.delete(key2)
     self.assertEquals(None, cache.get(key1))
     self.assertEquals(None, cache.get(key2))
Exemplo n.º 5
0
def get_count(name):
    '''
    Retrieve the value for a given sharded counter.
    
    Args:
        name: the name of the counter
    Returns:
        Integer value
    '''
    total = cache.get(CACHE_KEY_PREFIX + name)
    if total is None:
        total = 0
        for counter in ShardedCounter.all().filter('name =', name).fetch(1000):
            total += counter.count
        cache.set(CACHE_KEY_PREFIX + name, str(total))
        return total
    return int(total)
Exemplo n.º 6
0
 def is_too_many_retweets(rt):
     try:
         tid = str(rt.retweeted_status.id)
     except KeyError:
         logging.debug('-'*80 + 'KeyError' + rt.text)
         tid = str(rt.id)
     #print '----%s----%s----%s' % (rt, rt.retweeted_status.id, (tid))
     times = cache.get(tid)
     if times:
         logging.info("%s: %s times %s" % (tid, times, rt.retweeted_status.text))
         if times > 20:
             return True
         else:
             cache.incr(tid)
     else:
         cache.set(tid, '1')
     return False
Exemplo n.º 7
0
def _get_weather(city):
    if not city:
        return r'{"code":"500","Message":"Missing Input: city"}'
    data = cache.get("city-%s" % city)
    if data:
        logging.info("got data from cache")
        web.header("X-Cache", "Hit from cache")
        return data
    # check city:
    cities = list(db.select("city", where="yahoo_code=$code", vars=dict(code=city)))
    if not cities:
        return r'{"code":"500","Message":"Invalid Input: city"}'
    c = cities[0]
    w = c.yahoo_code
    logging.info("fetch from yahoo weather...")
    url = "http://weather.yahooapis.com/forecastjson?w=%s&u=c" % w
    resp = urllib2.urlopen(url)
    if resp.code == 200:
        logging.info("begin fetch...")
        data = json.loads(resp.read())
        data["city"] = c.name
        cond = data["forecast"][0]["condition"]
        codes = _get_weather_code(cond)
        data["forecast"][0]["code"] = codes[0]
        data["forecast"][0]["condition"] = codes[1]
        data["forecast"][0]["image"] = codes[2]
        cond = data["forecast"][1]["condition"]
        codes = _get_weather_code(cond)
        data["forecast"][1]["code"] = codes[0]
        data["forecast"][1]["condition"] = codes[1]
        data["forecast"][1]["image"] = codes[2]
        # cache it, but first calculate expires time:
        now = datetime.now()
        t = datetime(now.year, now.month, now.day)
        delta = now - t
        exp = 86400 - delta.seconds  # makes it expires in midnight (24:00)
        if exp > 3600:
            exp = 3600
        logging.info("fetched and cache for %d seconds." % exp)
        json_data = json.dumps(data)
        cache.set("city-%s" % city, json_data, exp)
        web.header("X-Cache", "Miss from cache")
        return json_data
    return None