import psycopg2
from psycopg2.extras import DictCursor
import datetime, redis
from intex.load_intex_collateral import intex_data
from .globeop import download_data
from .insert_tranche_quotes import insert_quotes
from task_server import app
from flask import request, g
import json

def get_db():
    db = getattr(g, 'db', None)
    if db is None:
        db = g.db = psycopg2.connect(database="ET",
                                     user="et_user",
                                     host="debian",
                                     cursor_factory=DictCursor)
    return db

def get_queue():
    q = getattr(g, 'queue', None)
    if q is None:
        q = g.queue = redis.Redis(unix_socket_path='/var/run/redis/redis.sock')
    return q

@app.teardown_appcontext
def close_db(error):
    """Closes the database again at the end of the request."""
    db = getattr(g, 'db', None)
    if db:
        db.close()

@app.route('/insert_intex_data')
def intex():
    db = get_db()
    intex_data(db, request.args.get('workdate', str(datetime.date.today())))
    return 'OK'

@app.route('/globeop', methods=['PUT'])
def globeop():
    download_data(request.args.get('workdate', datetime.date.today(), lambda s: datetime.datetime.
                                   strptime(s, "%Y-%m-%d").date()))
    return 'OK'

@app.route('/insert_quotes', methods=['PUT'])
def insert_tranches():
    insert_quotes(**request.args)
    return 'OK'

@app.route("/", methods=['POST'])
def run_tasks():
    workdate = str(datetime.date.today())
    with get_queue().pipeline() as pipe:
        for dealname, reinvflag in request.form.items():
            app.logger.info("Processing {0} with{1} reinvestment".format(dealname,
                                                                         "" if reinvflag=="TRUE" else "out"))
            pipe.rpush("tasks", json.dumps(("build_portfolio",
                                            [workdate, dealname, reinvflag])))
        pipe.execute()
    return 'OK'