|
|
@@ -0,0 +1,103 @@
|
|
|
+# -*- coding: utf-8 -*-
|
|
|
+
|
|
|
+from fastapi import APIRouter, Depends, Request, Header, Form, Body
|
|
|
+from fastapi.responses import FileResponse, StreamingResponse
|
|
|
+from sqlalchemy.orm import Session
|
|
|
+from fastapi.responses import JSONResponse
|
|
|
+from database import get_db
|
|
|
+from utils import *
|
|
|
+from utils.vcode import *
|
|
|
+from utils.redis_util import *
|
|
|
+import base64
|
|
|
+from common.const import *
|
|
|
+from io import BytesIO
|
|
|
+from utils.StripTagsHTMLParser import *
|
|
|
+from common import security
|
|
|
+from datetime import timedelta
|
|
|
+from common.security import valid_access_token
|
|
|
+from common.auth_user import *
|
|
|
+from common import YzyApi
|
|
|
+from models import *
|
|
|
+from urllib.parse import quote
|
|
|
+import requests
|
|
|
+import jwt
|
|
|
+import traceback
|
|
|
+
|
|
|
+router = APIRouter()
|
|
|
+
|
|
|
+
|
|
|
+# 提供给数科使用的单点登录token
|
|
|
+@router.post('/token/create')
|
|
|
+def sso_token(request: Request,
|
|
|
+ user_id: int = Depends(valid_access_token),
|
|
|
+ db: Session = Depends(get_db)):
|
|
|
+
|
|
|
+ sso_token_expires = timedelta(seconds = 300)
|
|
|
+ sso_token = security.create_access_token(
|
|
|
+ data={"sub": user_id}, expires_delta = sso_token_expires
|
|
|
+ )
|
|
|
+
|
|
|
+ return {
|
|
|
+ "code": 200,
|
|
|
+ "msg": "操作成功",
|
|
|
+ "data": {
|
|
|
+ "sso_token": sso_token
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+# token校验
|
|
|
+@router.post('/token/valid')
|
|
|
+def sso_token(request: Request,
|
|
|
+ sso_token: str,
|
|
|
+ db: Session = Depends(get_db)):
|
|
|
+
|
|
|
+ try:
|
|
|
+ payload = jwt.decode(sso_token, settings.SECRET_KEY, algorithms=[settings.ALGORITHM])
|
|
|
+ print(payload,payload.get("sub"))
|
|
|
+ user_id: str = payload.get("sub")
|
|
|
+
|
|
|
+ row = db.query(SysUser).filter(SysUser.user_id == int(user_id)).first()
|
|
|
+ if row is None:
|
|
|
+ return {
|
|
|
+ "code": 500,
|
|
|
+ "msg": "token异常"
|
|
|
+ }
|
|
|
+
|
|
|
+ # 角色信息
|
|
|
+ roles = []
|
|
|
+ role_ids = db.query(SysUserRole).filter(SysUserRole.user_id == int(user_id)).all()
|
|
|
+ for role in role_ids:
|
|
|
+ role_info = db.query(SysRole).filter(SysRole.role_id == role.role_id).first()
|
|
|
+ roles.append(
|
|
|
+ {
|
|
|
+ "roleId": role_info.role_id,
|
|
|
+ "roleName": role_info.role_name,
|
|
|
+ "roleKey": role_info.role_key
|
|
|
+ }
|
|
|
+ )
|
|
|
+
|
|
|
+ role_keys = [
|
|
|
+ n['roleKey']
|
|
|
+ for n in roles
|
|
|
+ ]
|
|
|
+
|
|
|
+ data = {
|
|
|
+ "userId": row.user_id,
|
|
|
+ "userName": row.user_name,
|
|
|
+ "nickName": row.nick_name,
|
|
|
+ "roles": role_keys
|
|
|
+ }
|
|
|
+
|
|
|
+ return {
|
|
|
+ "code": 200,
|
|
|
+ "msg": "操作成功",
|
|
|
+ "data": data
|
|
|
+ }
|
|
|
+
|
|
|
+ except Exception:
|
|
|
+ traceback.print_exc()
|
|
|
+ return {
|
|
|
+ "code": 500,
|
|
|
+ "msg": "token异常"
|
|
|
+ }
|
|
|
+
|
