Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a0196b8d64 | |||
| 53c3b7f60f |
@@ -0,0 +1,31 @@
|
|||||||
|
"""guide video play count is independent for coupon and comparison
|
||||||
|
|
||||||
|
Revision ID: guide_video_scene_unique
|
||||||
|
Revises: 6d2309208549
|
||||||
|
"""
|
||||||
|
from alembic import op
|
||||||
|
|
||||||
|
revision = "guide_video_scene_unique"
|
||||||
|
down_revision = "6d2309208549"
|
||||||
|
branch_labels = None
|
||||||
|
depends_on = None
|
||||||
|
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
op.drop_index("uq_guide_video_play_user_seq", table_name="guide_video_play")
|
||||||
|
op.create_index(
|
||||||
|
"uq_guide_video_play_user_scene_seq",
|
||||||
|
"guide_video_play",
|
||||||
|
["user_id", "scene", "seq"],
|
||||||
|
unique=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
op.drop_index("uq_guide_video_play_user_scene_seq", table_name="guide_video_play")
|
||||||
|
op.create_index(
|
||||||
|
"uq_guide_video_play_user_seq",
|
||||||
|
"guide_video_play",
|
||||||
|
["user_id", "seq"],
|
||||||
|
unique=True,
|
||||||
|
)
|
||||||
@@ -9,7 +9,7 @@ client_max_body_size,见 shaguabijia-admin-web/deploy/nginx/admin.shaguabijia.co
|
|||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from typing import Annotated
|
from typing import Annotated, Literal
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile
|
from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile
|
||||||
|
|
||||||
@@ -27,14 +27,21 @@ router = APIRouter(
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def _out(db: AdminDb) -> GuideVideoConfigOut:
|
GuideScene = Literal["coupon", "comparison"]
|
||||||
|
|
||||||
|
|
||||||
|
def _out(db: AdminDb, scene: GuideScene) -> GuideVideoConfigOut:
|
||||||
"""配置 + 播放统计合成响应(四个写接口都以最新状态返回,前端一次同步到位)。"""
|
"""配置 + 播放统计合成响应(四个写接口都以最新状态返回,前端一次同步到位)。"""
|
||||||
return GuideVideoConfigOut(**guide_video.get_config(db), **guide_video.play_stats(db))
|
return GuideVideoConfigOut(
|
||||||
|
scene=scene,
|
||||||
|
**guide_video.get_config(db, scene),
|
||||||
|
**guide_video.play_stats(db, scene),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@router.get("", response_model=GuideVideoConfigOut, summary="新手引导视频配置(领券浮层)")
|
@router.get("", response_model=GuideVideoConfigOut, summary="新手引导视频配置(领券浮层)")
|
||||||
def get_config(db: AdminDb) -> GuideVideoConfigOut:
|
def get_config(db: AdminDb, scene: GuideScene = "coupon") -> GuideVideoConfigOut:
|
||||||
return _out(db)
|
return _out(db, scene)
|
||||||
|
|
||||||
|
|
||||||
@router.patch("", response_model=GuideVideoConfigOut, summary="改开关/次数/金币(带审计)")
|
@router.patch("", response_model=GuideVideoConfigOut, summary="改开关/次数/金币(带审计)")
|
||||||
@@ -43,21 +50,23 @@ def update_config(
|
|||||||
request: Request,
|
request: Request,
|
||||||
admin: Annotated[AdminUser, Depends(require_role("operator"))],
|
admin: Annotated[AdminUser, Depends(require_role("operator"))],
|
||||||
db: AdminDb,
|
db: AdminDb,
|
||||||
|
scene: GuideScene = "coupon",
|
||||||
) -> GuideVideoConfigOut:
|
) -> GuideVideoConfigOut:
|
||||||
before, after = guide_video.update_config(
|
before, after = guide_video.update_config(
|
||||||
db,
|
db,
|
||||||
enabled=body.enabled,
|
enabled=body.enabled,
|
||||||
max_plays=body.max_plays,
|
max_plays=body.max_plays,
|
||||||
reward_coin=body.reward_coin,
|
reward_coin=body.reward_coin,
|
||||||
|
scene=scene,
|
||||||
admin_id=admin.id,
|
admin_id=admin.id,
|
||||||
commit=False,
|
commit=False,
|
||||||
)
|
)
|
||||||
write_audit(
|
write_audit(
|
||||||
db, admin, action="guide_video.update", target_type="guide_video", target_id=None,
|
db, admin, action="guide_video.update", target_type="guide_video", target_id=None,
|
||||||
detail={"before": before, "after": after}, ip=get_client_ip(request), commit=False,
|
detail={"scene": scene, "before": before, "after": after}, ip=get_client_ip(request), commit=False,
|
||||||
)
|
)
|
||||||
db.commit()
|
db.commit()
|
||||||
return _out(db)
|
return _out(db, scene)
|
||||||
|
|
||||||
|
|
||||||
@router.post("/video", response_model=GuideVideoConfigOut, summary="上传新手引导视频(MP4,带审计)")
|
@router.post("/video", response_model=GuideVideoConfigOut, summary="上传新手引导视频(MP4,带审计)")
|
||||||
@@ -65,23 +74,26 @@ async def upload_video(
|
|||||||
request: Request,
|
request: Request,
|
||||||
admin: Annotated[AdminUser, Depends(require_role("operator"))],
|
admin: Annotated[AdminUser, Depends(require_role("operator"))],
|
||||||
db: AdminDb,
|
db: AdminDb,
|
||||||
file: UploadFile = File(...),
|
file: Annotated[UploadFile, File()],
|
||||||
|
scene: GuideScene = "coupon",
|
||||||
) -> GuideVideoConfigOut:
|
) -> GuideVideoConfigOut:
|
||||||
data = await file.read()
|
data = await file.read()
|
||||||
try:
|
try:
|
||||||
url = media.save_guide_video(data)
|
url = media.save_guide_video(data)
|
||||||
except media.MediaError as e:
|
except media.MediaError as e:
|
||||||
raise HTTPException(status_code=400, detail=str(e)) from e
|
raise HTTPException(status_code=400, detail=str(e)) from e
|
||||||
before, after = guide_video.set_video(db, url, admin_id=admin.id, commit=False)
|
before, after = guide_video.set_video(
|
||||||
|
db, url, scene=scene, admin_id=admin.id, commit=False
|
||||||
|
)
|
||||||
write_audit(
|
write_audit(
|
||||||
db, admin, action="guide_video.set_video", target_type="guide_video", target_id=None,
|
db, admin, action="guide_video.set_video", target_type="guide_video", target_id=None,
|
||||||
detail={"before": before.get("video_url"), "after": url, "bytes": len(data)},
|
detail={"scene": scene, "before": before.get("video_url"), "after": url, "bytes": len(data)},
|
||||||
ip=get_client_ip(request), commit=False,
|
ip=get_client_ip(request), commit=False,
|
||||||
)
|
)
|
||||||
db.commit()
|
db.commit()
|
||||||
# 提交成功后再删旧片,避免新片没落库就把旧片丢了
|
# 提交成功后再删旧片,避免新片没落库就把旧片丢了
|
||||||
media.delete_guide_video(before.get("video_url"))
|
media.delete_guide_video(before.get("video_url"))
|
||||||
return _out(db)
|
return _out(db, scene)
|
||||||
|
|
||||||
|
|
||||||
@router.delete("/video", response_model=GuideVideoConfigOut, summary="移除新手引导视频(带审计)")
|
@router.delete("/video", response_model=GuideVideoConfigOut, summary="移除新手引导视频(带审计)")
|
||||||
@@ -89,13 +101,16 @@ def delete_video(
|
|||||||
request: Request,
|
request: Request,
|
||||||
admin: Annotated[AdminUser, Depends(require_role("operator"))],
|
admin: Annotated[AdminUser, Depends(require_role("operator"))],
|
||||||
db: AdminDb,
|
db: AdminDb,
|
||||||
|
scene: GuideScene = "coupon",
|
||||||
) -> GuideVideoConfigOut:
|
) -> GuideVideoConfigOut:
|
||||||
"""移除后 /guide-video/start 一律返回 should_play=false,领券浮层回到「只放广告」。"""
|
"""移除后 /guide-video/start 一律返回 should_play=false,领券浮层回到「只放广告」。"""
|
||||||
before, after = guide_video.set_video(db, None, admin_id=admin.id, commit=False)
|
before, after = guide_video.set_video(
|
||||||
|
db, None, scene=scene, admin_id=admin.id, commit=False
|
||||||
|
)
|
||||||
write_audit(
|
write_audit(
|
||||||
db, admin, action="guide_video.delete_video", target_type="guide_video", target_id=None,
|
db, admin, action="guide_video.delete_video", target_type="guide_video", target_id=None,
|
||||||
detail={"before": before.get("video_url")}, ip=get_client_ip(request), commit=False,
|
detail={"scene": scene, "before": before.get("video_url")}, ip=get_client_ip(request), commit=False,
|
||||||
)
|
)
|
||||||
db.commit()
|
db.commit()
|
||||||
media.delete_guide_video(before.get("video_url"))
|
media.delete_guide_video(before.get("video_url"))
|
||||||
return _out(db)
|
return _out(db, scene)
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ from app.repositories.guide_video import MAX_PLAYS_LIMIT, REWARD_COIN_LIMIT
|
|||||||
|
|
||||||
|
|
||||||
class GuideVideoConfigOut(BaseModel):
|
class GuideVideoConfigOut(BaseModel):
|
||||||
|
scene: str
|
||||||
enabled: bool
|
enabled: bool
|
||||||
video_url: str | None = None # 相对地址 /media/guide_video/xxx.mp4;未配片 = None
|
video_url: str | None = None # 相对地址 /media/guide_video/xxx.mp4;未配片 = None
|
||||||
max_plays: int
|
max_plays: int
|
||||||
|
|||||||
+242
-35
@@ -22,7 +22,7 @@ import uuid
|
|||||||
from collections.abc import Callable
|
from collections.abc import Callable
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import Any
|
from typing import Any
|
||||||
from urllib.parse import quote
|
from urllib.parse import quote, urlsplit
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
@@ -69,6 +69,18 @@ class _CachedToken:
|
|||||||
|
|
||||||
_token_cache: dict[str, _CachedToken] = {}
|
_token_cache: dict[str, _CachedToken] = {}
|
||||||
|
|
||||||
|
_LOG_SUMMARY_MAX_CHARS = 1500
|
||||||
|
_LOG_STRING_MAX_CHARS = 200
|
||||||
|
_SENSITIVE_LOG_KEYS = {
|
||||||
|
"accesstoken",
|
||||||
|
"appkey",
|
||||||
|
"authtoken",
|
||||||
|
"authorization",
|
||||||
|
"clientsecret",
|
||||||
|
"mastersecret",
|
||||||
|
"sign",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
def normalize_vendor(push_vendor: str | None) -> str | None:
|
def normalize_vendor(push_vendor: str | None) -> str | None:
|
||||||
if not push_vendor:
|
if not push_vendor:
|
||||||
@@ -115,8 +127,16 @@ def send_notification(
|
|||||||
|
|
||||||
if mock:
|
if mock:
|
||||||
logger.info(
|
logger.info(
|
||||||
"[mock push] vendor=%s token=%s... title=%s body=%s extras=%s",
|
"vendor push mock vendor=%s request=%s",
|
||||||
vendor, token[:12], title, body, extras,
|
vendor,
|
||||||
|
_log_summary(
|
||||||
|
{
|
||||||
|
"push_token": token,
|
||||||
|
"title": title,
|
||||||
|
"body": body,
|
||||||
|
"extras": extras,
|
||||||
|
}
|
||||||
|
),
|
||||||
)
|
)
|
||||||
return {
|
return {
|
||||||
"mock": True,
|
"mock": True,
|
||||||
@@ -133,7 +153,28 @@ def send_notification(
|
|||||||
"xiaomi": _send_xiaomi,
|
"xiaomi": _send_xiaomi,
|
||||||
"oppo": _send_oppo,
|
"oppo": _send_oppo,
|
||||||
}
|
}
|
||||||
return dispatch[vendor](token, title, body, extras)
|
started = time.perf_counter()
|
||||||
|
try:
|
||||||
|
result = dispatch[vendor](token, title, body, extras)
|
||||||
|
except Exception:
|
||||||
|
logger.exception(
|
||||||
|
"vendor push dispatch failed vendor=%s token=%s notification_type=%s "
|
||||||
|
"elapsed_ms=%.1f",
|
||||||
|
vendor,
|
||||||
|
token,
|
||||||
|
extras.get("type", ""),
|
||||||
|
_elapsed_ms(started),
|
||||||
|
)
|
||||||
|
raise
|
||||||
|
logger.info(
|
||||||
|
"vendor push dispatch succeeded vendor=%s token=%s notification_type=%s "
|
||||||
|
"elapsed_ms=%.1f",
|
||||||
|
vendor,
|
||||||
|
token,
|
||||||
|
extras.get("type", ""),
|
||||||
|
_elapsed_ms(started),
|
||||||
|
)
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
def send_accessibility_disabled(
|
def send_accessibility_disabled(
|
||||||
@@ -159,13 +200,120 @@ def _require(value: str, name: str) -> str:
|
|||||||
return value
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
def _elapsed_ms(started: float) -> float:
|
||||||
|
return (time.perf_counter() - started) * 1000
|
||||||
|
|
||||||
|
|
||||||
|
def _redacted_value(value: Any) -> str:
|
||||||
|
raw = json.dumps(value, ensure_ascii=False, sort_keys=True, default=str)
|
||||||
|
digest = hashlib.sha256(raw.encode()).hexdigest()[:10]
|
||||||
|
return f"<redacted len={len(raw)} sha256={digest}>"
|
||||||
|
|
||||||
|
|
||||||
|
def _sanitize_for_log(value: Any, *, key: str = "", depth: int = 0) -> Any:
|
||||||
|
"""生成有排障价值、但不暴露服务端鉴权凭据的紧凑日志摘要。"""
|
||||||
|
normalized_key = "".join(char for char in key.lower() if char.isalnum())
|
||||||
|
if normalized_key in _SENSITIVE_LOG_KEYS or normalized_key.endswith("secret"):
|
||||||
|
return _redacted_value(value)
|
||||||
|
if depth >= 10:
|
||||||
|
return f"<{type(value).__name__}>"
|
||||||
|
if isinstance(value, dict):
|
||||||
|
items = list(value.items())
|
||||||
|
sanitized = {
|
||||||
|
str(item_key): _sanitize_for_log(item_value, key=str(item_key), depth=depth + 1)
|
||||||
|
for item_key, item_value in items[:30]
|
||||||
|
}
|
||||||
|
if len(items) > 30:
|
||||||
|
sanitized["<omitted_keys>"] = len(items) - 30
|
||||||
|
return sanitized
|
||||||
|
if isinstance(value, (list, tuple)):
|
||||||
|
items = list(value)
|
||||||
|
sanitized = [_sanitize_for_log(item, key=key, depth=depth + 1) for item in items[:20]]
|
||||||
|
if len(items) > 20:
|
||||||
|
sanitized.append(f"<omitted_items={len(items) - 20}>")
|
||||||
|
return sanitized
|
||||||
|
if isinstance(value, str):
|
||||||
|
if value.lstrip().startswith(("{", "[")):
|
||||||
|
try:
|
||||||
|
decoded = json.loads(value)
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
return _sanitize_for_log(decoded, key=key, depth=depth + 1)
|
||||||
|
if len(value) > _LOG_STRING_MAX_CHARS:
|
||||||
|
return f"{value[:_LOG_STRING_MAX_CHARS]}…<len={len(value)}>"
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
def _log_summary(value: Any) -> str:
|
||||||
|
rendered = json.dumps(
|
||||||
|
_sanitize_for_log(value),
|
||||||
|
ensure_ascii=False,
|
||||||
|
sort_keys=True,
|
||||||
|
separators=(",", ":"),
|
||||||
|
default=str,
|
||||||
|
)
|
||||||
|
if len(rendered) > _LOG_SUMMARY_MAX_CHARS:
|
||||||
|
return f"{rendered[:_LOG_SUMMARY_MAX_CHARS]}…<len={len(rendered)}>"
|
||||||
|
return rendered
|
||||||
|
|
||||||
|
|
||||||
|
def _raw_log_summary(value: Any) -> str:
|
||||||
|
"""厂商响应摘要:保留原始字段和值,仅限制单条日志长度。"""
|
||||||
|
rendered = json.dumps(
|
||||||
|
value,
|
||||||
|
ensure_ascii=False,
|
||||||
|
sort_keys=True,
|
||||||
|
separators=(",", ":"),
|
||||||
|
default=str,
|
||||||
|
)
|
||||||
|
if len(rendered) > _LOG_SUMMARY_MAX_CHARS:
|
||||||
|
return f"{rendered[:_LOG_SUMMARY_MAX_CHARS]}…<len={len(rendered)}>"
|
||||||
|
return rendered
|
||||||
|
|
||||||
|
|
||||||
|
def _endpoint_for_log(url: str) -> str:
|
||||||
|
parsed = urlsplit(url)
|
||||||
|
return f"{parsed.scheme}://{parsed.netloc}{parsed.path}"
|
||||||
|
|
||||||
|
|
||||||
|
def _request_summary(kwargs: dict[str, Any]) -> str:
|
||||||
|
summary: dict[str, Any] = {}
|
||||||
|
if "json" in kwargs:
|
||||||
|
summary["body_type"] = "json"
|
||||||
|
summary["request_payload"] = kwargs["json"]
|
||||||
|
elif "data" in kwargs:
|
||||||
|
summary["body_type"] = "form"
|
||||||
|
summary["request_payload"] = kwargs["data"]
|
||||||
|
if kwargs.get("params"):
|
||||||
|
summary["params"] = kwargs["params"]
|
||||||
|
if kwargs.get("headers"):
|
||||||
|
summary["headers"] = kwargs["headers"]
|
||||||
|
return _log_summary(summary)
|
||||||
|
|
||||||
|
|
||||||
|
def _response_summary(resp: Any, parsed: Any | None = None) -> str:
|
||||||
|
if parsed is None:
|
||||||
|
try:
|
||||||
|
parsed = resp.json()
|
||||||
|
except ValueError:
|
||||||
|
pass
|
||||||
|
if parsed is not None:
|
||||||
|
return _raw_log_summary(parsed)
|
||||||
|
return _raw_log_summary(getattr(resp, "text", ""))
|
||||||
|
|
||||||
|
|
||||||
def _request_json(
|
def _request_json(
|
||||||
method: str,
|
method: str,
|
||||||
url: str,
|
url: str,
|
||||||
*,
|
*,
|
||||||
|
vendor: str,
|
||||||
|
operation: str,
|
||||||
expected_status: tuple[int, ...] = (200,),
|
expected_status: tuple[int, ...] = (200,),
|
||||||
**kwargs: Any,
|
**kwargs: Any,
|
||||||
) -> dict[str, Any]:
|
) -> dict[str, Any]:
|
||||||
|
started = time.perf_counter()
|
||||||
|
request_summary = _request_summary(kwargs)
|
||||||
try:
|
try:
|
||||||
resp = httpx.request(
|
resp = httpx.request(
|
||||||
method,
|
method,
|
||||||
@@ -174,41 +322,82 @@ def _request_json(
|
|||||||
**kwargs,
|
**kwargs,
|
||||||
)
|
)
|
||||||
except httpx.HTTPError as e:
|
except httpx.HTTPError as e:
|
||||||
|
logger.error(
|
||||||
|
"vendor push http completed vendor=%s operation=%s method=%s endpoint=%s "
|
||||||
|
"outcome=network_error request=%s response=%s elapsed_ms=%.1f error=%s",
|
||||||
|
vendor,
|
||||||
|
operation,
|
||||||
|
method,
|
||||||
|
_endpoint_for_log(url),
|
||||||
|
request_summary,
|
||||||
|
"<no_response>",
|
||||||
|
_elapsed_ms(started),
|
||||||
|
type(e).__name__,
|
||||||
|
)
|
||||||
raise VendorPushError(f"push http error: {e}") from e
|
raise VendorPushError(f"push http error: {e}") from e
|
||||||
|
|
||||||
if resp.status_code not in expected_status:
|
if resp.status_code not in expected_status:
|
||||||
logger.error("vendor push http failed url=%s http=%s body=%s", url, resp.status_code, resp.text[:500])
|
logger.error(
|
||||||
|
"vendor push http completed vendor=%s operation=%s method=%s endpoint=%s "
|
||||||
|
"outcome=http_error http_status=%s request=%s response=%s elapsed_ms=%.1f",
|
||||||
|
vendor,
|
||||||
|
operation,
|
||||||
|
method,
|
||||||
|
_endpoint_for_log(url),
|
||||||
|
resp.status_code,
|
||||||
|
request_summary,
|
||||||
|
_response_summary(resp),
|
||||||
|
_elapsed_ms(started),
|
||||||
|
)
|
||||||
raise VendorPushError(f"push http {resp.status_code}")
|
raise VendorPushError(f"push http {resp.status_code}")
|
||||||
try:
|
try:
|
||||||
return resp.json()
|
data = resp.json()
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
|
logger.error(
|
||||||
|
"vendor push http completed vendor=%s operation=%s method=%s endpoint=%s "
|
||||||
|
"outcome=invalid_json http_status=%s request=%s response=%s elapsed_ms=%.1f",
|
||||||
|
vendor,
|
||||||
|
operation,
|
||||||
|
method,
|
||||||
|
_endpoint_for_log(url),
|
||||||
|
resp.status_code,
|
||||||
|
request_summary,
|
||||||
|
_response_summary(resp),
|
||||||
|
_elapsed_ms(started),
|
||||||
|
)
|
||||||
raise VendorPushError(f"push invalid json: {resp.text[:200]}") from e
|
raise VendorPushError(f"push invalid json: {resp.text[:200]}") from e
|
||||||
|
logger.info(
|
||||||
|
"vendor push http completed vendor=%s operation=%s method=%s endpoint=%s "
|
||||||
|
"outcome=success http_status=%s request=%s response=%s elapsed_ms=%.1f",
|
||||||
|
vendor,
|
||||||
|
operation,
|
||||||
|
method,
|
||||||
|
_endpoint_for_log(url),
|
||||||
|
resp.status_code,
|
||||||
|
request_summary,
|
||||||
|
_response_summary(resp, data),
|
||||||
|
_elapsed_ms(started),
|
||||||
|
)
|
||||||
|
return data
|
||||||
|
|
||||||
|
|
||||||
def _request_form(
|
def _request_form(
|
||||||
method: str,
|
method: str,
|
||||||
url: str,
|
url: str,
|
||||||
*,
|
*,
|
||||||
|
vendor: str,
|
||||||
|
operation: str,
|
||||||
expected_status: tuple[int, ...] = (200,),
|
expected_status: tuple[int, ...] = (200,),
|
||||||
**kwargs: Any,
|
**kwargs: Any,
|
||||||
) -> dict[str, Any]:
|
) -> dict[str, Any]:
|
||||||
try:
|
return _request_json(
|
||||||
resp = httpx.request(
|
method,
|
||||||
method,
|
url,
|
||||||
url,
|
vendor=vendor,
|
||||||
timeout=settings.PUSH_REQUEST_TIMEOUT_SEC,
|
operation=operation,
|
||||||
**kwargs,
|
expected_status=expected_status,
|
||||||
)
|
**kwargs,
|
||||||
except httpx.HTTPError as e:
|
)
|
||||||
raise VendorPushError(f"push http error: {e}") from e
|
|
||||||
|
|
||||||
if resp.status_code not in expected_status:
|
|
||||||
logger.error("vendor push http failed url=%s http=%s body=%s", url, resp.status_code, resp.text[:500])
|
|
||||||
raise VendorPushError(f"push http {resp.status_code}")
|
|
||||||
try:
|
|
||||||
return resp.json()
|
|
||||||
except ValueError as e:
|
|
||||||
raise VendorPushError(f"push invalid json: {resp.text[:200]}") from e
|
|
||||||
|
|
||||||
|
|
||||||
def _cache_get(key: str) -> str | None:
|
def _cache_get(key: str) -> str | None:
|
||||||
@@ -234,6 +423,8 @@ def _honor_access_token() -> str:
|
|||||||
data = _request_form(
|
data = _request_form(
|
||||||
"POST",
|
"POST",
|
||||||
settings.HONOR_PUSH_TOKEN_ENDPOINT,
|
settings.HONOR_PUSH_TOKEN_ENDPOINT,
|
||||||
|
vendor="honor",
|
||||||
|
operation="authenticate",
|
||||||
data={
|
data={
|
||||||
"grant_type": "client_credentials",
|
"grant_type": "client_credentials",
|
||||||
"client_id": client_id,
|
"client_id": client_id,
|
||||||
@@ -243,7 +434,7 @@ def _honor_access_token() -> str:
|
|||||||
)
|
)
|
||||||
token = data.get("access_token")
|
token = data.get("access_token")
|
||||||
if not token:
|
if not token:
|
||||||
raise VendorPushError(f"honor auth failed: {data}")
|
raise VendorPushError(f"honor auth failed: {_raw_log_summary(data)}")
|
||||||
return _cache_put(cache_key, str(token), data.get("expires_in"))
|
return _cache_put(cache_key, str(token), data.get("expires_in"))
|
||||||
|
|
||||||
|
|
||||||
@@ -270,6 +461,8 @@ def _send_honor(token: str, title: str, body: str, extras: dict[str, str]) -> di
|
|||||||
data = _request_json(
|
data = _request_json(
|
||||||
"POST",
|
"POST",
|
||||||
settings.HONOR_PUSH_SEND_ENDPOINT_TEMPLATE.format(app_id=app_id),
|
settings.HONOR_PUSH_SEND_ENDPOINT_TEMPLATE.format(app_id=app_id),
|
||||||
|
vendor="honor",
|
||||||
|
operation="send_notification",
|
||||||
json=payload,
|
json=payload,
|
||||||
headers={
|
headers={
|
||||||
"Content-Type": "application/json; charset=UTF-8",
|
"Content-Type": "application/json; charset=UTF-8",
|
||||||
@@ -279,7 +472,7 @@ def _send_honor(token: str, title: str, body: str, extras: dict[str, str]) -> di
|
|||||||
)
|
)
|
||||||
code = data.get("code")
|
code = data.get("code")
|
||||||
if code is not None and int(code) != 200:
|
if code is not None and int(code) != 200:
|
||||||
raise VendorPushError(f"honor push failed: {data}")
|
raise VendorPushError(f"honor push failed: {_raw_log_summary(data)}")
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
@@ -294,6 +487,8 @@ def _huawei_access_token() -> str:
|
|||||||
data = _request_form(
|
data = _request_form(
|
||||||
"POST",
|
"POST",
|
||||||
settings.HUAWEI_PUSH_TOKEN_ENDPOINT,
|
settings.HUAWEI_PUSH_TOKEN_ENDPOINT,
|
||||||
|
vendor="huawei",
|
||||||
|
operation="authenticate",
|
||||||
data={
|
data={
|
||||||
"grant_type": "client_credentials",
|
"grant_type": "client_credentials",
|
||||||
"client_id": app_id,
|
"client_id": app_id,
|
||||||
@@ -303,7 +498,7 @@ def _huawei_access_token() -> str:
|
|||||||
)
|
)
|
||||||
token = data.get("access_token")
|
token = data.get("access_token")
|
||||||
if not token:
|
if not token:
|
||||||
raise VendorPushError(f"huawei auth failed: {data}")
|
raise VendorPushError(f"huawei auth failed: {_raw_log_summary(data)}")
|
||||||
return _cache_put(cache_key, str(token), data.get("expires_in"))
|
return _cache_put(cache_key, str(token), data.get("expires_in"))
|
||||||
|
|
||||||
|
|
||||||
@@ -333,6 +528,8 @@ def _send_huawei(token: str, title: str, body: str, extras: dict[str, str]) -> d
|
|||||||
data = _request_json(
|
data = _request_json(
|
||||||
"POST",
|
"POST",
|
||||||
settings.HUAWEI_PUSH_SEND_ENDPOINT_TEMPLATE.format(app_id=app_id),
|
settings.HUAWEI_PUSH_SEND_ENDPOINT_TEMPLATE.format(app_id=app_id),
|
||||||
|
vendor="huawei",
|
||||||
|
operation="send_notification",
|
||||||
json=payload,
|
json=payload,
|
||||||
headers={
|
headers={
|
||||||
"Content-Type": "application/json; charset=UTF-8",
|
"Content-Type": "application/json; charset=UTF-8",
|
||||||
@@ -340,7 +537,7 @@ def _send_huawei(token: str, title: str, body: str, extras: dict[str, str]) -> d
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
if str(data.get("code", "")) != "80000000":
|
if str(data.get("code", "")) != "80000000":
|
||||||
raise VendorPushError(f"huawei push failed: {data}")
|
raise VendorPushError(f"huawei push failed: {_raw_log_summary(data)}")
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
@@ -357,6 +554,8 @@ def _vivo_auth_token() -> str:
|
|||||||
data = _request_json(
|
data = _request_json(
|
||||||
"POST",
|
"POST",
|
||||||
settings.VIVO_PUSH_AUTH_ENDPOINT,
|
settings.VIVO_PUSH_AUTH_ENDPOINT,
|
||||||
|
vendor="vivo",
|
||||||
|
operation="authenticate",
|
||||||
json={
|
json={
|
||||||
"appId": app_id,
|
"appId": app_id,
|
||||||
"appKey": app_key,
|
"appKey": app_key,
|
||||||
@@ -366,10 +565,10 @@ def _vivo_auth_token() -> str:
|
|||||||
headers={"Content-Type": "application/json"},
|
headers={"Content-Type": "application/json"},
|
||||||
)
|
)
|
||||||
if int(data.get("result", -1)) != 0:
|
if int(data.get("result", -1)) != 0:
|
||||||
raise VendorPushError(f"vivo auth failed: {data}")
|
raise VendorPushError(f"vivo auth failed: {_raw_log_summary(data)}")
|
||||||
token = data.get("authToken")
|
token = data.get("authToken")
|
||||||
if not token:
|
if not token:
|
||||||
raise VendorPushError(f"vivo auth missing authToken: {data}")
|
raise VendorPushError(f"vivo auth missing authToken: {_raw_log_summary(data)}")
|
||||||
return _cache_put(cache_key, str(token), 24 * 3600)
|
return _cache_put(cache_key, str(token), 24 * 3600)
|
||||||
|
|
||||||
|
|
||||||
@@ -401,6 +600,8 @@ def _send_vivo(token: str, title: str, body: str, extras: dict[str, str]) -> dic
|
|||||||
data = _request_json(
|
data = _request_json(
|
||||||
"POST",
|
"POST",
|
||||||
settings.VIVO_PUSH_SEND_ENDPOINT,
|
settings.VIVO_PUSH_SEND_ENDPOINT,
|
||||||
|
vendor="vivo",
|
||||||
|
operation="send_notification",
|
||||||
json=payload,
|
json=payload,
|
||||||
headers={
|
headers={
|
||||||
"Content-Type": "application/json",
|
"Content-Type": "application/json",
|
||||||
@@ -408,7 +609,7 @@ def _send_vivo(token: str, title: str, body: str, extras: dict[str, str]) -> dic
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
if int(data.get("result", -1)) != 0:
|
if int(data.get("result", -1)) != 0:
|
||||||
raise VendorPushError(f"vivo push failed: {data}")
|
raise VendorPushError(f"vivo push failed: {_raw_log_summary(data)}")
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
@@ -446,14 +647,16 @@ def _send_xiaomi(token: str, title: str, body: str, extras: dict[str, str]) -> d
|
|||||||
data = _request_form(
|
data = _request_form(
|
||||||
"POST",
|
"POST",
|
||||||
settings.XIAOMI_PUSH_SEND_ENDPOINT,
|
settings.XIAOMI_PUSH_SEND_ENDPOINT,
|
||||||
|
vendor="xiaomi",
|
||||||
|
operation="send_notification",
|
||||||
data=form,
|
data=form,
|
||||||
headers={"Authorization": f"key={app_secret}"},
|
headers={"Authorization": f"key={app_secret}"},
|
||||||
)
|
)
|
||||||
code = data.get("code")
|
code = data.get("code")
|
||||||
if code not in (0, "0", None):
|
if code not in (0, "0", None):
|
||||||
raise VendorPushError(f"xiaomi push failed: {data}")
|
raise VendorPushError(f"xiaomi push failed: {_raw_log_summary(data)}")
|
||||||
if str(data.get("result", "ok")).lower() not in ("ok", "success"):
|
if str(data.get("result", "ok")).lower() not in ("ok", "success"):
|
||||||
raise VendorPushError(f"xiaomi push failed: {data}")
|
raise VendorPushError(f"xiaomi push failed: {_raw_log_summary(data)}")
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
@@ -522,6 +725,8 @@ def _oppo_auth_token() -> str:
|
|||||||
data = _request_form(
|
data = _request_form(
|
||||||
"POST",
|
"POST",
|
||||||
settings.OPPO_PUSH_AUTH_ENDPOINT,
|
settings.OPPO_PUSH_AUTH_ENDPOINT,
|
||||||
|
vendor="oppo",
|
||||||
|
operation="authenticate",
|
||||||
data={
|
data={
|
||||||
"app_key": app_key,
|
"app_key": app_key,
|
||||||
"timestamp": timestamp,
|
"timestamp": timestamp,
|
||||||
@@ -530,10 +735,10 @@ def _oppo_auth_token() -> str:
|
|||||||
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
||||||
)
|
)
|
||||||
if int(data.get("code", -1)) != 0:
|
if int(data.get("code", -1)) != 0:
|
||||||
raise VendorPushError(f"oppo auth failed: {data}")
|
raise VendorPushError(f"oppo auth failed: {_raw_log_summary(data)}")
|
||||||
token = (data.get("data") or {}).get("auth_token") or data.get("auth_token")
|
token = (data.get("data") or {}).get("auth_token") or data.get("auth_token")
|
||||||
if not token:
|
if not token:
|
||||||
raise VendorPushError(f"oppo auth missing auth_token: {data}")
|
raise VendorPushError(f"oppo auth missing auth_token: {_raw_log_summary(data)}")
|
||||||
return _cache_put(cache_key, str(token), 24 * 3600)
|
return _cache_put(cache_key, str(token), 24 * 3600)
|
||||||
|
|
||||||
|
|
||||||
@@ -573,6 +778,8 @@ def _send_oppo(token: str, title: str, body: str, extras: dict[str, str]) -> dic
|
|||||||
data = _request_form(
|
data = _request_form(
|
||||||
"POST",
|
"POST",
|
||||||
settings.OPPO_PUSH_SEND_ENDPOINT,
|
settings.OPPO_PUSH_SEND_ENDPOINT,
|
||||||
|
vendor="oppo",
|
||||||
|
operation="send_notification",
|
||||||
data={
|
data={
|
||||||
"auth_token": auth_token,
|
"auth_token": auth_token,
|
||||||
"message": json.dumps(message, ensure_ascii=False),
|
"message": json.dumps(message, ensure_ascii=False),
|
||||||
@@ -580,5 +787,5 @@ def _send_oppo(token: str, title: str, body: str, extras: dict[str, str]) -> dic
|
|||||||
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
headers={"Content-Type": "application/x-www-form-urlencoded"},
|
||||||
)
|
)
|
||||||
if int(data.get("code", -1)) != 0:
|
if int(data.get("code", -1)) != 0:
|
||||||
raise VendorPushError(f"oppo push failed: {data}")
|
raise VendorPushError(f"oppo push failed: {_raw_log_summary(data)}")
|
||||||
return data
|
return data
|
||||||
|
|||||||
@@ -38,7 +38,7 @@ class GuideVideoPlay(Base):
|
|||||||
# start_play 捕获 IntegrityError 降级成"这次不放视频"。
|
# start_play 捕获 IntegrityError 降级成"这次不放视频"。
|
||||||
# 用 unique Index 而非 UniqueConstraint:与迁移里的 create_index 对齐(SQLite 加约束
|
# 用 unique Index 而非 UniqueConstraint:与迁移里的 create_index 对齐(SQLite 加约束
|
||||||
# 要整表重建),autogenerate 才不会每次报一条假 diff。
|
# 要整表重建),autogenerate 才不会每次报一条假 diff。
|
||||||
Index("uq_guide_video_play_user_seq", "user_id", "seq", unique=True),
|
Index("uq_guide_video_play_user_scene_seq", "user_id", "scene", "seq", unique=True),
|
||||||
)
|
)
|
||||||
|
|
||||||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||||||
|
|||||||
@@ -30,7 +30,11 @@ from app.models.app_config import AppConfig
|
|||||||
from app.models.guide_video import GuideVideoPlay
|
from app.models.guide_video import GuideVideoPlay
|
||||||
from app.repositories import wallet as crud_wallet
|
from app.repositories import wallet as crud_wallet
|
||||||
|
|
||||||
_KEY = "coupon_guide_video"
|
SCENES = ("coupon", "comparison")
|
||||||
|
_KEY_BY_SCENE = {
|
||||||
|
"coupon": "coupon_guide_video",
|
||||||
|
"comparison": "comparison_guide_video",
|
||||||
|
}
|
||||||
|
|
||||||
#: 金币流水 biz_type。客户端收益明细按它显示「新手引导视频奖励」。
|
#: 金币流水 biz_type。客户端收益明细按它显示「新手引导视频奖励」。
|
||||||
BIZ_TYPE = "guide_video"
|
BIZ_TYPE = "guide_video"
|
||||||
@@ -41,7 +45,7 @@ _DEFAULTS: dict[str, Any] = {
|
|||||||
"enabled": True,
|
"enabled": True,
|
||||||
"video_url": None, # None/空 = 未配片 → 不下发,浮层照旧放广告
|
"video_url": None, # None/空 = 未配片 → 不下发,浮层照旧放广告
|
||||||
"max_plays": 3, # 每个账号前 N 次浮层放引导视频
|
"max_plays": 3, # 每个账号前 N 次浮层放引导视频
|
||||||
"reward_coin": 120, # 每次固定金币
|
"reward_coin": 100, # 每次固定金币
|
||||||
}
|
}
|
||||||
|
|
||||||
_FIELDS = tuple(_DEFAULTS.keys())
|
_FIELDS = tuple(_DEFAULTS.keys())
|
||||||
@@ -65,19 +69,28 @@ def _merge(raw: Any) -> dict[str, Any]:
|
|||||||
return out
|
return out
|
||||||
|
|
||||||
|
|
||||||
def get_config(db: Session) -> dict[str, Any]:
|
def _config_key(scene: str) -> str:
|
||||||
|
if scene not in _KEY_BY_SCENE:
|
||||||
|
raise ValueError(f"unsupported guide video scene: {scene}")
|
||||||
|
return _KEY_BY_SCENE[scene]
|
||||||
|
|
||||||
|
|
||||||
|
def get_config(db: Session, scene: str = "coupon") -> dict[str, Any]:
|
||||||
"""完整配置 + updated_at(admin 读 / 业务读共用)。"""
|
"""完整配置 + updated_at(admin 读 / 业务读共用)。"""
|
||||||
row = db.get(AppConfig, _KEY)
|
row = db.get(AppConfig, _config_key(scene))
|
||||||
cfg = _merge(row.value if row is not None else None)
|
cfg = _merge(row.value if row is not None else None)
|
||||||
cfg["updated_at"] = row.updated_at.isoformat() if row is not None and row.updated_at else None
|
cfg["updated_at"] = row.updated_at.isoformat() if row is not None and row.updated_at else None
|
||||||
return cfg
|
return cfg
|
||||||
|
|
||||||
|
|
||||||
def _write(db: Session, value: dict[str, Any], *, admin_id: int, commit: bool) -> dict[str, Any]:
|
def _write(
|
||||||
|
db: Session, value: dict[str, Any], *, scene: str, admin_id: int, commit: bool
|
||||||
|
) -> dict[str, Any]:
|
||||||
"""整体覆写该行(value 须为完整字段 dict),返回合并后的完整配置(含 updated_at)。"""
|
"""整体覆写该行(value 须为完整字段 dict),返回合并后的完整配置(含 updated_at)。"""
|
||||||
row = db.get(AppConfig, _KEY)
|
key = _config_key(scene)
|
||||||
|
row = db.get(AppConfig, key)
|
||||||
if row is None:
|
if row is None:
|
||||||
row = AppConfig(key=_KEY, value=value, updated_by_admin_id=admin_id)
|
row = AppConfig(key=key, value=value, updated_by_admin_id=admin_id)
|
||||||
db.add(row)
|
db.add(row)
|
||||||
else:
|
else:
|
||||||
row.value = value # 整体重新赋值,SQLAlchemy 才侦测得到变更
|
row.value = value # 整体重新赋值,SQLAlchemy 才侦测得到变更
|
||||||
@@ -98,11 +111,12 @@ def update_config(
|
|||||||
enabled: bool | None = None,
|
enabled: bool | None = None,
|
||||||
max_plays: int | None = None,
|
max_plays: int | None = None,
|
||||||
reward_coin: int | None = None,
|
reward_coin: int | None = None,
|
||||||
|
scene: str = "coupon",
|
||||||
admin_id: int,
|
admin_id: int,
|
||||||
commit: bool = True,
|
commit: bool = True,
|
||||||
) -> tuple[dict[str, Any], dict[str, Any]]:
|
) -> tuple[dict[str, Any], dict[str, Any]]:
|
||||||
"""改开关 / 次数 / 金币(只改传了的字段;视频走 set_video)。返回 (before, after) 供审计。"""
|
"""改开关 / 次数 / 金币(只改传了的字段;视频走 set_video)。返回 (before, after) 供审计。"""
|
||||||
row = db.get(AppConfig, _KEY)
|
row = db.get(AppConfig, _config_key(scene))
|
||||||
before = _merge(row.value if row is not None else None)
|
before = _merge(row.value if row is not None else None)
|
||||||
new_value = {k: before[k] for k in _FIELDS}
|
new_value = {k: before[k] for k in _FIELDS}
|
||||||
if enabled is not None:
|
if enabled is not None:
|
||||||
@@ -111,45 +125,52 @@ def update_config(
|
|||||||
new_value["max_plays"] = max(0, min(int(max_plays), MAX_PLAYS_LIMIT))
|
new_value["max_plays"] = max(0, min(int(max_plays), MAX_PLAYS_LIMIT))
|
||||||
if reward_coin is not None:
|
if reward_coin is not None:
|
||||||
new_value["reward_coin"] = max(0, min(int(reward_coin), REWARD_COIN_LIMIT))
|
new_value["reward_coin"] = max(0, min(int(reward_coin), REWARD_COIN_LIMIT))
|
||||||
after = _write(db, new_value, admin_id=admin_id, commit=commit)
|
after = _write(db, new_value, scene=scene, admin_id=admin_id, commit=commit)
|
||||||
return before, after
|
return before, after
|
||||||
|
|
||||||
|
|
||||||
def set_video(
|
def set_video(
|
||||||
db: Session, video_url: str | None, *, admin_id: int, commit: bool = True
|
db: Session, video_url: str | None, *, scene: str = "coupon",
|
||||||
|
admin_id: int, commit: bool = True
|
||||||
) -> tuple[dict[str, Any], dict[str, Any]]:
|
) -> tuple[dict[str, Any], dict[str, Any]]:
|
||||||
"""设置/清空引导视频地址。返回 (before, after);before['video_url'] 供调用方删旧文件。"""
|
"""设置/清空引导视频地址。返回 (before, after);before['video_url'] 供调用方删旧文件。"""
|
||||||
row = db.get(AppConfig, _KEY)
|
row = db.get(AppConfig, _config_key(scene))
|
||||||
before = _merge(row.value if row is not None else None)
|
before = _merge(row.value if row is not None else None)
|
||||||
new_value = {k: before[k] for k in _FIELDS}
|
new_value = {k: before[k] for k in _FIELDS}
|
||||||
new_value["video_url"] = video_url
|
new_value["video_url"] = video_url
|
||||||
after = _write(db, new_value, admin_id=admin_id, commit=commit)
|
after = _write(db, new_value, scene=scene, admin_id=admin_id, commit=commit)
|
||||||
return before, after
|
return before, after
|
||||||
|
|
||||||
|
|
||||||
# ===== 播放计次 =====
|
# ===== 播放计次 =====
|
||||||
|
|
||||||
|
|
||||||
def used_plays(db: Session, user_id: int) -> int:
|
def used_plays(db: Session, user_id: int, scene: str = "coupon") -> int:
|
||||||
"""该账号已用掉的引导视频次数(开播即算,含未发币的)。"""
|
"""该账号已用掉的引导视频次数(开播即算,含未发币的)。"""
|
||||||
return int(
|
return int(
|
||||||
db.execute(
|
db.execute(
|
||||||
select(func.count()).select_from(GuideVideoPlay).where(
|
select(func.count()).select_from(GuideVideoPlay).where(
|
||||||
GuideVideoPlay.user_id == user_id
|
GuideVideoPlay.user_id == user_id,
|
||||||
|
GuideVideoPlay.scene == scene,
|
||||||
)
|
)
|
||||||
).scalar_one()
|
).scalar_one()
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def play_stats(db: Session) -> dict[str, int]:
|
def play_stats(db: Session, scene: str = "coupon") -> dict[str, int]:
|
||||||
"""全站播放统计(admin 页展示):总播放次数 / 其中已发币次数。"""
|
"""全站播放统计(admin 页展示):总播放次数 / 其中已发币次数。"""
|
||||||
total = int(
|
total = int(
|
||||||
db.execute(select(func.count()).select_from(GuideVideoPlay)).scalar_one()
|
db.execute(
|
||||||
|
select(func.count()).select_from(GuideVideoPlay).where(
|
||||||
|
GuideVideoPlay.scene == scene
|
||||||
|
)
|
||||||
|
).scalar_one()
|
||||||
)
|
)
|
||||||
granted = int(
|
granted = int(
|
||||||
db.execute(
|
db.execute(
|
||||||
select(func.count()).select_from(GuideVideoPlay).where(
|
select(func.count()).select_from(GuideVideoPlay).where(
|
||||||
GuideVideoPlay.status == "granted"
|
GuideVideoPlay.status == "granted",
|
||||||
|
GuideVideoPlay.scene == scene,
|
||||||
)
|
)
|
||||||
).scalar_one()
|
).scalar_one()
|
||||||
)
|
)
|
||||||
@@ -168,11 +189,12 @@ def start_play(
|
|||||||
reward_coin 播完/中途关闭都发的固定金币
|
reward_coin 播完/中途关闭都发的固定金币
|
||||||
seq / remaining 第几次 / 发完这次还剩几次(仅展示与排查用)
|
seq / remaining 第几次 / 发完这次还剩几次(仅展示与排查用)
|
||||||
"""
|
"""
|
||||||
cfg = get_config(db)
|
_config_key(scene)
|
||||||
|
cfg = get_config(db, scene)
|
||||||
video_url = (cfg.get("video_url") or "").strip()
|
video_url = (cfg.get("video_url") or "").strip()
|
||||||
max_plays = int(cfg.get("max_plays") or 0)
|
max_plays = int(cfg.get("max_plays") or 0)
|
||||||
reward_coin = int(cfg.get("reward_coin") or 0)
|
reward_coin = int(cfg.get("reward_coin") or 0)
|
||||||
used = used_plays(db, user_id)
|
used = used_plays(db, user_id, scene)
|
||||||
|
|
||||||
def _miss(used_now: int) -> dict[str, Any]:
|
def _miss(used_now: int) -> dict[str, Any]:
|
||||||
return {
|
return {
|
||||||
@@ -194,7 +216,7 @@ def start_play(
|
|||||||
scene=scene,
|
scene=scene,
|
||||||
seq=seq,
|
seq=seq,
|
||||||
video_url=video_url,
|
video_url=video_url,
|
||||||
coin=0,
|
coin=reward_coin,
|
||||||
status="playing",
|
status="playing",
|
||||||
completed=0,
|
completed=0,
|
||||||
started_at=datetime.now(rewards.CN_TZ).replace(tzinfo=None),
|
started_at=datetime.now(rewards.CN_TZ).replace(tzinfo=None),
|
||||||
@@ -210,7 +232,7 @@ def start_play(
|
|||||||
db.flush()
|
db.flush()
|
||||||
except IntegrityError:
|
except IntegrityError:
|
||||||
db.rollback()
|
db.rollback()
|
||||||
return _miss(used_plays(db, user_id))
|
return _miss(used_plays(db, user_id, scene))
|
||||||
return {
|
return {
|
||||||
"should_play": True,
|
"should_play": True,
|
||||||
"video_url": video_url,
|
"video_url": video_url,
|
||||||
@@ -241,7 +263,8 @@ def grant_play(
|
|||||||
"""
|
"""
|
||||||
token = (play_token or "").strip()
|
token = (play_token or "").strip()
|
||||||
# 金币额度以**服务端配置**为准,不信客户端(客户端只上报"播完/关闭")。
|
# 金币额度以**服务端配置**为准,不信客户端(客户端只上报"播完/关闭")。
|
||||||
coin = int(get_config(db).get("reward_coin") or 0)
|
play = _find_play(db, user_id, token)
|
||||||
|
coin = int(play.coin if play is not None else 0)
|
||||||
|
|
||||||
# 幂等核心:把 status 放进 WHERE 做条件更新(compare-and-set),而不是"先读再判再写"。
|
# 幂等核心:把 status 放进 WHERE 做条件更新(compare-and-set),而不是"先读再判再写"。
|
||||||
# 「播完」与「✕ 关闭」抢跑、或客户端超时重试时,两个请求会都读到 status='playing',
|
# 「播完」与「✕ 关闭」抢跑、或客户端超时重试时,两个请求会都读到 status='playing',
|
||||||
|
|||||||
@@ -1,13 +1,15 @@
|
|||||||
"""新手引导视频(领券等候浮层前 N 次替代广告)的客户端请求/响应契约。"""
|
"""新手引导视频(领券等候浮层前 N 次替代广告)的客户端请求/响应契约。"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from typing import Literal
|
||||||
|
|
||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
class GuideVideoStartIn(BaseModel):
|
class GuideVideoStartIn(BaseModel):
|
||||||
"""开播询问。scene 目前只有 coupon(领券浮层);预留给日后比价等场景。"""
|
"""开播询问。scene 目前只有 coupon(领券浮层);预留给日后比价等场景。"""
|
||||||
|
|
||||||
scene: str = Field(default="coupon", max_length=16)
|
scene: Literal["coupon", "comparison"] = "coupon"
|
||||||
|
|
||||||
|
|
||||||
class GuideVideoStartOut(BaseModel):
|
class GuideVideoStartOut(BaseModel):
|
||||||
|
|||||||
@@ -201,7 +201,7 @@ def test_start_loses_seq_race_degrades_to_ad(client, guide_configured, monkeypat
|
|||||||
assert first["should_play"] is True and first["seq"] == 1
|
assert first["should_play"] is True and first["seq"] == 1
|
||||||
|
|
||||||
# 本次请求读到的是过期计数 → 仍会算出 seq=1
|
# 本次请求读到的是过期计数 → 仍会算出 seq=1
|
||||||
monkeypatch.setattr(crud_guide, "used_plays", lambda db, user_id: 0)
|
monkeypatch.setattr(crud_guide, "used_plays", lambda db, user_id, scene="coupon": 0)
|
||||||
with SessionLocal() as db:
|
with SessionLocal() as db:
|
||||||
result = crud_guide.start_play(db, uid)
|
result = crud_guide.start_play(db, uid)
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,9 @@ mock/真发/变量覆盖/站内联动/设备反查/参数校验。厂商 HTTP
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
|
|
||||||
|
import httpx
|
||||||
import pytest
|
import pytest
|
||||||
from fastapi.testclient import TestClient
|
from fastapi.testclient import TestClient
|
||||||
|
|
||||||
@@ -110,6 +112,99 @@ def test_huawei_non_success_code_raises(monkeypatch) -> None:
|
|||||||
vendor_push.send_notification("huawei", "bad-token", title="t", body="b")
|
vendor_push.send_notification("huawei", "bad-token", title="t", body="b")
|
||||||
|
|
||||||
|
|
||||||
|
def test_vendor_http_logs_device_token_raw_response_and_elapsed(monkeypatch, caplog) -> None:
|
||||||
|
vendor_push._token_cache.clear()
|
||||||
|
|
||||||
|
def _fake_request(method, url, **kwargs): # noqa: ANN001
|
||||||
|
if url == vendor_push.settings.HUAWEI_PUSH_TOKEN_ENDPOINT:
|
||||||
|
return _Resp({"access_token": "raw-access-token", "expires_in": 3600})
|
||||||
|
return _Resp({"code": "80000000", "msg": "Success", "requestId": "vendor-request-1"})
|
||||||
|
|
||||||
|
monkeypatch.setattr(vendor_push.settings, "HUAWEI_PUSH_APP_ID", "10086001")
|
||||||
|
monkeypatch.setattr(vendor_push.settings, "HUAWEI_PUSH_APP_SECRET", "raw-app-secret")
|
||||||
|
monkeypatch.setattr(vendor_push.httpx, "request", _fake_request)
|
||||||
|
|
||||||
|
with caplog.at_level(logging.INFO, logger=vendor_push.logger.name):
|
||||||
|
vendor_push.send_notification(
|
||||||
|
"huawei",
|
||||||
|
"raw-device-token",
|
||||||
|
title="需要进入日志的标题",
|
||||||
|
body="需要进入日志的正文",
|
||||||
|
extras={"type": "withdraw_success", "notificationId": "90001"},
|
||||||
|
)
|
||||||
|
|
||||||
|
messages = "\n".join(record.getMessage() for record in caplog.records)
|
||||||
|
assert "operation=authenticate" in messages
|
||||||
|
assert "operation=send_notification" in messages
|
||||||
|
assert "request=" in messages
|
||||||
|
assert "response=" in messages
|
||||||
|
assert "elapsed_ms=" in messages
|
||||||
|
assert "vendor-request-1" in messages
|
||||||
|
assert "withdraw_success" in messages
|
||||||
|
assert "raw-access-token" in messages
|
||||||
|
assert "raw-device-token" in messages
|
||||||
|
assert "需要进入日志的标题" in messages
|
||||||
|
assert "需要进入日志的正文" in messages
|
||||||
|
assert "<redacted" in messages
|
||||||
|
assert "raw-app-secret" not in messages
|
||||||
|
send_http_log = next(
|
||||||
|
record.getMessage()
|
||||||
|
for record in caplog.records
|
||||||
|
if "operation=send_notification" in record.getMessage()
|
||||||
|
and "vendor push http completed" in record.getMessage()
|
||||||
|
)
|
||||||
|
assert "raw-device-token" in send_http_log
|
||||||
|
assert "raw-access-token" not in send_http_log # 请求头中的厂商鉴权 token 仍脱敏
|
||||||
|
|
||||||
|
|
||||||
|
def test_vendor_http_network_error_logs_request_context(monkeypatch, caplog) -> None:
|
||||||
|
def _fake_request(*args, **kwargs): # noqa: ANN002, ANN003
|
||||||
|
raise httpx.ConnectError("connection refused")
|
||||||
|
|
||||||
|
monkeypatch.setattr(vendor_push.settings, "XIAOMI_PUSH_APP_SECRET", "raw-xiaomi-secret")
|
||||||
|
monkeypatch.setattr(vendor_push.httpx, "request", _fake_request)
|
||||||
|
|
||||||
|
with (
|
||||||
|
caplog.at_level(logging.ERROR, logger=vendor_push.logger.name),
|
||||||
|
pytest.raises(vendor_push.VendorPushError, match="push http error"),
|
||||||
|
):
|
||||||
|
vendor_push.send_notification(
|
||||||
|
"xiaomi",
|
||||||
|
"raw-xiaomi-token",
|
||||||
|
title="title",
|
||||||
|
body="body",
|
||||||
|
extras={"type": "push_test"},
|
||||||
|
)
|
||||||
|
|
||||||
|
messages = "\n".join(record.getMessage() for record in caplog.records)
|
||||||
|
assert "outcome=network_error" in messages
|
||||||
|
assert "response=<no_response>" in messages
|
||||||
|
assert "operation=send_notification" in messages
|
||||||
|
assert "elapsed_ms=" in messages
|
||||||
|
assert "raw-xiaomi-secret" not in messages
|
||||||
|
assert "raw-xiaomi-token" in messages
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("request_kwargs", "device_token"),
|
||||||
|
[
|
||||||
|
({"json": {"token": ["honor-device-token"]}}, "honor-device-token"),
|
||||||
|
({"json": {"message": {"token": ["huawei-device-token"]}}}, "huawei-device-token"),
|
||||||
|
({"json": {"regId": "vivo-device-token"}}, "vivo-device-token"),
|
||||||
|
({"data": {"registration_id": "xiaomi-device-token"}}, "xiaomi-device-token"),
|
||||||
|
(
|
||||||
|
{"data": {"message": '{"target_value":"oppo-device-token"}'}},
|
||||||
|
"oppo-device-token",
|
||||||
|
),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_all_vendor_device_token_fields_remain_visible_in_request_summary(
|
||||||
|
request_kwargs, device_token
|
||||||
|
) -> None:
|
||||||
|
summary = vendor_push._request_summary(request_kwargs)
|
||||||
|
assert device_token in summary
|
||||||
|
|
||||||
|
|
||||||
def test_vendor_aliases_normalize() -> None:
|
def test_vendor_aliases_normalize() -> None:
|
||||||
assert vendor_push.normalize_vendor("华为") == "huawei"
|
assert vendor_push.normalize_vendor("华为") == "huawei"
|
||||||
assert vendor_push.normalize_vendor("HMS") == "huawei"
|
assert vendor_push.normalize_vendor("HMS") == "huawei"
|
||||||
|
|||||||
Reference in New Issue
Block a user