stock/server.py at master · zhangsheng377/stock · 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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
import datetime
import json
import time
import requests

from flask import Flask, request
import xmltodict

from UTILS.db_sheets import get_users, insert_users, update_one_user
from UTILS.config_port import user_send_host, user_send_port
from UTILS.config import VERSION, LOGGING_LEVEL

application = Flask(__name__)
# application.debug = True
application.logger.setLevel(LOGGING_LEVEL)


def get_filter_users(filter: dict):
    users = get_users()
    result = []
    for user in users:
        is_find = True
        for key, value in filter.items():
            if key not in user or user[key] != value:
                is_find = False
                break
        if is_find:
            result.append(user)
    return result


@application.route('/')
def hello_world():
    return 'Hello, World!' + '<br /><br />' + str(datetime.datetime.now())


@application.route('/wx', methods=["GET", "POST"])
def get():
    if request.method == "GET":  # 判断请求方式是GET请求
        my_echostr = request.args.get('echostr')  # 获取携带的echostr参数
        return my_echostr
    else:
        # 表示微信服务器转发消息过来
        xml_str = request.data
        if not xml_str:
            return ""
        resp_dict = None
        re_content = "信息错误"
        # 对xml字符串进行解析
        xml_dict = xmltodict.parse(xml_str)
        xml_dict = xml_dict.get("xml")

        application.logger.info(f"xml_dict:{xml_dict}")

        # 提取消息类型
        msg_type = xml_dict.get("MsgType")
        if msg_type == "text":  # 表示发送的是文本消息
            content = xml_dict.get("Content")
            if content == "查询已订阅股票":
                users = get_filter_users(filter={'wechat': xml_dict.get("FromUserName")})
                if users:
                    re_content = str(users[0]['stocks'])
                else:
                    re_content = "尚未绑定微信"
            # elif re.fullmatch(r'\d{6}\.\w{2}', content):
            #     re_content = "code: " + content
            elif content.startswith("查询 "):
                try:
                    datas = content.split(" ")
                    stock_id = datas[1]
                    users = get_filter_users(filter={'wechat': xml_dict.get("FromUserName")})
                    if users:
                        user_name = users[0]['_id']
                        url = f'http://{user_send_host}:{user_send_port}/send_user'
                        data = {'user_id': json.dumps(user_name), 'stock_id': json.dumps(stock_id),
                                'old_result_len': json.dumps(0)}  # 将携带的参数传给params
                        re_len = requests.get(url, params=data).json()
                        re_content = "发送成功: {} {} {}".format(user_name, stock_id, re_len)
                    else:
                        re_content = "尚未绑定微信"
                except Exception as e:
                    re_content = "发送失败: " + str(e)
                    application.logger.error(re_content, exc_info=True)
            elif content.startswith("绑定 "):
                datas = content.split(" ")
                user_name = datas[1]
                # 此处逻辑需要细考
                if get_filter_users(filter={'_id': user_name}):
                    re_content = "您要绑定的用户名:{}，已被人绑定!请联系微信435878393".format(user_name)
                elif get_filter_users(filter={'wechat': xml_dict.get("FromUserName")}):
                    re_content = "您的微信已被绑定!请联系微信435878393"
                else:
                    if insert_users(document={'_id': user_name, 'wechat': xml_dict.get("FromUserName")}):
                        re_content = "绑定成功"
                    else:
                        re_content = "绑定失败"
            elif content.startswith("订阅 "):
                datas = content.split(" ")
                stock_id = datas[1]
                users = get_filter_users(filter={'wechat': xml_dict.get("FromUserName")})
                if users:
                    user = users[0]
                    if 'stocks' not in user.keys():
                        user['stocks'] = []
                    stocks = set(user['stocks'])
                    stocks.add(stock_id)
                    _filter = {'wechat': xml_dict.get("FromUserName")}
                    _update = {'$set': {'stocks': list(stocks)}}
                    result = update_one_user(filter=_filter, update=_update)
                    re_content = f"订阅失败. {_filter} {_update} {result.raw_result}"
                    if result is not None and result.modified_count > 0:
                        re_content = "订阅成功"
                else:
                    re_content = "尚未绑定微信"
            elif content.startswith("取消订阅 "):
                datas = content.split(" ")
                stock_id = datas[1]
                users = get_filter_users(filter={'wechat': xml_dict.get("FromUserName")})
                if users:
                    user = users[0]
                    if 'stocks' not in user.keys():
                        user['stocks'] = []
                    stocks = set(user['stocks'])
                    if stock_id in stocks:
                        stocks.remove(stock_id)
                        _filter = {'wechat': xml_dict.get("FromUserName")}
                        _update = {'$set': {'stocks': list(stocks)}}
                        result = update_one_user(filter=_filter, update=_update)
                        re_content = f"尚未订阅{stock_id}. {_filter} {_update} {result.raw_result}"
                        if result is not None and result.modified_count > 0:
                            re_content = "取消订阅成功"
                    else:
                        re_content = f"尚未订阅{stock_id}"
                else:
                    re_content = "尚未绑定微信"
            else:
                re_content = content

        if not resp_dict:
            # 构造返回值，经由微信服务器回复给用户的消息内容
            resp_dict = {
                "xml": {
                    "ToUserName": xml_dict.get("FromUserName"),
                    "FromUserName": xml_dict.get("ToUserName"),
                    "CreateTime": int(time.time()),
                    "MsgType": "text",
                    "Content": re_content,
                }
            }

        # 将字典转换为xml字符串
        resp_xml_str = xmltodict.unparse(resp_dict)

        application.logger.info(f"resp_xml_str:{resp_xml_str}")

        # 返回消息数据给微信服务器
        return resp_xml_str


if __name__ == "__main__":
    application.logger.info(f"VERSION: {VERSION}")
    # application.run(host="0.0.0.0", port=5000)
    application.run(host="0.0.0.0")