plain.oauth
Let users log in with OAuth providers.
This library is intentionally minimal. It has no dependencies and a single database model. If you simply want users to log in with GitHub, Google, Twitter, etc. (and maybe use that access token for API calls), then this is the library for you.
There are three OAuth flows that it makes possible:
- Signup via OAuth (new user, new OAuth connection)
- Login via OAuth (existing user, existing OAuth connection)
- Connect/disconnect OAuth accounts to a user (existing user, new OAuth connection)
Usage
Install the package from PyPi:
pip install plain-oauth
Add plain.oauth to your INSTALLED_PACKAGES in settings.py:
INSTALLED_PACKAGES = [
...
"plain.oauth",
]
In your urls.py, include plain.oauth.urls:
urlpatterns = [
path("oauth/", include("plain.oauth.urls")),
...
]
Then run migrations:
python manage.py migrate plain.oauth
Create a new OAuth provider (or copy one from our examples):
# yourapp/oauth.py
import requests
from plain.oauth.providers import OAuthProvider, OAuthToken, OAuthUser
class ExampleOAuthProvider(OAuthProvider):
authorization_url = "https://example.com/login/oauth/authorize"
def get_oauth_token(self, *, code, request):
response = requests.post(
"https://example.com/login/oauth/token",
headers={
"Accept": "application/json",
},
data={
"client_id": self.get_client_id(),
"client_secret": self.get_client_secret(),
"code": code,
},
)
response.raise_for_status()
data = response.json()
return OAuthToken(
access_token=data["access_token"],
)
def get_oauth_user(self, *, oauth_token):
response = requests.get(
"https://example.com/api/user",
headers={
"Accept": "application/json",
"Authorization": f"token {oauth_token.access_token}",
},
)
response.raise_for_status()
data = response.json()
return OAuthUser(
# The provider ID is required
id=data["id"],
# And you can populate any of your User model fields with additional kwargs
email=data["email"],
username=data["username"],
)
Create your OAuth app/consumer on the provider's site (GitHub, Google, etc.).
When setting it up, you'll likely need to give it a callback URL.
In development this can be http://localhost:8000/oauth/github/callback/ (if you name it "github" like in the example below).
At the end you should get some sort of "client id" and "client secret" which you can then use in your settings.py:
OAUTH_LOGIN_PROVIDERS = {
"github": {
"class": "yourapp.oauth.GitHubOAuthProvider",
"kwargs": {
"client_id": environ["GITHUB_CLIENT_ID"],
"client_secret": environ["GITHUB_CLIENT_SECRET"],
# "scope" is optional, defaults to ""
# You can add other fields if you have additional kwargs in your class __init__
# def __init__(self, *args, custom_arg="default", **kwargs):
# self.custom_arg = custom_arg
# super().__init__(*args, **kwargs)
},
},
}
Then add a login button (which is a form using POST rather than a basic link, for security purposes):
<h1>Login</h1>
<form action="{% url 'oauth:login' 'github' %}" method="post">
{{ csrf_input }}
<button type="submit">Login with GitHub</button>
</form>
Depending on your URL and provider names,
your OAuth callback will be something like https://example.com/oauth/{provider}/callback/.
That's pretty much it!
Advanced usage
Handling OAuth errors
The most common error you'll run into is if an existing user clicks a login button, but they haven't yet connected that provider to their account. For security reasons, the required flow here is that the user actually logs in with another method (however they signed up) and then connects the OAuth provider from a settings page.
For this error (and a couple others),
there is an error template that is rendered.
You can customize this by copying oauth/error.html to one of your own template directories:
{% extends "base.html" %}
{% block content %}
<h1>OAuth Error</h1>
<p>{{ oauth_error }}</p>
{% endblock %}
Connecting and disconnecting OAuth accounts
To connect and disconnect OAuth accounts, you can add a series of forms to a user/profile settings page. Here's an very basic example:
{% extends "base.html" %}
{% block content %}
Hello {{ request.user }}!
<h2>Existing connections</h2>
<ul>
{% for connection in request.user.oauth_connections.all %}
<li>
{{ connection.provider_key }} [ID: {{ connection.provider_user_id }}]
<form action="{% url 'oauth:disconnect' connection.provider_key %}" method="post">
{{ csrf_input }}
<input type="hidden" name="provider_user_id" value="{{ connection.provider_user_id }}">
<button type="submit">Disconnect</button>
</form>
</li>
{% endfor %}
</ul>
<h2>Add a connection</h2>
<ul>
{% for provider_key in oauth_provider_keys %}
<li>
{{ provider_key}}
<form action="{% url 'oauth:connect' provider_key %}" method="post">
{{ csrf_input }}
<button type="submit">Connect</button>
</form>
</li>
{% endfor %}
</ul>
{% endblock %}
The get_provider_keys function can help populate the list of options:
from plain.oauth.providers import get_provider_keys
class ExampleView(TemplateView):
template_name = "index.html"
def get_context(self, **kwargs):
context = super().get_context(**kwargs)
context["oauth_provider_keys"] = get_provider_keys()
return context
Using a saved access token
import requests
# Get the OAuth connection for a user
connection = user.oauth_connections.get(provider_key="github")
# If the token can expire, check and refresh it
if connection.access_token_expired():
connection.refresh_access_token()
# Use the token in an API call
token = connection.access_token
response = requests.get(...)
Using the Django system check
This library comes with a Django system check to ensure you don't remove a provider from settings.py that is still in use in your database.
You do need to specify the --database for this to run when using the check command by itself:
python manage.py check --database default
FAQs
How is this different from Django OAuth libraries?
The short answer is that it does less.
In django-allauth
(maybe the most popular alternative)
you get all kinds of other features like managing multiple email addresses,
email verification,
a long list of supported providers,
and a whole suite of forms/urls/views/templates/signals/tags.
And in my experience,
it's too much.
It often adds more complexity to your app than you actually need (or want) and honestly it can just be a lot to wrap your head around.
Personally, I don't like the way that your OAuth settings are stored in the database vs when you use settings.py,
and the implications for doing it one way or another.
The other popular OAuth libraries have similar issues, and I think their weight outweighs their usefulness for 80% of the use cases.
Why aren't providers included in the library itself?
One thing you'll notice is that we don't have a long list of pre-configured providers in this library. Instead, we have some examples (which you can usually just copy, paste, and use) and otherwise encourage you to wire up the provider yourself. Often times all this means is finding the two OAuth URLs ("oauth/authorize" and "oauth/token") in their docs, and writing two class methods that do the actual work of getting the user's data (which is often customized anyway).
We've written examples for the following providers:
Just copy that code and paste it in your project. Tweak as necessary!
This might sound strange at first. But in the long run we think it's actually much more maintainable for both us (as library authors) and you (as app author). If something breaks with a provider, you can fix it immediately! You don't need to try to run changes through us or wait for an upstream update. You're welcome to contribute an example to this repo, and there won't be an expectation that it "works perfectly for every use case until the end of time".
Redirect/callback URL mismatch in local development?
If you're doing local development through a proxy/tunnel like ngrok,
then the callback URL might be automatically built as http instead of https.
This is the Django setting you're probably looking for:
SECURE_PROXY_SSL_HEADER = ("HTTP_X_FORWARDED_PROTO", "https")
1import datetime
2import secrets
3from typing import Any
4from urllib.parse import urlencode
5
6from plain.auth import login as auth_login
7from plain.http import HttpRequest, Response, ResponseRedirect
8from plain.runtime import settings
9from plain.urls import reverse
10from plain.utils.crypto import get_random_string
11from plain.utils.module_loading import import_string
12
13from .exceptions import OAuthError, OAuthStateMismatchError
14from .models import OAuthConnection
15
16SESSION_STATE_KEY = "plainoauth_state"
17SESSION_NEXT_KEY = "plainoauth_next"
18
19
20class OAuthToken:
21 def __init__(
22 self,
23 *,
24 access_token: str,
25 refresh_token: str = "",
26 access_token_expires_at: datetime.datetime = None,
27 refresh_token_expires_at: datetime.datetime = None,
28 ):
29 self.access_token = access_token
30 self.refresh_token = refresh_token
31 self.access_token_expires_at = access_token_expires_at
32 self.refresh_token_expires_at = refresh_token_expires_at
33
34
35class OAuthUser:
36 def __init__(self, *, id: str, **user_model_fields: dict):
37 self.id = id # ID on the provider's system
38 self.user_model_fields = user_model_fields
39
40 def __str__(self):
41 if "email" in self.user_model_fields:
42 return self.user_model_fields["email"]
43 if "username" in self.user_model_fields:
44 return self.user_model_fields["username"]
45 return str(self.id)
46
47
48class OAuthProvider:
49 authorization_url = ""
50
51 def __init__(
52 self,
53 *,
54 # Provided automatically
55 provider_key: str,
56 # Required as kwargs in OAUTH_LOGIN_PROVIDERS setting
57 client_id: str,
58 client_secret: str,
59 # Not necessarily required, but commonly used
60 scope: str = "",
61 ):
62 self.provider_key = provider_key
63 self.client_id = client_id
64 self.client_secret = client_secret
65 self.scope = scope
66
67 def get_authorization_url_params(self, *, request: HttpRequest) -> dict:
68 return {
69 "redirect_uri": self.get_callback_url(request=request),
70 "client_id": self.get_client_id(),
71 "scope": self.get_scope(),
72 "state": self.generate_state(),
73 "response_type": "code",
74 }
75
76 def refresh_oauth_token(self, *, oauth_token: OAuthToken) -> OAuthToken:
77 raise NotImplementedError()
78
79 def get_oauth_token(self, *, code: str, request: HttpRequest) -> OAuthToken:
80 raise NotImplementedError()
81
82 def get_oauth_user(self, *, oauth_token: OAuthToken) -> OAuthUser:
83 raise NotImplementedError()
84
85 def get_authorization_url(self, *, request: HttpRequest) -> str:
86 return self.authorization_url
87
88 def get_client_id(self) -> str:
89 return self.client_id
90
91 def get_client_secret(self) -> str:
92 return self.client_secret
93
94 def get_scope(self) -> str:
95 return self.scope
96
97 def get_callback_url(self, *, request: HttpRequest) -> str:
98 url = reverse("oauth:callback", kwargs={"provider": self.provider_key})
99 return request.build_absolute_uri(url)
100
101 def generate_state(self) -> str:
102 return get_random_string(length=32)
103
104 def check_request_state(self, *, request: HttpRequest) -> None:
105 if error := request.GET.get("error"):
106 raise OAuthError(error)
107
108 state = request.GET["state"]
109 expected_state = request.session.pop(SESSION_STATE_KEY)
110 if not secrets.compare_digest(state, expected_state):
111 raise OAuthStateMismatchError()
112
113 def handle_login_request(
114 self, *, request: HttpRequest, redirect_to: str = ""
115 ) -> Response:
116 authorization_url = self.get_authorization_url(request=request)
117 authorization_params = self.get_authorization_url_params(request=request)
118
119 if "state" in authorization_params:
120 # Store the state in the session so we can check on callback
121 request.session[SESSION_STATE_KEY] = authorization_params["state"]
122
123 # Store next url in session so we can get it on the callback request
124 if redirect_to:
125 request.session[SESSION_NEXT_KEY] = redirect_to
126 elif "next" in request.POST:
127 request.session[SESSION_NEXT_KEY] = request.POST["next"]
128
129 # Sort authorization params for consistency
130 sorted_authorization_params = sorted(authorization_params.items())
131 redirect_url = authorization_url + "?" + urlencode(sorted_authorization_params)
132 return ResponseRedirect(redirect_url)
133
134 def handle_connect_request(
135 self, *, request: HttpRequest, redirect_to: str = ""
136 ) -> Response:
137 return self.handle_login_request(request=request, redirect_to=redirect_to)
138
139 def handle_disconnect_request(self, *, request: HttpRequest) -> Response:
140 provider_user_id = request.POST["provider_user_id"]
141 connection = OAuthConnection.objects.get(
142 provider_key=self.provider_key, provider_user_id=provider_user_id
143 )
144 connection.delete()
145 redirect_url = self.get_disconnect_redirect_url(request=request)
146 return ResponseRedirect(redirect_url)
147
148 def handle_callback_request(self, *, request: HttpRequest) -> Response:
149 self.check_request_state(request=request)
150
151 oauth_token = self.get_oauth_token(code=request.GET["code"], request=request)
152 oauth_user = self.get_oauth_user(oauth_token=oauth_token)
153
154 if request.user:
155 connection = OAuthConnection.connect(
156 user=request.user,
157 provider_key=self.provider_key,
158 oauth_token=oauth_token,
159 oauth_user=oauth_user,
160 )
161 user = connection.user
162 else:
163 connection = OAuthConnection.get_or_create_user(
164 provider_key=self.provider_key,
165 oauth_token=oauth_token,
166 oauth_user=oauth_user,
167 )
168
169 user = connection.user
170
171 self.login(request=request, user=user)
172
173 redirect_url = self.get_login_redirect_url(request=request)
174 return ResponseRedirect(redirect_url)
175
176 def login(self, *, request: HttpRequest, user: Any) -> Response:
177 auth_login(request=request, user=user)
178
179 def get_login_redirect_url(self, *, request: HttpRequest) -> str:
180 return request.session.pop(SESSION_NEXT_KEY, "/")
181
182 def get_disconnect_redirect_url(self, *, request: HttpRequest) -> str:
183 return request.POST.get("next", "/")
184
185
186def get_oauth_provider_instance(*, provider_key: str) -> OAuthProvider:
187 OAUTH_LOGIN_PROVIDERS = getattr(settings, "OAUTH_LOGIN_PROVIDERS", {})
188 provider_class_path = OAUTH_LOGIN_PROVIDERS[provider_key]["class"]
189 provider_class = import_string(provider_class_path)
190 provider_kwargs = OAUTH_LOGIN_PROVIDERS[provider_key].get("kwargs", {})
191 return provider_class(provider_key=provider_key, **provider_kwargs)
192
193
194def get_provider_keys() -> list[str]:
195 return list(getattr(settings, "OAUTH_LOGIN_PROVIDERS", {}).keys())