DjangoAdmin_FastAPI_EleVue/modules/region/service.py

# +======================================================================
# | DjangoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
# +----------------------------------------------------------------------
# | Copyright (c) 2021-2026 一米阳光团队 版权所有
# +----------------------------------------------------------------------
# | 开源协议: Apache-2.0
# | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
# +----------------------------------------------------------------------
# | 官方网站: https://www.djangoadmin.cn
# +----------------------------------------------------------------------
# | 研发团队: 一米阳光团队 荣誉出品
# +======================================================================
# | 版权和免责声明:
# | 一、知识产权声明
# | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
# | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
# | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
# | 二、使用限制与禁止条款
# | 您理解并同意，禁止将本产品用于以下任何行为：
# | 1. 任何违反中华人民共和国法律法规的用途；
# | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
# | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
# | 三、责任豁免
# | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
# | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
# | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
# | 四、违约与法律责任
# | 如发现任何违反本声明的行为，本团队有权：
# | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
# | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
# | 3. 保留进一步采取刑事控告等法律措施的权利。
# | 五、许可范围说明
# | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
# | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
# | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
# | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
# | 成的一切经济损失及商誉损害。
# | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
# | 许可协议》附件为准。
# +======================================================================

from datetime import datetime, timezone

from fastapi.logger import logger
from sqlalchemy import and_


from shared.enums import CITY_LEVEL_LIST
from modules.region.schemas import CityForm
from modules.region.models import City
from core.database import get_db_session
from core import R
from core.security import get_user_id
from utils.request import like_contains


# 查询城市数据

def get_city_page(request):
    db = get_db_session()
    # 实例化查询对象
    query = db.query(City).filter(City.is_delete == 0)
    # 上级ID
    pid = request.query_params.get('pid', 0)
    query = query.filter(City.pid == pid)
    # 城市名称
    name = request.query_params.get('name')
    if name:
        # 城市名称模糊查询
        query = query.filter(City.name.like(like_contains(name), escape='\\'))
    # 查询数据
    city_list = query.order_by(City.id.asc()).all()
    # 实例化数组对象
    result = []
    # 遍历数据源
    if city_list:
        for item in city_list:
            # 对象转字典
            data = item.to_dict()
            # 城市级别
            data['level_name'] = CITY_LEVEL_LIST.get(item.level)
            # 是否有子级
            data['haveChild'] = True if item.level < 3 else False,
            # 加入列表
            result.append(data)
    # 返回结果
    return R.ok(data=result)
# 根据城市ID获取详情

def get_city_detail(city_id):
    db = get_db_session()
    # 根据ID查询城市
    city = db.query(City).filter(and_(City.id == city_id, City.is_delete == 0)).first()
    # 查询结果判空
    if not city:
        return None
    # 对象转字典
    data = city.to_dict()
    # 返回结果
    return data
# 添加城市

def add_city(request, data: CityForm):
    db = get_db_session()
    # 表单数据赋值给对象
    city = City(**data.model_dump(exclude={'id'}))
    city.create_user = get_user_id(request)
    # 插入数据
    city.save()
    # 返回结果
    return R.ok(msg="添加成功")


# 更新城市

def update_city(request, data: CityForm):
    db = get_db_session()
    # 记录ID判空
    if not data.id or data.id <= 0:
        return R.failed("记录ID不能为空")

    # 根据ID查询记录
    city = db.query(City).filter(and_(City.id == data.id, City.is_delete == 0)).first()
    # 查询结果判空
    if not city:
        return R.failed("记录不存在")
    # 构造更新数据（排除 id）
    updData = data.model_dump(exclude={'id'})
    updData['update_user'] = get_user_id(request)
    updData['update_time'] = datetime.now(tz=timezone.utc)
    result = db.query(City).filter_by(id=data.id).update(updData)
    # 提交数据
    if not result:
        return R.failed("更新失败")
    # 返回结果
    return R.ok(msg="更新成功")
# 删除城市

def delete_city(city_id):
    return City.batch_delete_r(city_id)
# 根据城市ID获取子级城市

def get_child_list(city_code):
    db = get_db_session()
    # 查询子级城市
    childList = db.query(City).filter(and_(City.parent_code == city_code, City.is_delete == 0)).all()
    # 实例化子级城市对象
    city_list = []
    # 遍历数据源
    if childList:
        for v in childList:
            item = {
                'area_code': v.area_code,
                'name': v.name
            }
            # 加入列表
            city_list.append(item)
    # 返回结果
    return city_list