74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
"""公司业务服务：查找或创建公司基础记录，上传 Logo。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import threading
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import insert, text
|
|
|
|
from app.core.database import MysqlSession
|
|
from app.core.id_gen import next_id
|
|
from app.core.logger import log
|
|
from app.core.oss import upload_image_url
|
|
from app.models.company import Company
|
|
|
|
# 公司创建锁（防止并发重复插入同一公司）
|
|
_company_lock = threading.Lock()
|
|
|
|
|
|
def find_or_create_company(company_name: str, logo_url: str | None = None) -> int:
|
|
"""查找或创建公司，上传 Logo 并回填地址。
|
|
|
|
Args:
|
|
company_name: 公司名称（用于查重和创建）。
|
|
logo_url: Logo 图片原始地址，非空时下载并上传 OSS。
|
|
|
|
Returns:
|
|
公司 ID。
|
|
"""
|
|
if not company_name:
|
|
company_name = "未知公司"
|
|
|
|
with _company_lock:
|
|
with MysqlSession() as session:
|
|
row = session.execute(
|
|
text("SELECT id FROM bg_company WHERE short_name = :name LIMIT 1"),
|
|
{"name": company_name},
|
|
)
|
|
existing = row.scalar()
|
|
if existing:
|
|
return existing
|
|
|
|
company_id = next_id()
|
|
now = datetime.now()
|
|
session.execute(
|
|
insert(Company).values(
|
|
id=company_id,
|
|
name=company_name,
|
|
short_name=company_name,
|
|
status=0,
|
|
create_time=now,
|
|
update_time=now,
|
|
)
|
|
)
|
|
session.commit()
|
|
log.info("公司创建成功 [id={}]: {}", company_id, company_name)
|
|
|
|
# 锁外处理 Logo 上传
|
|
if logo_url:
|
|
try:
|
|
oss_url = upload_image_url(logo_url)
|
|
if oss_url:
|
|
with MysqlSession() as session:
|
|
session.execute(
|
|
text("UPDATE bg_company SET logo_url = :url, update_time = :t WHERE id = :id"),
|
|
{"url": oss_url, "t": datetime.now(), "id": company_id},
|
|
)
|
|
session.commit()
|
|
log.info("公司 Logo 上传成功 [id={}]: {}", company_id, oss_url)
|
|
except Exception as exc:
|
|
log.warning("公司 Logo 上传失败 [id={}]: {}", company_id, exc)
|
|
|
|
return company_id
|