【Python】【websocket請求/ws請求】

 

def websocket(url, headers_dict=None, params_dict=None, result: list = None):
    """
    websocket請求
    :param url:
    :param headers_dict:
    :param params_dict:
    :param result: 請求結果列表
    """
    print('websocket請求url：{}'.format(url))
    # step1 創建連接
    while True:
        try:
            if headers_dict is not None:
                ws = create_connection(url, header=headers_dict)
            else:
                ws = create_connection(url)
            break
        except Exception as e:
            print('連接錯誤:{}'.format(e))
            time.sleep(5)
    print('請求是否成功：{}'.format('成功' if ws.status == 101 else '失敗'))

    # step2：請求/獲取 響應
    while True:
        if params_dict is not None:
            params = params_dict
            ws.send(str(params))
        response = ws.recv()
        if result is not None:
            result.append(response)
        print(response)
    ws.close()