mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-10-01 08:10:13 +03:00
fix(typing): resolve mypy errors in api.routes.controller.dependencies.authentication
This commit is contained in:
parent
dfea615248
commit
a3cb73da3d
@ -55,7 +55,7 @@ async def get_user_from_token(
|
|||||||
user_repo: UsersRepository = Depends(get_repository(UsersRepository)),
|
user_repo: UsersRepository = Depends(get_repository(UsersRepository)),
|
||||||
api_keys_repo: ApiKeysRepository = Depends(get_repository(ApiKeysRepository)),
|
api_keys_repo: ApiKeysRepository = Depends(get_repository(ApiKeysRepository)),
|
||||||
token: Optional[str] = Query(None, include_in_schema=False),
|
token: Optional[str] = Query(None, include_in_schema=False),
|
||||||
) -> schemas.User:
|
) -> models.User:
|
||||||
|
|
||||||
if bearer_token:
|
if bearer_token:
|
||||||
# bearer token is used first, then any token passed as a URL parameter
|
# bearer token is used first, then any token passed as a URL parameter
|
||||||
@ -169,7 +169,7 @@ async def get_current_active_user_from_websocket(
|
|||||||
websocket: WebSocket,
|
websocket: WebSocket,
|
||||||
token: str = Query(...),
|
token: str = Query(...),
|
||||||
user_repo: UsersRepository = Depends(get_repository(UsersRepository)),
|
user_repo: UsersRepository = Depends(get_repository(UsersRepository)),
|
||||||
) -> Optional[schemas.User]:
|
) -> Optional[models.User]:
|
||||||
|
|
||||||
# Extract requested subprotocols from headers for proper WebSocket negotiation
|
# Extract requested subprotocols from headers for proper WebSocket negotiation
|
||||||
# This is critical for protocols like xpra that require specific subprotocols
|
# This is critical for protocols like xpra that require specific subprotocols
|
||||||
@ -238,4 +238,5 @@ async def get_current_active_user_from_websocket(
|
|||||||
websocket_error = {"action": "log.error", "event": {"message": err_msg}}
|
websocket_error = {"action": "log.error", "event": {"message": err_msg}}
|
||||||
await websocket.send_json(websocket_error)
|
await websocket.send_json(websocket_error)
|
||||||
log.error(err_msg)
|
log.error(err_msg)
|
||||||
return await websocket.close(code=1008)
|
await websocket.close(code=1008)
|
||||||
|
return None
|
||||||
|
|||||||
@ -14,7 +14,10 @@
|
|||||||
# You should have received a copy of the GNU General Public License
|
# You should have received a copy of the GNU General Public License
|
||||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
import uuid
|
||||||
|
|
||||||
from sqlalchemy import Column, String, Boolean, DateTime, ForeignKey, func
|
from sqlalchemy import Column, String, Boolean, DateTime, ForeignKey, func
|
||||||
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
|
||||||
from .base import BaseTable, GUID
|
from .base import BaseTable, GUID
|
||||||
|
|
||||||
@ -22,8 +25,10 @@ from .base import BaseTable, GUID
|
|||||||
class ApiKey(BaseTable):
|
class ApiKey(BaseTable):
|
||||||
__tablename__ = "api_keys"
|
__tablename__ = "api_keys"
|
||||||
|
|
||||||
api_key_id = Column(GUID, primary_key=True)
|
api_key_id: Mapped[uuid.UUID] = mapped_column(GUID, primary_key=True)
|
||||||
user_id = Column(GUID, ForeignKey("users.user_id", ondelete="CASCADE"), nullable=False, index=True)
|
user_id: Mapped[uuid.UUID] = mapped_column(
|
||||||
|
GUID, ForeignKey("users.user_id", ondelete="CASCADE"), nullable=False, index=True
|
||||||
|
)
|
||||||
name = Column(String(128), nullable=False)
|
name = Column(String(128), nullable=False)
|
||||||
key_hash = Column(String(128), nullable=False)
|
key_hash = Column(String(128), nullable=False)
|
||||||
key_prefix = Column(String(8), nullable=False)
|
key_prefix = Column(String(8), nullable=False)
|
||||||
|
|||||||
@ -25,7 +25,7 @@ class Token(BaseModel):
|
|||||||
|
|
||||||
|
|
||||||
class TokenData(BaseModel):
|
class TokenData(BaseModel):
|
||||||
username: Optional[str] = None
|
username: str
|
||||||
token_version: int = 0
|
token_version: int = 0
|
||||||
token_use: str = "access"
|
token_use: str = "access"
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user