rsich-server/server.py at master · one8loop/rsich-server · 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
from flask_restful import Resource, Api, abort, reqparse
import json, sqlite3 as lite
from quiz import *

app = Flask(__name__)
api = Api(app)

userparser = reqparse.RequestParser()
userparser.add_argument('givePoints', type=int)
quizparser = reqparse.RequestParser()
quizparser.add_argument('channel', type=str, required=True, help='You must specify channel argument (la1/la2)')
quizparser.add_argument('getAnswersForId', type=int)
quizparser.add_argument('getCorrectForId', type=int)
quizparser.add_argument('getPoll', type=int)

class Quiz(Resource):
    def post(self):
        args = quizparser.parse_args()
        if args['getPoll'] == 1:
            pass
            # todo, prepare poll
        else: # Quiz
            if args['getAnswersForId'] != None:
                response = getAnswersForId(args['getAnswersForId'])
                if response == None:
                    abort(400, message='Invalid question id')
                return response
            elif args['getCorrectForId'] != None:
                response = getCorrectForId(args['getCorrectForId'])
                if response == None:
                    abort(400, message='Invalid question id')
                return response
            else:
                response = getQuiz(args['channel'])
                if response == None:
                    abort(400, message='No questions available at the moment')
                return response


        return response


class Leaderboard(Resource):
    def get(self):
        con = lite.connect('RSI.db')
        with con:
            cur = con.cursor()
            cur.execute('SELECT Username, Points, Gen_Points FROM User ORDER BY Points DESC LIMIT 10')
            rows = cur.fetchall()

        if rows == None:
            abort(400, message='No user found')

        leaderboard = []
        for pos, user in zip(range(len(rows)), rows):
            leaderboard.append({ '_id': pos+1,
                                 'id': str(pos+1),
                                 'username': user[0],
                                 'points': str(user[1]),
                                 'general_points': str(user[2]) })
        return leaderboard


class User(Resource):
    def get(self, userId):
        con = lite.connect('RSI.db')
        with con:
            cur = con.cursor()
            cur.execute('SELECT * FROM User WHERE rowid = ?', userId)
            row = cur.fetchone()

        if row == None:
            abort(400, message='User not found')

        user = { 'username': row[0],
                 'name': row[1],
                 'surname': row[2],
                 'points': row[3],
                 'general_points': row[4] }

        return user

    def put(self, userId):
        args = userparser.parse_args()
        if args['givePoints'] == 1:
            con = lite.connect('RSI.db')
            with con:
                cur = con.cursor()
                cur.execute('UPDATE User SET Points = Points + 50, Gen_Points = Gen_Points + 50 WHERE rowid = ?', userId)
        return 'Accepted', 202


api.add_resource(Quiz, '/quiz')
api.add_resource(Leaderboard, '/leaderboard')
api.add_resource(User, '/user/<string:userId>')

if __name__ == '__main__':
#    app.run(debug=False)
    app.run(host='0.0.0.0', port=5689, debug=False) # To open to the network