core.py
1.7 KB
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2020-02-07 13:59
# @Author : Lemon
# @File : core.py
# @Software: PyCharm
import json
from tornado import ioloop
from conf import settings
from core.monitor import Monitor
from lib import control_server
from lib.socket_ import MessageSocketClient
monitor = Monitor()
def message_receive(message):
print("收到消息", message)
if message == 'login1':
monitor.login("[email protected]", "nantian888")
elif message == 'exit1':
monitor.logout('100047151842270')
else:
try:
req = json.loads(message)
type_ = req.get('type')
data_ = req.get('data')
if type_ == 'task':
commands = json.loads(data_)
for cmd in commands:
monitor.execute(cmd)
elif type_ == 'initialize':
commands = json.loads(data_)
monitor.init_config = commands
except:
import traceback
print(traceback.format_exc())
def server_replace(ws_url):
monitor._temp_socket = MessageSocketClient(connect_success, message_receive)
monitor._temp_socket.connect(ws_url, reconnect=False)
def connect_success(sock):
if monitor._temp_socket == sock:
sock.auto_reconnet = True
monitor._socket.quit() # 旧的断开
monitor._temp_socket = None
monitor.bind(sock)
monitor._init_server_()
monitor._socket.new_server_func = lambda: control_server.get_ws_address(monitor._name, sock.ws_url)
monitor._socket.server_replace = server_replace
def run():
socket = MessageSocketClient(connect_success, message_receive)
socket.connect(**settings.get_server())
ioloop.IOLoop.instance().start()