diff options
Diffstat (limited to 'python/http-server.py')
| -rw-r--r-- | python/http-server.py | 79 |
1 files changed, 6 insertions, 73 deletions
diff --git a/python/http-server.py b/python/http-server.py index 38df18f2..c2058cd0 100644 --- a/python/http-server.py +++ b/python/http-server.py @@ -1,75 +1,8 @@ -from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler -from urlparse import urlparse, parse_qsl -from daemon import Daemon -import sys, datetime, logging, json, redis -from insert_tranche_quotes import insert_quotes -from load_intex_collateral import intex_data -import psycopg2 -from psycopg2.extras import DictCursor -from globeop import download_data +import logging +from task_server import app -class MyHandler(BaseHTTPRequestHandler): - def do_GET(self): - if self.path.startswith("/insert_quotes"): - q = urlparse(self.path).query - if q: - args = dict(parse_qsl(q)) - insert_quotes(**args) - else: - insert_quotes() - if self.path == "/insert_intex_data": - workdate = str(datetime.date.today()) - intex_data(self.server.conn, workdate) +file_handler = logging.FileHandler(filename='/home/share/CorpCDOs/logs/tasks.log') +file_handler.setLevel(logging.INFO) +app.logger.addHandler(file_handler) - if self.path.startswith("/globeop"): - q = urlparse(self.path).query - if q: - args = dict(parse_qsl(q)) - args['workdate'] = datetime.datetime.strptime(args['workdate'], "%Y-%m-%d").date() - download_data(**args) - else: - download_data() - self.send_response(200) - self.end_headers() - - def do_POST(self): - length = int(self.headers['content-length']) - d = parse_qsl(self.rfile.read(length).decode('utf-8')) - self.log_message("%s" , json.dumps(d)) - workdate = str(datetime.date.today()) - pipe = self.server.queue.pipeline() - for dealname, reinvflag in d: - pipe.rpush("tasks", json.dumps(("build_portfolio", - [workdate, dealname, reinvflag]))) - pipe.execute() - self.send_response(200) - self.end_headers() - - def log_message(self, format, *args): - logging.info("%s - - [%s] %s" % - (self.address_string(), - self.log_date_time_string(), - format%args)) - -class MyServer(HTTPServer): - def __init__(self, addr, handler, queue, conn): - HTTPServer.__init__(self, addr, handler) - self.queue = queue - self.conn = conn - -class MyDaemon(Daemon): - def run(self): - server_address = ('',8000) - logging.basicConfig(filename='/home/share/CorpCDOs/logs/tasks.log', level=logging.INFO) - q = redis.Redis(unix_socket_path='/var/run/redis/redis.sock') - self.conn = psycopg2.connect(database="ET", - user="et_user", - host="debian", - cursor_factory=DictCursor) - http = MyServer(server_address, MyHandler, q, self.conn) - http.serve_forever() - -if __name__=="__main__": - d = MyDaemon('/tmp/tasks.pid') - command = getattr(d, sys.argv[1]) - command() +app.run(host='0.0.0.0', port = 8000, debug=True) |
