core.py
3.1 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
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2020-02-07 13:59
# @Author : Lemon
# @File : core.py
# @Software: PyCharm
import datetime
import functools
import json
from apscheduler.schedulers.tornado import TornadoScheduler
from tornado import ioloop
from conf import settings
from core.monitor import Monitor
from lib import control_server
from lib.socket_ import MessageSocketClient
try:
from lib.message_queue import MessageQueue as Queue
except:
Queue = None
monitor = Monitor()
monitor.version = settings.get_version()
def server_replace(ws_url):
'''新开一个socket尝试连接新地址，存储在临时变量里'''
monitor._temp_socket = MessageSocketClient(connect_success, message_receive)
set_new_server_func(monitor._temp_socket)
monitor._temp_socket.connect(ws_url, monitor._name, monitor._imei, reconnect=False) # 失败不重试
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':
monitor.init_config = json.loads(data_)
except:
import traceback
print(traceback.format_exc())
def host():
return 'ws://fbchat.data.gzntkj.com:8081', True
def connect_success(sock):
print(monitor._name, '连接成功->', sock.ws_url)
if monitor._temp_socket == sock:
sock.auto_reconnet = True # 新的连接成功后把断线重连开启
monitor._socket.quit() # 旧的断开
monitor._temp_socket = None # 清除临时变量
monitor.bind(sock) # 替换Sock
monitor.initialize() # 发送初始化消息
def set_new_server_func(sock):
sock.new_server_func = functools.partial(
control_server.post_terminal_info,
lambda: monitor._name,
lambda: monitor._socket.ws_url,
lambda: monitor.size,
lambda: monitor.version.name
)
sock.server_replace = server_replace
def run():
# init socket start
socket = MessageSocketClient(connect_success, message_receive)
set_new_server_func(socket)
conf = settings.get_server()
conf.update({'url': monitor._host, 'name': monitor._name, 'uuid': monitor._imei})
socket.connect(**conf)
monitor.bind(socket)
# init socket end
# init schedule start
sched = TornadoScheduler()
sched.add_job(monitor._socket.sendheartbeat, 'interval', seconds=60)
sched.add_job(monitor._socket.auto_move_server, 'interval', seconds=31)
sched.add_job(monitor.go_login, next_run_time=datetime.datetime.now() + datetime.timedelta(seconds=5))
sched.start()
# init schedule end
# message queue start
if Queue:
monitor.queue = Queue(monitor._name, monitor.execute).start()
# message queue end
ioloop.IOLoop.instance().start()