mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-11-26 19:41:11 +00:00
Python formatting
This commit is contained in:
parent
2fe9d1044a
commit
0caea7e9ff
2 changed files with 36 additions and 33 deletions
|
@ -18,11 +18,11 @@ class ActivityStream(ABC):
|
|||
|
||||
def stream_id(self, user):
|
||||
""" the redis key for this user's instance of this stream """
|
||||
return '{}-{}'.format(user.id, self.key)
|
||||
return "{}-{}".format(user.id, self.key)
|
||||
|
||||
def unread_id(self, user):
|
||||
""" the redis key for this user's unread count for this stream """
|
||||
return '{}-unread'.format(self.stream_id(user))
|
||||
return "{}-unread".format(self.stream_id(user))
|
||||
|
||||
def add_status(self, status):
|
||||
""" add a status to users' feeds """
|
||||
|
@ -38,25 +38,24 @@ class ActivityStream(ABC):
|
|||
# and go!
|
||||
pipeline.execute()
|
||||
|
||||
|
||||
def get_value(self, status): # pylint: disable=no-self-use
|
||||
""" the status id and the rank (ie, published date) """
|
||||
return {status.id: status.published_date.timestamp()}
|
||||
|
||||
|
||||
def get_activity_stream(self, user):
|
||||
""" load the ids for statuses to be displayed """
|
||||
# clear unreads for this feed
|
||||
r.set(self.unread_id(user), 0)
|
||||
|
||||
statuses = r.zrevrange(self.stream_id(user), 0, -1)
|
||||
return models.Status.objects.select_subclasses().filter(
|
||||
id__in=statuses
|
||||
).order_by('-published_date')
|
||||
|
||||
return (
|
||||
models.Status.objects.select_subclasses()
|
||||
.filter(id__in=statuses)
|
||||
.order_by("-published_date")
|
||||
)
|
||||
|
||||
def populate_stream(self, user):
|
||||
''' go from zero to a timeline '''
|
||||
""" go from zero to a timeline """
|
||||
pipeline = r.pipeline()
|
||||
statuses = self.stream_statuses(user)
|
||||
|
||||
|
@ -65,42 +64,41 @@ class ActivityStream(ABC):
|
|||
pipeline.zadd(stream_id, self.get_value(status))
|
||||
pipeline.execute()
|
||||
|
||||
|
||||
def stream_users(self, status): # pylint: disable=no-self-use
|
||||
""" given a status, what users should see it """
|
||||
# direct messages don't appeard in feeds.
|
||||
if status.privacy == 'direct':
|
||||
if status.privacy == "direct":
|
||||
return None
|
||||
|
||||
# everybody who could plausibly see this status
|
||||
audience = models.User.objects.filter(
|
||||
is_active=True,
|
||||
local=True # we only create feeds for users of this instance
|
||||
local=True, # we only create feeds for users of this instance
|
||||
).exclude(
|
||||
Q(id__in=status.user.blocks.all()) | Q(blocks=status.user) # not blocked
|
||||
)
|
||||
|
||||
# only visible to the poster's followers and tagged users
|
||||
if status.privacy == 'followers':
|
||||
if status.privacy == "followers":
|
||||
audience = audience.filter(
|
||||
Q(id=status.user.id) # if the user is the post's author
|
||||
| Q(following=status.user) # if the user is following the author
|
||||
)
|
||||
return audience
|
||||
|
||||
|
||||
def stream_statuses(self, user): # pylint: disable=no-self-use
|
||||
""" given a user, what statuses should they see on this stream """
|
||||
return privacy_filter(
|
||||
user,
|
||||
models.Status.objects.select_subclasses(),
|
||||
privacy_levels=["public", 'unlisted', 'followers'],
|
||||
privacy_levels=["public", "unlisted", "followers"],
|
||||
)
|
||||
|
||||
|
||||
class HomeStream(ActivityStream):
|
||||
""" users you follow """
|
||||
key = 'home'
|
||||
|
||||
key = "home"
|
||||
|
||||
def stream_users(self, status):
|
||||
audience = super().stream_users(status)
|
||||
|
@ -114,18 +112,19 @@ class HomeStream(ActivityStream):
|
|||
return privacy_filter(
|
||||
user,
|
||||
models.Status.objects.select_subclasses(),
|
||||
privacy_levels=["public", 'unlisted', 'followers'],
|
||||
following_only=True
|
||||
privacy_levels=["public", "unlisted", "followers"],
|
||||
following_only=True,
|
||||
)
|
||||
|
||||
|
||||
class LocalStream(ActivityStream):
|
||||
""" users you follow """
|
||||
key = 'local'
|
||||
|
||||
key = "local"
|
||||
|
||||
def stream_users(self, status):
|
||||
# this stream wants no part in non-public statuses
|
||||
if status.privacy != 'public':
|
||||
if status.privacy != "public":
|
||||
return None
|
||||
return super().stream_users(status)
|
||||
|
||||
|
@ -140,11 +139,12 @@ class LocalStream(ActivityStream):
|
|||
|
||||
class FederatedStream(ActivityStream):
|
||||
""" users you follow """
|
||||
key = 'federated'
|
||||
|
||||
key = "federated"
|
||||
|
||||
def stream_users(self, status):
|
||||
# this stream wants no part in non-public statuses
|
||||
if status.privacy != 'public':
|
||||
if status.privacy != "public":
|
||||
return None
|
||||
return super().stream_users(status)
|
||||
|
||||
|
@ -156,20 +156,23 @@ class FederatedStream(ActivityStream):
|
|||
)
|
||||
|
||||
|
||||
|
||||
streams = {
|
||||
'home': HomeStream(),
|
||||
'local': LocalStream(),
|
||||
'federated': FederatedStream(),
|
||||
"home": HomeStream(),
|
||||
"local": LocalStream(),
|
||||
"federated": FederatedStream(),
|
||||
}
|
||||
|
||||
|
||||
@receiver(signals.post_save)
|
||||
# pylint: disable=unused-argument
|
||||
def update_feeds(sender, instance, created, *args, **kwargs):
|
||||
""" add statuses to activity feeds """
|
||||
# we're only interested in new statuses that aren't dms
|
||||
if not created or not issubclass(sender, models.Status) or \
|
||||
instance.privacy == 'direct':
|
||||
if (
|
||||
not created
|
||||
or not issubclass(sender, models.Status)
|
||||
or instance.privacy == "direct"
|
||||
):
|
||||
return
|
||||
|
||||
for stream in streams.values():
|
||||
|
|
|
@ -28,7 +28,7 @@ class Feed(View):
|
|||
page = 1
|
||||
|
||||
if not tab in STREAMS:
|
||||
tab = 'home'
|
||||
tab = "home"
|
||||
|
||||
activities = activitystreams.streams[tab].get_activity_stream(request.user)
|
||||
|
||||
|
|
Loading…
Reference in a new issue