Add user.get_token() helper

HumanSignal · Nov 4, 2024 · 56e1493 · 56e1493
1 parent 04ac6fe
commit 56e1493
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/label_studio/users/models.py b/label_studio/users/models.py
@@ -177,6 +177,9 @@ def get_full_name(self):
     def get_short_name(self):
         """Return the short name for the user."""
         return self.first_name
+
+    def get_token(self) -> Token:
+        return Token.objects.filter(user=self).first()
 
     def reset_token(self) -> Token:
         Token.objects.filter(user=self).delete()