use social auth model

gauge-sh · Jan 7, 2025 · 4389a56 · 4389a56
1 parent 1b9f3fe
commit 4389a56
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/sentry/users/services/usersocialauth/model.py b/src/sentry/users/services/usersocialauth/model.py
@@ -9,6 +9,7 @@
 
 from sentry.hybridcloud.rpc import RpcModel
 from social_auth.backends import BaseAuth
+from social_auth.models import UserSocialAuth
 from social_auth.utils import get_backend, tokens
 
 
@@ -26,6 +27,8 @@ def get_backend(self) -> type[BaseAuth] | None:
     def tokens(self) -> dict[str, Any]:
         return tokens(instance=self)
 
+    def auth(self) -> "UserSocialAuth":
+        return UserSocialAuth.objects.get(id=self.id).auth()
 
 class UserSocialAuthFilterArgs(TypedDict, total=False):
     id: int