Computer-Networks-Project/userDataServer.py at main · rafaelandrealves/Computer-Networks-Project · 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
##################### Library Imports #####################

import userData
from flask import Flask, request, jsonify, abort

# -- ADINT Final Project
# -- Made by: Diogo Ferreira and Rafael Cordeiro

# ----------------------------------------
# --------------User Data Server----------
# ----------------------------------------


#Flask
app = Flask(__name__)


# Database endpoint for adding new user
@app.route("/users/newuser",methods = ['POST'])
def newUserRequest():

    #retrieve data from input JSON body
    data = request.json

    try:
        int(data["user_id"])
        data["token"]
        data["secret_code"]
    except:
        abort(400)

    try:
        status = userData.newUser(int(data["user_id"]),data["token"],data["secret_code"])
    except:
        return jsonify({'StatusCode':'2', 'Description':'Err'})
    if status:

        return jsonify({'StatusCode':'1','Description':'OK'})
    else:
        # The ID already exists - Not Admitted
        userData.UpdateuserToken(int(data["user_id"]),data["token"])
        return jsonify({'StatusCode':'3','Description':'Updated Token'})


@app.route("/user/check",methods = ['GET'])
def CheckUser():
    #retrieve data from input JSON body
    data = request.json

    try:
        int(data["istID"])
        data["token"]
    except:
        abort(400)

    if userData.CheckuserID(int(data["istID"])) and userData.CheckuserToken(data["token"]):
        return jsonify({'StatusCode':'1', 'Description':'OK'})
    else:
        return jsonify({'StatusCode':'2','Description':'Err'})


@app.route("/user/<path:istID>/updateCode",methods = ['POST'])
def UpdateUserCode(istID):
    #retrieve data from input JSON body
    data = request.json

    try:
        int(data["istID"])
        data["token"]
        data["secret"]
    except:
        abort(400)

    Status = userData.UpdateuserSecret(int(data["istID"]),data["secret"])
    if Status:
        return jsonify({'StatusCode':'1', 'Description':'OK'})
    else:
        return jsonify({'StatusCode':'2','Description':'Err'})


@app.route("/user/bycode")
def CheckbyCode():
    data = request.json

    try:
        data['code']
    except:
        abort(400)

    user = userData.GetUserbyAC(data['code'])

    if user:
        if userData.CheckCreaTime(user.user_id) and userData.UpdateuserSecret(user.user_id,"",1):
            return jsonify({'userID': user.user_id})
        else:
            return jsonify({})
    else:
        return jsonify({})

#Start server
if __name__ == "__main__":
    app.run(host='0.0.0.0', port=8001, debug=True)