*(minor): small refactor

passbook/sources/oauth/models.py

@@ -0,0 +1,149 @@
+"""OAuth Client models"""
+
+from django.db import models
+from django.urls import reverse, reverse_lazy
+from django.utils.translation import gettext as _
+
+from passbook.core.models import Source, UserSourceConnection
+from passbook.sources.oauth.clients import get_client
+
+
+class OAuthSource(Source):
+    """Configuration for OAuth provider."""
+
+    provider_type = models.CharField(max_length=255)
+    request_token_url = models.CharField(blank=True, max_length=255)
+    authorization_url = models.CharField(max_length=255)
+    access_token_url = models.CharField(max_length=255)
+    profile_url = models.CharField(max_length=255)
+    consumer_key = models.TextField()
+    consumer_secret = models.TextField()
+
+    form = 'passbook.sources.oauth.forms.OAuthSourceForm'
+
+    @property
+    def is_link(self):
+        return True
+
+    @property
+    def get_login_button(self):
+        url = reverse_lazy('passbook_sources_oauth:oauth-client-login',
+                           kwargs={'source_slug': self.slug})
+        return url, self.provider_type, self.name
+
+    @property
+    def additional_info(self):
+        return "Callback URL: <pre>%s</pre>" % \
+            reverse_lazy('passbook_sources_oauth:oauth-client-callback',
+                         kwargs={'source_slug': self.slug})
+
+    def has_user_settings(self):
+        """Entrypoint to integrate with User settings. Can either return False if no
+        user settings are available, or a tuple or string, string, string where the first string
+        is the name the item has, the second string is the icon and the third is the view-name."""
+        icon_type = self.provider_type
+        if icon_type == 'azure ad':
+            icon_type = 'windows'
+        icon_class = 'fa fa-%s' % icon_type
+        view_name = 'passbook_sources_oauth:oauth-client-user'
+        return self.name, icon_class, reverse((view_name), kwargs={
+            'source_slug': self.slug
+        })
+
+    class Meta:
+
+        verbose_name = _('Generic OAuth Source')
+        verbose_name_plural = _('Generic OAuth Sources')
+
+
+class GitHubOAuthSource(OAuthSource):
+    """Abstract subclass of OAuthSource to specify GitHub Form"""
+
+    form = 'passbook.sources.oauth.forms.GitHubOAuthSourceForm'
+
+    class Meta:
+
+        abstract = True
+        verbose_name = _('GitHub OAuth Source')
+        verbose_name_plural = _('GitHub OAuth Sources')
+
+
+class TwitterOAuthSource(OAuthSource):
+    """Abstract subclass of OAuthSource to specify Twitter Form"""
+
+    form = 'passbook.sources.oauth.forms.TwitterOAuthSourceForm'
+
+    class Meta:
+
+        abstract = True
+        verbose_name = _('Twitter OAuth Source')
+        verbose_name_plural = _('Twitter OAuth Sources')
+
+
+class FacebookOAuthSource(OAuthSource):
+    """Abstract subclass of OAuthSource to specify Facebook Form"""
+
+    form = 'passbook.sources.oauth.forms.FacebookOAuthSourceForm'
+
+    class Meta:
+
+        abstract = True
+        verbose_name = _('Facebook OAuth Source')
+        verbose_name_plural = _('Facebook OAuth Sources')
+
+
+class DiscordOAuthSource(OAuthSource):
+    """Abstract subclass of OAuthSource to specify Discord Form"""
+
+    form = 'passbook.sources.oauth.forms.DiscordOAuthSourceForm'
+
+    class Meta:
+
+        abstract = True
+        verbose_name = _('Discord OAuth Source')
+        verbose_name_plural = _('Discord OAuth Sources')
+
+
+class GoogleOAuthSource(OAuthSource):
+    """Abstract subclass of OAuthSource to specify Google Form"""
+
+    form = 'passbook.sources.oauth.forms.GoogleOAuthSourceForm'
+
+    class Meta:
+
+        abstract = True
+        verbose_name = _('Google OAuth Source')
+        verbose_name_plural = _('Google OAuth Sources')
+
+
+class AzureADOAuthSource(OAuthSource):
+    """Abstract subclass of OAuthSource to specify AzureAD Form"""
+
+    form = 'passbook.sources.oauth.forms.AzureADOAuthSourceForm'
+
+    class Meta:
+
+        abstract = True
+        verbose_name = _('Azure AD OAuth Source')
+        verbose_name_plural = _('Azure AD OAuth Sources')
+
+
+class UserOAuthSourceConnection(UserSourceConnection):
+    """Authorized remote OAuth provider."""
+
+    identifier = models.CharField(max_length=255)
+    access_token = models.TextField(blank=True, null=True, default=None)
+
+    def save(self, *args, **kwargs):
+        self.access_token = self.access_token or None
+        super().save(*args, **kwargs)
+
+    @property
+    def api_client(self):
+        """Get API Client"""
+        return get_client(self.source, self.access_token or '')
+
+    class Meta:
+
+        verbose_name = _('User OAuth Source Connection')
+        verbose_name_plural = _('User OAuth Source Connections')