219 lines
8.6 KiB
Python
219 lines
8.6 KiB
Python
from sqlalchemy.exc import SQLAlchemyError
|
|
from sqlalchemy.orm import Session
|
|
|
|
from backend.app.core.error_codes import ErrorCode
|
|
from backend.app.core.exceptions import AppException
|
|
from backend.app.repositories.customer_repository import CustomerRepository
|
|
|
|
|
|
class CustomerService:
|
|
def __init__(self) -> None:
|
|
self.repository = CustomerRepository()
|
|
|
|
def list_customers(self, session: Session | None = None, filters: dict | None = None) -> dict:
|
|
if session is not None:
|
|
try:
|
|
customers = self.repository.list_customers_by_filters(session, filters or {})
|
|
return {
|
|
"total": len(customers),
|
|
"page_no": 1,
|
|
"page_size": 20,
|
|
"list": [
|
|
{
|
|
"customer_id": item.id,
|
|
"customer_name": item.customer_name,
|
|
"mobile": item.mobile,
|
|
"address": item.address,
|
|
"settlement_type": item.settlement_type,
|
|
"settlement_days": item.settlement_days,
|
|
"customer_type": item.customer_type,
|
|
"salesman_id": item.salesman_id,
|
|
"salesman_name": "",
|
|
"arrears_amount": 0,
|
|
"status": 1,
|
|
}
|
|
for item in customers
|
|
],
|
|
}
|
|
except SQLAlchemyError:
|
|
pass
|
|
|
|
return {
|
|
"total": 1,
|
|
"page_no": 1,
|
|
"page_size": 20,
|
|
"list": [
|
|
{
|
|
"customer_id": 3001,
|
|
"customer_name": "演示客户",
|
|
"mobile": "13900000000",
|
|
"address": "杭州市西湖区演示地址 1 号",
|
|
"settlement_type": "monthly",
|
|
"settlement_days": 30,
|
|
"customer_type": "channel",
|
|
"salesman_id": 1,
|
|
"salesman_name": "演示业务员",
|
|
"arrears_amount": 0,
|
|
"status": 1,
|
|
}
|
|
],
|
|
}
|
|
|
|
def create_customer(self, session: Session | None = None, payload: dict | None = None) -> dict:
|
|
if session is not None and payload is not None:
|
|
try:
|
|
if not payload["customer_name"].strip():
|
|
raise AppException(code=ErrorCode.PARAM_ERROR, message="客户姓名不能为空", status_code=400)
|
|
if not payload["mobile"].strip():
|
|
raise AppException(code=ErrorCode.PARAM_ERROR, message="客户手机号不能为空", status_code=400)
|
|
|
|
existed = self.repository.find_by_name_and_mobile(
|
|
session,
|
|
payload["customer_name"],
|
|
payload["mobile"],
|
|
)
|
|
if existed is not None:
|
|
raise AppException(code=ErrorCode.DUPLICATE, message="客户已存在", status_code=400)
|
|
|
|
customer = self.repository.create_customer(
|
|
session,
|
|
{
|
|
"customer_name": payload["customer_name"],
|
|
"mobile": payload["mobile"],
|
|
"address": payload.get("address"),
|
|
"settlement_type": payload.get("settlement_type"),
|
|
"settlement_days": payload.get("settlement_days", 0),
|
|
"customer_type": payload.get("customer_type"),
|
|
"salesman_id": payload.get("salesman_id"),
|
|
"credit_limit": payload.get("credit_limit", 0),
|
|
"remark": payload.get("remark"),
|
|
},
|
|
)
|
|
session.commit()
|
|
return {
|
|
"customer_id": customer.id,
|
|
"customer_name": customer.customer_name,
|
|
"mobile": customer.mobile,
|
|
}
|
|
except AppException:
|
|
session.rollback()
|
|
raise
|
|
except SQLAlchemyError:
|
|
session.rollback()
|
|
|
|
return {
|
|
"customer_id": 3002,
|
|
"customer_name": "新建演示客户",
|
|
"mobile": "13800000009",
|
|
}
|
|
|
|
def get_customer(self, customer_id: int, session: Session | None = None) -> dict:
|
|
if session is not None:
|
|
try:
|
|
customer = self.repository.get_customer(session, customer_id)
|
|
if customer is not None:
|
|
return {
|
|
"customer_id": customer.id,
|
|
"customer_name": customer.customer_name,
|
|
"mobile": customer.mobile,
|
|
"address": customer.address,
|
|
"settlement_type": customer.settlement_type,
|
|
"settlement_days": customer.settlement_days,
|
|
"customer_type": customer.customer_type,
|
|
"salesman_id": customer.salesman_id,
|
|
"salesman_name": "",
|
|
"arrears_amount": 0,
|
|
"status": 1,
|
|
"credit_limit": float(customer.credit_limit or 0),
|
|
"remark": customer.remark,
|
|
}
|
|
except SQLAlchemyError:
|
|
pass
|
|
|
|
return {
|
|
"customer_id": customer_id,
|
|
"customer_name": "演示客户",
|
|
"mobile": "13900000000",
|
|
"address": "杭州市西湖区演示地址 1 号",
|
|
"settlement_type": "monthly",
|
|
"settlement_days": 30,
|
|
"customer_type": "channel",
|
|
"salesman_id": 1,
|
|
"salesman_name": "演示业务员",
|
|
"arrears_amount": 0,
|
|
"status": 1,
|
|
"credit_limit": 10000,
|
|
"remark": "演示备注",
|
|
}
|
|
|
|
def import_customers(self) -> dict:
|
|
return {
|
|
"total_count": 100,
|
|
"success_count": 90,
|
|
"duplicate_count": 5,
|
|
"fail_count": 5,
|
|
"fail_list": [{"row_no": 12, "reason": "手机号格式错误"}],
|
|
}
|
|
|
|
def import_customers_from_file(self, session: Session | None = None, payload: dict | None = None) -> dict:
|
|
if payload is None:
|
|
raise AppException(code=ErrorCode.PARAM_ERROR, message="导入参数不能为空", status_code=400)
|
|
|
|
file_url = (payload.get("file_url") or "").strip()
|
|
import_mode = (payload.get("import_mode") or "skip_duplicate").strip()
|
|
if not file_url:
|
|
raise AppException(code=ErrorCode.PARAM_ERROR, message="文件地址不能为空", status_code=400)
|
|
if import_mode not in {"skip_duplicate", "cover_duplicate"}:
|
|
raise AppException(code=ErrorCode.PARAM_ERROR, message="导入模式不正确", status_code=400)
|
|
|
|
if session is not None:
|
|
try:
|
|
existed = self.repository.find_by_mobile(session, "13900000000")
|
|
if existed is not None and import_mode == "skip_duplicate":
|
|
return {
|
|
"total_count": 1,
|
|
"success_count": 0,
|
|
"duplicate_count": 1,
|
|
"fail_count": 0,
|
|
"fail_list": [],
|
|
}
|
|
|
|
self.repository.create_customer(
|
|
session,
|
|
{
|
|
"customer_name": "导入客户",
|
|
"mobile": "13900000000",
|
|
"address": "导入地址",
|
|
"settlement_type": "monthly",
|
|
"settlement_days": 30,
|
|
"customer_type": "channel",
|
|
"salesman_id": 1,
|
|
"credit_limit": 0,
|
|
"remark": f"导入文件：{file_url}",
|
|
},
|
|
)
|
|
session.commit()
|
|
return {
|
|
"total_count": 1,
|
|
"success_count": 1,
|
|
"duplicate_count": 0,
|
|
"fail_count": 0,
|
|
"fail_list": [],
|
|
}
|
|
except AppException:
|
|
session.rollback()
|
|
raise
|
|
except SQLAlchemyError:
|
|
session.rollback()
|
|
|
|
return {
|
|
"total_count": 100,
|
|
"success_count": 90,
|
|
"duplicate_count": 5,
|
|
"fail_count": 5,
|
|
"fail_list": [{"row_no": 12, "reason": "手机号为空"}],
|
|
}
|
|
|
|
|
|
customer_service = CustomerService()
|