1
from flask import Flask, jsonify, request
2
from flask_cors import CORS
3
import mysql.connector
4
from mysql.connector import Error
5
# 引入 string
6
import string
7

8
app = Flask(__name__)
9
# enable CORS
10
CORS(app)
11

12
# 註冊 email 驗證後的緩存
13
registerTempData = {
14
    "id": "",
15
    "username": "",
16
    "email": "",
17
    "phone": "",
18
    "password": "",
19
}
20

21
# 全局驗證碼
22
captcha = 0
23

24
# 驗證 註冊的 Email
25
@app.route("/user/validateEmail", methods=['GET', 'POST'])
26
def user_register_email_validate():
27
    if request.method == 'POST':
28
        response_object = {'status': 'success'}
29
        response_object['code'] = 200
30
        requestCaptcha = request.form.get('captcha')
31
        print("requestCaptcha: " + str(requestCaptcha))
32
        print("captcha: " + str(captcha))
33
        # 如果驗證碼相同
34
        if str(requestCaptcha) == str(captcha):
35
            global registerTempData
36
            # 新增資料
37
            sql = "INSERT INTO user (id, username, email, phone, password) VALUES (%s, %s, %s, %s, %s);"
38
            # 將緩存的註冊資料加入
39
            new_data = (registerTempData["id"], registerTempData["username"],
40
                        registerTempData["email"], registerTempData["phone"], registerTempData["password"])
41
            print(new_data)
42
            cursor = connection.cursor()
43
            cursor.execute(sql, new_data)
44

45
            # 確認資料有存入資料庫
46
            connection.commit()
47
            response_object['message'] = 'Your account is activated!'
48
            response_object['data'] = True
49

50
            # 初始化
51
            registerTempData = {
52
                "id": "",
53
                "username": "",
54
                "email": "",
55
                "phone": "",
56
                "password": "",
57
            }
58
        else:
59
            response_object['message'] = 'The captcha is not true!'
60
            response_object['data'] = False
61

62
        return jsonify(response_object)

1
# 返回用戶資訊
2
@app.route("/user/getDetail", methods=['GET', 'POST'])
3
@jwt_optional
4
def user_getDetail():
5
    if request.method == 'POST':
6
        response_object = {'status': 'success'}
7
        response_object['code'] = 200
8
        requestId = request.form.get('id')
9
        sqlSearchResult = []
10

11
        # 驗證 jwt_token
12
        current_user = get_jwt_identity()
13

14
        # 驗證失敗 回傳失敗
15
        if not current_user:
16
            response_object['message'] = 'anonymous user'
17
            response_object['data'] = False
18

19
            return jsonify(response_object)
20

21
        # 用 id 查詢用戶資訊
22
        cursor.execute(
23
            "SELECT id, username, email, phone, password FROM user WHERE id = %s;", (requestId,))
24
        for (id, username, email, phone, password) in cursor:
25
            sqlSearchResult.append({
26
                "id": id,
27
                "username": username,
28
                "email": email,
29
                "phone": phone,
30
                "password": password
31
            })
32

33
        print(sqlSearchResult)
34

35
        # 如果有資料
36
        if cursor.rowcount > 0:
37
            response_object['message'] = 'Searching is Success. Have Data'
38
            response_object['data'] = True
39
            # 將搜到的數據添入 response_object
40
            response_object.update({
41
                'userData': {
42
                    'id': sqlSearchResult[0]['id'],
43
                    'username': sqlSearchResult[0]['username'],
44
                    'email': sqlSearchResult[0]['email'],
45
                    'phone': sqlSearchResult[0]['phone'],
46
                    'password': sqlSearchResult[0]['password']
47
                }
48
            })
49

50
        # 如果沒有資料
51
        else:
52
            response_object['message'] = 'There is not this account in the database.'
53
            response_object['data'] = False
54

55
        return jsonify(response_object)

1
# 模糊搜索 所有 channelType 是 group 的群
2
@app.route("/user/channels/searchChannels", methods=['GET', 'POST'])
3
@jwt_optional
4
def user_channels_search_channels():
5
    if request.method == 'POST':
6
        response_object = {'status': 'success'}
7
        response_object['code'] = 200
8
        requestKeyword = request.form.get('keyword')
9

10
        sqlSearchResult = []
11

12
        # 驗證 jwt_token
13
        current_user = get_jwt_identity()
14

15
        # 驗證失敗 回傳失敗
16
        if not current_user:
17
            response_object['message'] = 'anonymous user'
18
            response_object['data'] = False
19

20
            return jsonify(response_object)
21

22
        # 用 keyword 模糊查詢 所有 channel (channelType = group)
23
        cursor.execute(
24
            "SELECT id, channelType, adminId, channelName FROM channel WHERE channelType = 'group' \
25
             AND (id LIKE CONCAT('%', %s, '%') OR adminId LIKE CONCAT('%', %s, '%') OR \
26
              channelName LIKE CONCAT('%', %s, '%')) GROUP BY id;",
27
               (requestKeyword, requestKeyword, requestKeyword))
28
        for (id, channelType, adminId, channelName) in cursor:
29
            sqlSearchResult.append({
30
                "id": id,
31
                "channelType": channelType,
32
                "adminId": adminId,
33
                "channelName": channelName,
34
            })
35
        # 如果有 資料
36
        if cursor.rowcount > 0:
37
            response_object['message'] = 'Get the channel List success.'
38
            response_object['data'] = True
39
            # 將資料添加到 response_object
40
            response_object.update({
41
                'searchResult': sqlSearchResult
42
            })
43
        # 沒有 message
44
        else:
45
            response_object['message'] = 'There is no data.'
46
            response_object['data'] = False
47

48
        return jsonify(response_object)