FreakU
/
takahe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Catch all the subtypes too

This commit is contained in:
Andrew Godwin 2023-07-07 16:39:02 -06:00
parent 2fda9ad2b4
commit 933f6660d5
1 changed files with 5 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
users/models/inbox_message.py
View File

@ -69,9 +69,7 @@ class InboxMessageStates(StateGraph):
# It's a string object, but these will only be for Follows
Follow.handle_accept_ap(instance.message)
case unknown:
raise ValueError(
f"Cannot handle activity of type accept.{unknown}"
)
return cls.errored
case "reject":
match instance.message_object_type:
case "follow":
@ -80,9 +78,7 @@ class InboxMessageStates(StateGraph):
# It's a string object, but these will only be for Follows
Follow.handle_reject_ap(instance.message)
case unknown:
raise ValueError(
f"Cannot handle activity of type reject.{unknown}"
)
return cls.errored
case "undo":
match instance.message_object_type:
case "follow":
@ -97,9 +93,7 @@ class InboxMessageStates(StateGraph):
# We're ignoring emoji reactions for now
pass
case unknown:
raise ValueError(
f"Cannot handle activity of type undo.{unknown}"
)
return cls.errored
case "delete":
# If there is no object type, we need to see if it's a profile or a post
if not isinstance(instance.message["object"], dict):
@ -121,9 +115,7 @@ class InboxMessageStates(StateGraph):
case "note":
Post.handle_delete_ap(instance.message)
case unknown:
raise ValueError(
f"Cannot handle activity of type delete.{unknown}"
)
return cls.errored
case "add":
PostInteraction.handle_add_ap(instance.message)
case "remove":
@ -150,9 +142,7 @@ class InboxMessageStates(StateGraph):
instance.message["object"]
)
case unknown:
raise ValueError(
f"Cannot handle activity of type __internal__.{unknown}"
)
return cls.errored
case unknown:
return cls.errored
return cls.processed