mirror of
https://github.com/jointakahe/takahe.git
synced 2024-11-29 10:40:59 +00:00
parent
33b24d99bc
commit
427744f5a7
1 changed files with 46 additions and 14 deletions
|
@ -1,3 +1,5 @@
|
||||||
|
from collections.abc import Iterable
|
||||||
|
|
||||||
from django.db import models, transaction
|
from django.db import models, transaction
|
||||||
from django.utils import timezone
|
from django.utils import timezone
|
||||||
|
|
||||||
|
@ -31,23 +33,12 @@ class PostInteractionStates(StateGraph):
|
||||||
"""
|
"""
|
||||||
interaction = await instance.afetch_full()
|
interaction = await instance.afetch_full()
|
||||||
# Boost: send a copy to all people who follow this user (limiting
|
# Boost: send a copy to all people who follow this user (limiting
|
||||||
# to just local follows if it's a remote post)
|
# to just local follows if it's a remote boost)
|
||||||
if interaction.type == interaction.Types.boost:
|
if interaction.type == interaction.Types.boost:
|
||||||
async for follow in interaction.identity.inbound_follows.filter(
|
for target in await interaction.aget_boost_targets():
|
||||||
boosts=True
|
|
||||||
).select_related("source", "target"):
|
|
||||||
if interaction.post.local or follow.source.local:
|
|
||||||
await FanOut.objects.acreate(
|
await FanOut.objects.acreate(
|
||||||
type=FanOut.Types.interaction,
|
type=FanOut.Types.interaction,
|
||||||
identity_id=follow.source_id,
|
identity=target,
|
||||||
subject_post=interaction.post,
|
|
||||||
subject_post_interaction=interaction,
|
|
||||||
)
|
|
||||||
# And one to the post's author, if the booster is local or they are
|
|
||||||
if interaction.identity.local or interaction.post.local:
|
|
||||||
await FanOut.objects.acreate(
|
|
||||||
type=FanOut.Types.interaction,
|
|
||||||
identity_id=interaction.post.author_id,
|
|
||||||
subject_post=interaction.post,
|
subject_post=interaction.post,
|
||||||
subject_post_interaction=interaction,
|
subject_post_interaction=interaction,
|
||||||
)
|
)
|
||||||
|
@ -224,6 +215,47 @@ class PostInteraction(StatorModel):
|
||||||
"identity", "post", "post__author"
|
"identity", "post", "post__author"
|
||||||
).aget(pk=self.pk)
|
).aget(pk=self.pk)
|
||||||
|
|
||||||
|
async def aget_boost_targets(self) -> Iterable[Identity]:
|
||||||
|
"""
|
||||||
|
Returns an iterable with Identities of followers that have unique
|
||||||
|
shared_inbox among each other to be used as target to the boost
|
||||||
|
"""
|
||||||
|
# Start including the post author
|
||||||
|
targets = {self.post.author}
|
||||||
|
|
||||||
|
# Include all followers that are following the boosts
|
||||||
|
async for follow in self.identity.inbound_follows.active().filter(
|
||||||
|
boosts=True
|
||||||
|
).select_related("source"):
|
||||||
|
targets.add(follow.source)
|
||||||
|
|
||||||
|
# Fetch the full blocks and remove them as targets
|
||||||
|
async for block in self.identity.outbound_blocks.active().filter(
|
||||||
|
mute=False
|
||||||
|
).select_related("target"):
|
||||||
|
try:
|
||||||
|
targets.remove(block.target)
|
||||||
|
except KeyError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
deduped_targets = set()
|
||||||
|
shared_inboxes = set()
|
||||||
|
for target in targets:
|
||||||
|
if target.local:
|
||||||
|
# Local targets always gets the boosts
|
||||||
|
# despite its creator locality
|
||||||
|
deduped_targets.add(target)
|
||||||
|
elif self.identity.local:
|
||||||
|
# Dedupe the targets based on shared inboxes
|
||||||
|
# (we only keep one per shared inbox)
|
||||||
|
if not target.shared_inbox_uri:
|
||||||
|
deduped_targets.add(target)
|
||||||
|
elif target.shared_inbox_uri not in shared_inboxes:
|
||||||
|
shared_inboxes.add(target.shared_inbox_uri)
|
||||||
|
deduped_targets.add(target)
|
||||||
|
|
||||||
|
return deduped_targets
|
||||||
|
|
||||||
### Create helpers ###
|
### Create helpers ###
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
|
Loading…
Reference in a new issue