public-mirrors/NewsBlur
1
0
Fork 0
mirror of https://github.com/samuelclay/NewsBlur.git synced 2025-09-18 21:50:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixing missing usernames in msocialprofile, due to botched deleted user refactor.

Browse source
This commit is contained in:
Samuel Clay 2021-08-09 11:36:56 -04:00
parent 9bd863f74f
commit d67d591973
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
apps/social/models.py
View file

@ -161,6 +161,9 @@ class MSocialProfile(mongo.Document):
profile = cls.objects.create(user_id=user_id)
profile.save()
if not profile.username:
profile.save()
return profile
@property
@ -171,6 +174,8 @@ class MSocialProfile(mongo.Document):
return None
def save(self, *args, **kwargs):
if not self.username:
self.import_user_fields()
if not self.subscription_count:
self.count_follows(skip_save=True)
if self.bio and len(self.bio) > MSocialProfile.bio.max_length:
@ -433,6 +438,11 @@ class MSocialProfile(mongo.Document):
return [u for u in self.follower_user_ids if u != self.user_id]
return self.follower_user_ids
def import_user_fields(self):
user = User.objects.get(pk=self.user_id)
self.username = user.username
self.email = user.email
def count_follows(self, skip_save=False):
self.subscription_count = UserSubscription.objects.filter(user__pk=self.user_id).count()
self.shared_stories_count = MSharedStory.objects.filter(user_id=self.user_id).count()