Как передать в данном примере объект в функцию в данном примере websocket.WebSocketApp?

Question

[MasterCopipaster]

как я в данном случае могу передать перменную c в строке c = Config() в функцию on_message что бы каждый раз когда WebSocketApp вызывает ее, я имел доступ к конфигу внутри неё не используя глобальные переменные

from modules.config import Config
import json
import websocket
import _thread
import time
import rel


def on_message(ws, message):
    print(message)


def on_error(ws, error):
    print(error)

def on_close(ws, close_status_code, close_msg):
    print("### closed ###")

def on_open(ws):
    print("Opened connection")

if __name__ == '__main__':
    c = Config()
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://echo.websocket.events",
                                on_open=on_open,
                                on_message=on_message,
                                on_error=on_error,
                                on_close=on_close)

    ws.run_forever(dispatcher=rel,
                   reconnect=5)  # Set dispatcher to automatic reconnection, 5 second reconnect delay if connection closed unexpectedly
    rel.signal(2, rel.abort)  # Keyboard Interrupt
    rel.dispatch()

Comments

[Алан Гибизов]

Любопытный вопрос.
Первое, что мне пришло в голову - попытаться добавить в класс WebSocketApp атрибут, который будет ссылаться на ваш конфиг. Или, если конфиг для каждого экземпляра должен быть уникальным, то не в класс, а в экземпляр.

Но как это лучше сделать, я не знаю. Я бы попытался «в лоб» создать новый атрибут, либо можно сделать наследника класса с новым атрибутом. И уже в экземпляре обращаться к этому атрибуту или ссылаться на этот атрибут класса.

Но, признаться, я так делать не пробовал и не могу даже предположить, какие грабли это может создать.

[Алан Гибизов]

Кстати, вот здесь есть любопытная схема с наследованием - посмотрите на устройство class BasePageElement - в нем не определен, но вроде как используется атрибут self.locator, а потом от класса наследуется другой класс и в нем этот атрибут класса определяется.
Схема не очевидная, и прямо тут не применима, но возможно наведет на какие-то мысли.

[MasterCopipaster]

Алан Гибизов, спасибо ознокомлюсь

Answer 1

[Александр Храпач]

Можно передать объект c в функцию on_message с использованием lambda-функции или partial-функции из модуля functools.

from modules.config import Config
import json
import websocket
import _thread
import time
import rel
from functools import partial

def on_message(c, ws, message):
    print(c.some_property)  # Пример использования свойства объекта Config
    print(message)

# Остальные функции on_error, on_close, on_open остаются без изменений

if __name__ == '__main__':
    c = Config()
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp("ws://echo.websocket.events",
                                on_open=on_open,
                                on_message=partial(on_message, c),  # Передача объекта c через partial-функцию
                                on_error=on_error,
                                on_close=on_close)

    ws.run_forever(dispatcher=rel,
                   reconnect=5)
    rel.signal(2, rel.abort)
    rel.dispatch()