#!/usr/bin/env python

# -*- coding: utf-8 -*-

from datetime import datetime, timedelta

from flask import Response, request #@UnresolvedImport

from flaskext.sqlalchemy import get_debug_queries

from iri_tweet.models import TweetSource, Tweet

from pyrfc3339.generator import generate

from sqlalchemy import func, desc #@UnresolvedImport

from sqlalchemy.orm import joinedload

from tweetgserver import app, db, cache

import pytz

import simplejson as json #@UnresolvedImport

import re

import math

def jsonpwrap(resobj):

    resp_str = json.dumps(resobj)

    callback_param = request.args.get("callback")

    if callback_param:

        resp_str = "%s( %s );" % (callback_param, resp_str)

        mime_type = 'text/javascript'

    else:

        mime_type = 'application/json'

    return Response(resp_str, mimetype=mime_type)

@app.route('/', methods=['GET'])

def index():

    query = db.session.query(func.count(TweetSource.id)) #@UndefinedVariable

    ts_list_count = query.scalar()

    return 'Nb of tweets ' + str(ts_list_count)

@app.route('/total', methods=['GET'])

def total():

    query = db.session.query(func.count(TweetSource.id)) #@UndefinedVariable

    ts_list_count = query.scalar()

    return jsonpwrap({"total":ts_list_count})

@app.route('/podium/<tokens>', methods=['GET'])

def podium(tokens):

    token_list = tokens.split(",")

    query_base = db.session.query(func.count(Tweet.id)) #@UndefinedVariable

    podium_res = {}

    for token in token_list:

        query = query_base.filter(Tweet.text.op('~*')(token)) #@UndefinedVariable

        podium_res[token] = query.scalar()

    res = {

        "podium" : podium_res,

        "total" : query_base.scalar()

    }

    return jsonpwrap(res)