Use default avatar for User/Orgs in OAuth services

Author: humitos

readthedocs/oauth/services/base.py

@@ -31,6 +31,9 @@ class Service(object):
     adapter = None
     url_pattern = None
 
+    default_user_avatar_url = settings.OAUTH_AVATAR_USER_DEFAULT_URL
+    default_org_avatar_url = settings.OAUTH_AVATAR_ORG_DEFAULT_URL
+
     def __init__(self, user, account):
         self.session = None
         self.user = user

readthedocs/oauth/services/bitbucket.py

@@ -136,6 +136,8 @@ def create_repository(self, fields, privacy=DEFAULT_PRIVACY_LEVEL,
 
             avatar_url = fields['links']['avatar']['href'] or ''
             repo.avatar_url = re.sub(r'\/16\/$', r'/32/', avatar_url)
+            if not repo.avatar_url:
+                repo.avatar_url = self.default_user_avatar_url
 
             repo.json = json.dumps(fields)
             repo.save()
@@ -157,6 +159,8 @@ def create_organization(self, fields):
         organization.name = fields.get('display_name')
         organization.email = fields.get('email')
         organization.avatar_url = fields['links']['avatar']['href']
+        if not organization.avatar_url:
+            organization.avatar_url = self.default_org_avatar_url
         organization.url = fields['links']['html']['href']
         organization.json = json.dumps(fields)
         organization.account = self.account

readthedocs/oauth/services/github.py

@@ -114,6 +114,8 @@ def create_repository(self, fields, privacy=DEFAULT_PRIVACY_LEVEL,
             repo.vcs = 'git'
             repo.account = self.account
             repo.avatar_url = fields.get('owner', {}).get('avatar_url')
+            if not repo.avatar_url:
+                repo.avatar_url = self.default_user_avatar_url
             repo.json = json.dumps(fields)
             repo.save()
             return repo
@@ -143,6 +145,8 @@ def create_organization(self, fields):
         organization.name = fields.get('name')
         organization.email = fields.get('email')
         organization.avatar_url = fields.get('avatar_url')
+        if not organization.avatar_url:
+            organization.avatar_url = self.default_org_avatar_url
         organization.json = json.dumps(fields)
         organization.account = self.account
         organization.save()

readthedocs/oauth/services/gitlab.py

@@ -162,7 +162,10 @@ def create_repository(
 
             owner = fields.get('owner') or {}
             repo.avatar_url = (
-                fields.get('avatar_url') or owner.get('avatar_url'))
+                fields.get('avatar_url') or owner.get('avatar_url')
+            )
+            if not repo.avatar_url:
+                repo.avatar_url = self.default_user_avatar_url
 
             repo.json = json.dumps(fields)
             repo.save()
@@ -201,6 +204,8 @@ def create_organization(self, fields):
             path=fields.get('path'),
         )
         organization.avatar_url = fields.get('avatar_url')
+        if not organization.avatar_url:
+            organization.avatar_url = self.default_user_avatar_url
         organization.json = json.dumps(fields)
         organization.save()
         return organization

readthedocs/settings/base.py

@@ -314,6 +314,8 @@ def INSTALLED_APPS(self):  # noqa
     # Misc application settings
     GLOBAL_ANALYTICS_CODE = 'UA-17997319-1'
     GRAVATAR_DEFAULT_IMAGE = 'https://media.readthedocs.org/images/silhouette.png'  # NOQA
+    OAUTH_AVATAR_USER_DEFAULT_URL = GRAVATAR_DEFAULT_IMAGE
+    OAUTH_AVATAR_ORG_DEFAULT_URL = GRAVATAR_DEFAULT_IMAGE
     RESTRICTEDSESSIONS_AUTHED_ONLY = True
     RESTRUCTUREDTEXT_FILTER_SETTINGS = {
         'cloak_email_addresses': True,