Handle cases when the user does not have site added yet

This commit is contained in:
shortcutme 2018-04-18 22:06:18 +02:00
parent 40cf4f2c2a
commit 6b33ad8a32
No known key found for this signature in database
GPG Key ID: 5B63BAE6CB9613AE
1 changed files with 1 additions and 1 deletions

View File

@ -10,7 +10,7 @@ from Debug import Debug
class UiWebsocketPlugin(object):
def formatSiteInfo(self, site, create_user=True):
site_info = super(UiWebsocketPlugin, self).formatSiteInfo(site, create_user=create_user)
feed_following = self.user.sites[site.address].get("follow", None)
feed_following = self.user.sites.get(site.address, {}).get("follow", None)
if feed_following == None:
site_info["feed_follow_num"] = None
else: