You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.5 KiB
40 lines
1.5 KiB
import json |
|
# from update.views import get_cinema_update_status |
|
from channels.generic.websocket import WebsocketConsumer |
|
from django_redis import get_redis_connection |
|
|
|
|
|
class UpdateConsumer(WebsocketConsumer): |
|
def connect(self): |
|
self.accept() |
|
self.send('你已经连接成功') |
|
|
|
def receive(self, text_data=None, bytes_data=None): |
|
|
|
redis_conn = get_redis_connection() |
|
|
|
if text_data is None: |
|
self.send('你发了啥') |
|
else: |
|
data = json.loads(text_data) |
|
# cinema_update_status = get_cinema_update_status() |
|
if data.get('msg', False): |
|
if data['msg'] == 'ping': |
|
self.send(json.dumps({'msg': 'pong'})) |
|
if data.get('finish', False): |
|
ip = data['finish'] |
|
redis_key = f'cinema_update_status_{ip}' |
|
if redis_conn.exists(redis_key): |
|
redis_conn.delete(redis_key) |
|
data = json.loads(redis_conn.get(redis_key)) |
|
print('ws任务完成消息', data) |
|
|
|
if data.get('ip', False): |
|
ip = data['ip'] |
|
redis_key = f'cinema_update_status_{ip}' |
|
# print('ws发送消息前', cinema_update_status) |
|
data = json.loads(redis_conn.get(redis_key)) |
|
print('ws发送消息前', data) |
|
self.send(json.dumps({'status': {'ip': ip, 'result': data}})) |
|
else: |
|
self.send(text_data)
|
|
|