fixing auth for token

ereuse_devicehub/auth.py

@@ -3,12 +3,17 @@ from teal.auth import TokenAuth
 from teal.db import ResourceNotFound
 from werkzeug.exceptions import Unauthorized
 
-from ereuse_devicehub.resources.user.models import User
+from ereuse_devicehub.resources.user.models import User, Session
 
 
 class Auth(TokenAuth):
     def authenticate(self, token: str, *args, **kw) -> User:
         try:
-            return User.query.filter_by(token=token).one()
+            user = User.query.filter_by(token=token).first()
+            if user:
+                return user
+
+            ses = Session.query.filter_by(token=token).one()
+            return ses.user
         except (ResourceNotFound, DataError):
             raise Unauthorized('Provide a suitable token.')