Skip to content

Commit f72fd08

Browse files
committed
refactor: minor cleanup
1 parent b7ef547 commit f72fd08

2 files changed

Lines changed: 14 additions & 11 deletions

File tree

raven/omni_channel_chat/api/webhooks.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,6 @@ def handle() -> Response:
3333
"""
3434
slug = extract_provider_slug()
3535
provider = get_omni_channel_chat_provider(slug=slug)
36-
return provider.handle_frappe_api(
37-
callback=OmniChannelRavenConnector(provider=provider).receive_from_provider
38-
)
36+
connector = OmniChannelRavenConnector(provider=provider)
37+
38+
return provider.handle_frappe_api(callback=connector.receive_from_provider)

raven/omni_channel_chat/omni_channel_raven_connector.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@
55

66
from raven.omni_channel_chat.models.message import (
77
BaseMessage,
8+
FileContent,
89
FileMessage,
910
FileUrl,
10-
FileContent,
1111
ImageMessage,
1212
SenderInfo,
1313
TextMessage,
@@ -47,7 +47,9 @@ def _build_outbound_message(
4747
) -> BaseMessage:
4848
msg_type = raven_message.message_type
4949
if msg_type == "Text":
50-
return TextMessage(provider=provider_name, user_id="", text=raven_message.content or "", sender=sender)
50+
return TextMessage(
51+
provider=provider_name, user_id="", text=raven_message.content or "", sender=sender
52+
)
5153
file_url = raven_message.file
5254
if file_url and file_url.startswith("/"):
5355
file_url = get_url(file_url)
@@ -62,7 +64,6 @@ def _build_outbound_message(
6264
raise ValueError(f"Unsupported outbound message type: {msg_type}")
6365

6466

65-
6667
class OmniChannelRavenConnector:
6768
"""Bridges Raven and an external omni-channel provider.
6869
@@ -195,7 +196,7 @@ def handle_webhook(self, body: bytes, headers: dict) -> None:
195196
for message in messages:
196197
self.receive_from_provider(message)
197198

198-
def receive_from_provider(self, message: BaseMessage) -> "RavenChannel":
199+
def receive_from_provider(self, message: BaseMessage) -> None:
199200
"""Inbound: turn a provider webhook payload into a Raven message.
200201
201202
Creates the Frappe user, Raven user, and channel on first contact,
@@ -210,9 +211,9 @@ def receive_from_provider(self, message: BaseMessage) -> "RavenChannel":
210211
raven_channel = self._get_or_create_channel(raven_user=raven_user)
211212
self._save_inbound_message(raven_channel=raven_channel, message=message)
212213

213-
return raven_channel
214-
215-
def _save_inbound_message(self, *, raven_channel: "RavenChannel", message: BaseMessage) -> None:
214+
def _save_inbound_message(
215+
self, *, raven_channel: "RavenChannel", message: BaseMessage
216+
) -> None:
216217
doc = frappe.new_doc(doctype="Raven Message")
217218
doc.update(
218219
{
@@ -225,7 +226,9 @@ def _save_inbound_message(self, *, raven_channel: "RavenChannel", message: BaseM
225226
)
226227
if isinstance(message, TextMessage):
227228
doc.text = message.text
228-
elif isinstance(message, (ImageMessage, FileMessage)) and isinstance(message.file, FileContent):
229+
elif isinstance(message, (ImageMessage, FileMessage)) and isinstance(
230+
message.file, FileContent
231+
):
229232
file_doc = frappe.get_doc(
230233
{
231234
"doctype": "File",

0 commit comments

Comments
 (0)