44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
|
"""从 URL 解析当前环境腾讯云 COS object key（仅当 host 与配置一致时）。"""
|
|||
|
|
|
|||
|
|
from urllib.parse import urlparse
|
|||
|
|
|
|||
|
|
from app.core.config import settings
|
|||
|
|
|
|||
|
|
|
|||
|
|
def extract_cos_object_key_if_owned(url: str | None) -> str | None:
|
|||
|
|
"""
|
|||
|
|
若 url 指向 settings 中配置的 COS 域名，返回去掉前导 / 的 object key。
|
|||
|
|
非 http(s)、或 host 不匹配时返回 None。
|
|||
|
|
"""
|
|||
|
|
if not url:
|
|||
|
|
return None
|
|||
|
|
s = str(url).strip()
|
|||
|
|
if not s.startswith(("http://", "https://")):
|
|||
|
|
return None
|
|||
|
|
parsed = urlparse(s)
|
|||
|
|
host = (parsed.netloc or "").lower()
|
|||
|
|
if not host:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
candidates: list[str] = []
|
|||
|
|
bucket = (settings.tencent_cos_bucket or "").strip().lower()
|
|||
|
|
region = (settings.tencent_cos_region or "").strip().lower()
|
|||
|
|
if bucket and region:
|
|||
|
|
candidates.append(f"{bucket}.cos.{region}.myqcloud.com")
|
|||
|
|
base = (settings.tencent_cos_base_url or "").strip()
|
|||
|
|
if base:
|
|||
|
|
base_parsed = urlparse(base if "://" in base else f"https://{base}")
|
|||
|
|
bh = (base_parsed.netloc or "").lower()
|
|||
|
|
if bh:
|
|||
|
|
candidates.append(bh)
|
|||
|
|
|
|||
|
|
if not candidates:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
matched = any(host == c for c in candidates if c)
|
|||
|
|
if not matched:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
key = (parsed.path or "").lstrip("/")
|
|||
|
|
return key or None
|