queryapp/flask_query_api.py at master · c4n2012/queryapp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from flask import Flask, request, jsonify
import yaml
import requests
import sys
#from .query_class import Query

if len(sys.argv) != 1 :
    app_run_address = sys.argv[1]
    app_run_port = sys.argv[2]
    config_api_url = sys.argv[3]
else:
    app_run_address = "0.0.0.0"
    app_run_port = "5003"
    config_api_url = "http://cfgmanapp.dev.svc:5004/conf/query"

class Query(object):
    def __init__(self,symbols,query_type,day_range = 0,time_interval = 0):
        self.config_api_url = config_api_url
        self.symbols = symbols
        self.query_type = query_type
        self.day_range = day_range
        self.time_interval = time_interval

    def make_query(self, remote_url):
        res = requests.get(remote_url)
        data = res.json()
        yml_data = yaml.dump(data)
        return yml_data

    # {remote_api_url: ..., query_template: ..., remote_api_token: ...}

    def url_constructor(self, config_data):
        if self.query_type == 'realtime':
            api_url = config_data["remote_api_url"] + config_data["query_template"] + 'symbol=' + ','.join(self.symbols) + "&api_token=" + config_data["remote_api_token"]
            return api_url
        elif self.query_type == 'intraday':
            api_url = config_data["remote_api_url"] + config_data["query_template"] + 'symbol=' + "".join(self.symbols) +  "&interval=" + str(self.time_interval) + "&range=" + str(self.day_range) +"&api_token=" + config_data["remote_api_token"]
            return api_url
        else:
            print("UNKNOWN QUERY TYPE")
            return "UNKNOWN QUERY TYPE"

    def get_config(self):
        payload = "query_type=" + self.query_type
        try:
            res = requests.get(self.config_api_url, params=payload)
        except:
            print("Config manager service is not responding")
            return "Config manager service is not responding"
        # print(res.json())
        return res.json()


app = Flask(__name__)
@app.route("/")
def hello():
    return "There is no service here. Use '/query/yml_data'."

@app.route('/query/yml_data', methods=['POST'])
def receive_yml_data():
    if request.method == 'POST':

        yml_req_src = request.get_data(cache=True, as_text=True)
        yml_data = yaml.load(yml_req_src,Loader=yaml.FullLoader)
        query_obj = Query(yml_data['symbols'],yml_data['query_type'])

        if query_obj.query_type == 'intraday':
            query_obj.day_range = yml_data['day_range']
            query_obj.time_interval = yml_data['time_interval']

        remote_url = query_obj.url_constructor(query_obj.get_config())
        print (remote_url)
        yml_response = query_obj.make_query(remote_url)

        return yml_response

# @app.route('/conf/query', methods=['GET'])
# def config_data():
#     remote_realtime_api_url = "https://api.worldtradingdata.com"
#     remote_intraday_api_url = "https://intraday.worldtradingdata.com"
#     query_template_realtime = "/api/v1/stock?"
#     query_template_intraday = "/api/v1/intraday?"
#     remote_api_token = "UL0eS95gqeGZMgOtgDW9PZXSuQB9xQex3luk6RhDOErkm74JKch6ps7vXDdQ"
#
#     if request.method == 'GET':
#         query_type = request.args.get('query_type')
#         if query_type == "realtime":
#             response = {"remote_api_url" : remote_realtime_api_url, "query_template" : query_template_realtime, "remote_api_token": remote_api_token }
#             # print (response)
#             return jsonify(response)
#         elif query_type == "intraday" :
#             response = {"remote_api_url" : remote_intraday_api_url, "query_template" : query_template_intraday, "remote_api_token": remote_api_token }
#             # print ("query_type = intraday" )
#             return jsonify(response)
#     return "UNKNOWN QUERY TYPE"


if __name__ == '__main__':
    app.run(host=app_run_address, port=app_run_port)