ISSTrackerBE/application.py at master · ISSTrackr/ISSTrackerBE · 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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
from flask import Flask, Response
from flask import request as frequest
from Backend.Requests.issCurrPos import currPos as issCurrentPosition
from Backend.Requests.rssFeed import rssFeed as rssFeed
from Backend.Requests.ISSCountryPasses import ISScountryPasses
from Backend.Requests.userPosition import getUserPosition as userPosition
from Backend.Core.database import redisDB
from Backend.Requests.issPastPasses import pastPasses
from Backend.Requests.issFuturePasses import getFuturePass
from Backend.Core.XMLParser import reformatData, parseRequestParamsXMLToDic, XMLvalidate
from Backend.Requests.addressGeocoding import geocoder
import multiprocessing
from Backend.Core.checkThread import checkThread
from flask_cors import CORS, cross_origin

app = Flask(__name__, static_url_path='')

cors = CORS(app)
BadRequest = '<?xml version="1.0" encoding="UTF-8"?>' \
             '<message>' \
             '<error>Error 400: Bad Request</error>' \
             '<description></description>' \
             '</message>'

multiprocessing.Process(target=checkThread).start()


def makeResponse(data, status):
    return Response(data, status=status, content_type='text/xml')  # headers={}


def _checkData(requestName, data):
    correct = False
    allowedKeys = {
        "ISSDB": [
            "startTime",
            "endTime",
        ],
        "ISSpos": True,
        "AstrosOnISS": True,
        "CountryList": True,
        "GeoJson": [
            "country"
        ],
        "ISSCountryPasses": [
            "startTime",
            "endTime",
            "country"
        ],
        "RSS-Feed": [
            "startID",
            "endID"
        ],
        "ISSpastPasses": [
            "latitude",
            "longitude",
            "radius"
        ],
        "ISSfuturePasses": [
            "latitude",
            "longitude",
            "number"
        ],
        "GeocodingAddress": [
            "q"
        ]
    }

    if allowedKeys[requestName] is not None:
        if data is not None:
            if "params" in data and data["params"] is not None:
                for key in allowedKeys[requestName]:
                    correct = True if key in data["params"] and data["params"][key] is not None else False
                    if not correct:
                        break
            # print(correct)
            return correct
        elif allowedKeys[requestName] is True:
            return True
        else:
            return False
    else:
        return False


@app.route('/<requestName>', methods=['GET', 'POST'])
@cross_origin()  # Adds 'Access-Control-Allow-Origin': '*' to answer
def request(requestName):
    if frequest.method == 'GET':
        if requestName == 'AstrosOnISS':
            return makeResponse(
                data=reformatData(requestData=redisDB().getData(None, requestName), requestName=requestName),
                status=200)
        elif requestName == 'ISSpos':
            return makeResponse(data=reformatData(requestData=issCurrentPosition(), requestName=requestName),
                                status=200)
        elif requestName == "CountryList":
            return makeResponse(
                data=reformatData(requestData=redisDB().getData(None, requestName), requestName=requestName),
                status=200)
        else:
            return makeResponse(BadRequest, 400)

    elif frequest.method == 'POST':
        resp = ""
        try:
            body = parseRequestParamsXMLToDic(frequest.data)
        except TypeError:
            body = None

        if body == "INVALID XML" or not _checkData(requestName=requestName, data=body):
            return makeResponse(BadRequest, 400)
        else:
            if requestName == "ISSCountryPasses":
                resp = reformatData(requestData=ISScountryPasses(requestData=body), requestName=requestName)
            elif requestName == "ISSDB":
                resp = reformatData(requestData=redisDB().getData(body, requestName), requestName=requestName)
            elif requestName == "GeoJson":
                resp = reformatData(requestData=redisDB().getData(body, requestName), requestName=requestName)
            elif requestName == "RSS-Feed":
                resp = reformatData(requestData=redisDB().getData(requestData=body, requestName=requestName),
                                    requestName=requestName)
            elif requestName == "ISSpastPasses":
                resp = reformatData(requestData=pastPasses().pastPasses(requestData=body), requestName=requestName)
            elif requestName == "ISSfuturePasses":
                resp = reformatData(requestData=getFuturePass(params=body["params"]), requestName=requestName)
            elif requestName == "GeocodingAddress":
                resp = reformatData(requestData=geocoder(params=body["params"]), requestName=requestName)
            else:
                return makeResponse(BadRequest, 400)
            if XMLvalidate(resp) and not resp == "":
                return makeResponse(data=resp, status=200)
            else:
                return makeResponse(data='<?xml version="1.0" encoding="UTF-8"?>' \
                                  '<message>' \
                                  '<error>Error 500: Bad XML Response</error>' \
                                  '<description>An Error occured while validating the XML file</description>' \
                                  '</message>', status=500)