rainycode 1.6.1__tar.gz → 1.6.3__tar.gz

{rainycode-1.6.1 → rainycode-1.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rainycode
-Version: 1.6.1
+Version: 1.6.3
 Summary: FastAPI base modules
 Requires-Python: >=3.10
 Requires-Dist: aerich==0.9.2

{rainycode-1.6.1 → rainycode-1.6.3}/common_depends/auth_depend.py

@@ -11,7 +11,7 @@ from common_base.exception import CustomException
 from common_models.user_model import User
 from common_utils.jwt_util import JwtUtil
 from core.databases.aioredis import AioRedis
-from core.start import base_config
+from core.base_config import base_config
 
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl=f"/api/{base_config.app_name}/auth/login", auto_error=False)
 
@@ -21,7 +21,7 @@ async def get_current_user(request: Request, token: str, security_scopes: Securi
     """
     payload = JwtUtil.decode_token(token)
     if not payload:
-        raise CustomException(msg=f"无效的令牌 {token}", status_code=401)
+        raise CustomException(msg="无效的令牌", status_code=401)
 
     # 检查是否在黑名单中
     if await AioRedis.get(f"{REDIS_PREFIX_BLOCKLIST}{token}"):
@@ -29,7 +29,7 @@ async def get_current_user(request: Request, token: str, security_scopes: Securi
 
     user_id = payload.get("sub")
     if not user_id:
-        raise CustomException(msg=f"无效的令牌 {token}", status_code=401)
+        raise CustomException(msg="无效的令牌", status_code=401)
 
     # 1. 尝试从缓存获取
     user = None

{rainycode-1.6.1 → rainycode-1.6.3}/common_servers/api/auth_api.py

@@ -55,7 +55,7 @@ async def logout(
 
 @router.get("/me", response_model=UserInfo, summary="获取当前用户信息")
 async def get_me(user: User = Security(login_require)):
-    return SuccessReturn(data=user)
+    return SuccessReturn(data=await User.model2dict(user), msg="查询成功")
 
 
 @router.post("/wechat/mini", response_model=TokenResponse, summary="微信小程序登录")

{rainycode-1.6.1 → rainycode-1.6.3}/common_servers/api/wechat_api.py

@@ -89,8 +89,10 @@ async def wechat_oauth_callback(
                     "error": error,
                     "error_description": error_description or "用户拒绝授权",
                 }
-                # redirect_uri 现在可能已带目标路由（...#/pages/xxx），拼接符按需选，
-                # 使错误信息落在该页自己的 query 上（路由之后）
+                # 拼接符按 redirect_uri 现状选：前端现在把目标页路由挂在 query 上
+                # （?dw_route=%2Fpages%2Fxxx，原因见 AuthService._build_redirect_response），
+                # 此时用 & 续接；存量授权链接仍是 #/pages/xxx 形态，用 ? 续接。
+                # 两种形态下错误信息都会落在该页自己的参数区，页面能取到并提示。
                 sep = "&" if "?" in redirect_uri else "?"
                 error_url = f"{redirect_uri}{sep}{urlencode(params)}"
                 return RedirectResponse(url=error_url, status_code=302)

{rainycode-1.6.1 → rainycode-1.6.3}/common_servers/service/auth_svc.py

@@ -521,7 +521,7 @@ class AuthService:
     async def _build_redirect_response(
         cls, redirect_uri: str, token_data: dict, state: str | None = None
     ) -> RedirectResponse:
-        """构建带 token 的重定向响应（使用 URL Fragment）"""
+        """构建带 token 的重定向响应（凭证拼在 redirect_uri 的 query 上）"""
         params = {
             "access_token": token_data["access_token"],
             "refresh_token": token_data["refresh_token"],
@@ -531,9 +531,18 @@ class AuthService:
         if state:
             params["state"] = state
 
-        # 凭证一律挂在 query 上（token 不落 fragment）：redirect_uri 由前端带上目标页
-        # 路由（https://host/#/pages/xxx），拼出来即 ...#/pages/xxx?access_token=..
-        # 它自身已带参数时（...#/pages/xxx?id=3）用 & 续接。
+        # 凭证一律拼在 query 上，绝不落到 `#` 之后。
+        #
+        # 为什么：PC 微信跟随 302 时会丢掉 `#` 之后整段（微信官方社区原话：fragment
+        #「仅用作浏览器端的处理，不会参与到服务端请求中，故回跳不会带上这部分信息」）。
+        # 凭证若落在 fragment，就等于在电脑端凭空消失，用户表现为「点了登录回到首页
+        # 且没登录」，后端日志里连 /user/profile 都不会出现。
+        # 因此前端 guard.js 连目标页路由也一并挪到 query 上（?dw_route=%2Fpages%2Fxxx），
+        # 由 index.html 在 app 启动前还原成 hash 路由。这里只按 redirect_uri 是否已带
+        # 参数来选拼接符：
+        #   ?dw_route=..（当前形态，路由挂在 query）→ 用 & 续接
+        #   #/pages/xxx（存量授权链接，路由挂在 hash）→ 用 ? 续接
+        # 注：变量名 fragment 是历史沿用，实际拼在 query 上。
         fragment = urlencode(params)
         sep = "&" if "?" in redirect_uri else "?"
         redirect_url = f"{redirect_uri}{sep}{fragment}"

{rainycode-1.6.1 → rainycode-1.6.3}/common_servers/service/file_svc.py

@@ -16,8 +16,8 @@ class FileService:
             cls,
             user: User,
             file,
-            *,
             subdir: str,
+            upload_root: str = base_config.upload_root,
             allowed_exts: set[str] = base_config.upload_allowed_image_exts,
             max_bytes: int = base_config.upload_max_bytes,
     ) -> dict:
@@ -37,7 +37,7 @@ class FileService:
 
         filename = f"{uuid.uuid4().hex}.{ext}"
         relative = f"{subdir}/{filename}"  # 磁盘相对路径 & File.url（不含前缀）
-        path = Path(base_config.upload_root) / relative
+        path = Path(upload_root) / relative
         path.parent.mkdir(parents=True, exist_ok=True)
         path.write_bytes(content)
 

rainycode-1.6.3/common_utils/pagination.py

@@ -0,0 +1,34 @@
+class Pagination:
+    """
+    对 Tortoise ORM 查询集执行分页，并返回总记录数和当前页数据。
+
+    示例：
+        qs = Order.filter(couple_id=couple.id).order_by('-id')
+        page = await Pagination.page(qs, page_no=page_no, page_size=page_size)
+
+    返回结构为 {'items': 当前页模型对象列表, 'count': 符合查询条件的总数}。
+    本类只负责查询计数和 limit/offset，不负责序列化或组装业务字段。
+    调用方应在需要时转换模型对象，并确保查询集具有稳定的排序规则。
+    """
+    DEFAULT_PAGE_SIZE = 20
+    MAX_PAGE_SIZE = 999
+
+    @classmethod
+    async def page(cls, qs, page_no: int | None = None, page_size: int | None = None) -> dict:
+        """获取指定页的数据。
+
+        Args:
+            qs: 已完成筛选及排序的 Tortoise ORM 查询集。
+            page_no: 页码，从 1 开始；未提供或小于 1 时按第 1 页处理。
+            page_size: 每页条数；未提供时使用默认值，并限制在 1 到 MAX_PAGE_SIZE 之间。
+
+        Returns:
+            包含当前页模型对象列表 ``items`` 和匹配记录总数 ``count`` 的字典。
+        """
+        page_no = page_no or 1
+        page_no = page_no if page_no > 0 else 1
+        page_size = page_size or cls.DEFAULT_PAGE_SIZE
+        page_size = min(max(page_size, 1), cls.MAX_PAGE_SIZE)
+        count = await qs.count()
+        items = await qs.limit(page_size).offset((page_no - 1) * page_size)
+        return {'items': items, 'count': count}

{rainycode-1.6.1 → rainycode-1.6.3}/core/base_config.py

@@ -1,5 +1,10 @@
+import os
+
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+# 包内 static 目录（core/ 的上一级），默认的 /static 挂载点
+_STATIC_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../static"))
+
 
 class BaseConfig(BaseSettings):
     app_debug: bool = False
@@ -13,10 +18,15 @@ class BaseConfig(BaseSettings):
     # 基础 URL
     base_url: str = ""
 
+    # 静态文件：/static 的挂载根目录，默认包内 static；
+    # 想让静态资源落到项目目录，子类里覆盖这一项即可。
+    static_root: str = _STATIC_DIR
+
     # 文件上传：本地存储根目录 + 访问 URL 前缀。
-    # upload_url_prefix 可为相对路径（/uploads）或完整 URL（https://cdn.xxx.com/uploads），
-    upload_root: str = 'uploads'
-    upload_url_prefix: str = '/uploads'
+    # upload_root 必须在 static_root 之下，否则会出现「写得进、读不到」。
+    # upload_url_prefix 可为相对路径（/static/uploads）或完整 URL（https://cdn.xxx.com/uploads），
+    upload_root: str = os.path.join(_STATIC_DIR, 'uploads')
+    upload_url_prefix: str = '/static/uploads'
 
     # 跨域
     cors_origins: list[str] = ['*']

{rainycode-1.6.1 → rainycode-1.6.3}/core/start.py

@@ -13,7 +13,7 @@ from common_base.exception import (
     ValueExceptionHandler,
 )
 from common_base.logging import logger
-from core.base_config import BaseConfig, base_config
+from core.base_config import BaseConfig
 from core.databases.aiodb import AioDb
 from core.middleware.http_middleware import RequestHttpMiddleware
 from fastapi import FastAPI, HTTPException
@@ -39,9 +39,9 @@ def start_server(config: BaseConfig):
     # 注册异常处理
     register_exception(app)
     # 注册中间件
-    register_middlewares(app)
+    register_middlewares(app, config)
     # 注册静态文件
-    register_files(app)
+    register_files(app, config)
 
     return app
 
@@ -79,25 +79,25 @@ def register_exception(app: FastAPI):
     app.add_exception_handler(Exception, AllExceptionHandler) # pyright: ignore[reportArgumentType]
 
 
-def register_middlewares(app: FastAPI) -> None:
+def register_middlewares(app: FastAPI, config: BaseConfig) -> None:
     """
     注册中间件
     """
     app.add_middleware(RequestHttpMiddleware)
     app.add_middleware(
         CORSMiddleware,
-        allow_origins=base_config.cors_origins,
-        allow_credentials=base_config.cors_allow_credentials,
-        allow_methods=base_config.cors_allow_methods,
-        allow_headers=base_config.cors_allow_headers
+        allow_origins=config.cors_origins,
+        allow_credentials=config.cors_allow_credentials,
+        allow_methods=config.cors_allow_methods,
+        allow_headers=config.cors_allow_headers
     )
 
 
-def register_files(app: FastAPI) -> None:
+def register_files(app: FastAPI, config: BaseConfig) -> None:
     """
     注册文件相关配置
     """
-    static_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../static"))
+    static_dir = config.static_root
     if not os.path.exists(static_dir):
         os.mkdir(static_dir)
     app.mount('/static', StaticFiles(directory=static_dir))

{rainycode-1.6.1 → rainycode-1.6.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rainycode"
-version = "1.6.1"
+version = "1.6.3"
 description = "FastAPI base modules"
 requires-python = ">=3.10"
 

{rainycode-1.6.1 → rainycode-1.6.3}/rainycode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rainycode
-Version: 1.6.1
+Version: 1.6.3
 Summary: FastAPI base modules
 Requires-Python: >=3.10
 Requires-Dist: aerich==0.9.2

{rainycode-1.6.1 → rainycode-1.6.3}/rainycode.egg-info/SOURCES.txt

@@ -22,6 +22,7 @@ common_utils/captcha_util.py
 common_utils/datetime_util.py
 common_utils/ip_util.py
 common_utils/jwt_util.py
+common_utils/pagination.py
 common_utils/snowflake_util.py
 common_utils/wechat_util.py
 core/base_config.py