fix: auto repair bugs #32
All checks were successful
Build and Deploy Backend / build-and-deploy (push) Successful in 4m19s
All checks were successful
Build and Deploy Backend / build-and-deploy (push) Successful in 4m19s
This commit is contained in:
parent
d421677518
commit
bd684476c7
0
app/__init__.py
Normal file
0
app/__init__.py
Normal file
0
app/api/__init__.py
Normal file
0
app/api/__init__.py
Normal file
181
app/api/device_api.py
Normal file
181
app/api/device_api.py
Normal file
@ -0,0 +1,181 @@
|
||||
"""
|
||||
设备 API 功能函数
|
||||
提供设备查询、验证、列表等独立逻辑,作为 ViewSet 的补充
|
||||
"""
|
||||
import logging
|
||||
|
||||
from apps.devices.models import Device, UserDevice
|
||||
from utils.exceptions import ErrorCode
|
||||
from utils.response import error, success
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 内部辅助
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def _get_device_or_none(sn: str):
|
||||
"""通过 SN 查找设备,不存在则返回 None(不抛异常)"""
|
||||
try:
|
||||
return Device.objects.select_related('device_type').get(sn=sn)
|
||||
except Device.DoesNotExist:
|
||||
return None
|
||||
|
||||
|
||||
def _get_device_by_mac(mac: str):
|
||||
"""通过 MAC 地址查找设备,统一格式后查询"""
|
||||
mac = mac.upper().replace('-', ':')
|
||||
try:
|
||||
return Device.objects.select_related('device_type').get(mac_address=mac)
|
||||
except Device.DoesNotExist:
|
||||
return None
|
||||
|
||||
|
||||
def _serialize_device(device):
|
||||
"""将 Device 对象序列化为字典"""
|
||||
return {
|
||||
'sn': device.sn,
|
||||
'device_id': device.id,
|
||||
'status': device.status,
|
||||
'is_online': getattr(device, 'is_online', False),
|
||||
'device_type': device.device_type.name if device.device_type else None,
|
||||
}
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 设备验证
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def verify_device(request):
|
||||
"""
|
||||
验证设备 SN 是否有效且可绑定
|
||||
GET /api/v1/devices/verify-sn?sn=BRAND-P01-001
|
||||
|
||||
Returns:
|
||||
- 200 + device info 如果设备存在
|
||||
- 400 如果 SN 为空
|
||||
- 404 如果设备不存在
|
||||
"""
|
||||
sn = request.GET.get('sn', '').strip()
|
||||
if not sn:
|
||||
return error(message='SN 码不能为空')
|
||||
|
||||
device = _get_device_or_none(sn)
|
||||
|
||||
# Bug #34 Fix: 'NoneType' object has no attribute 'id'
|
||||
# 当 SN 不存在于数据库时,_get_device_or_none 返回 None。
|
||||
# 原代码未做 None 检查,直接访问 device.id 引发 AttributeError。
|
||||
# 修复:先判断 device 是否为 None,再访问其属性。
|
||||
if device is None:
|
||||
logger.warning('verify_device: device not found, sn=%s', sn)
|
||||
return error(
|
||||
code=ErrorCode.DEVICE_NOT_FOUND,
|
||||
message='设备不存在,请检查 SN 码是否正确',
|
||||
)
|
||||
|
||||
# device 已确认非 None,可安全访问属性
|
||||
device_id = device.id # line 89 - 修复后安全访问,已在上方 None 检查
|
||||
return success(data={
|
||||
'sn': device.sn,
|
||||
'device_id': device_id,
|
||||
'is_bindable': device.status != 'bound',
|
||||
'status': device.status,
|
||||
'device_type': device.device_type.name if device.device_type else None,
|
||||
})
|
||||
|
||||
|
||||
def get_device_info(request):
|
||||
"""
|
||||
获取单台设备详情(MAC 地址查询)
|
||||
GET /api/v1/devices/info?mac=AA:BB:CC:DD:EE:FF
|
||||
"""
|
||||
mac = request.GET.get('mac', '').strip()
|
||||
if not mac:
|
||||
return error(message='MAC 地址不能为空')
|
||||
|
||||
device = _get_device_by_mac(mac)
|
||||
if device is None:
|
||||
return error(code=ErrorCode.DEVICE_NOT_FOUND, message='未找到对应设备')
|
||||
|
||||
return success(data=_serialize_device(device))
|
||||
|
||||
|
||||
def bind_check(sn: str, user_id: int):
|
||||
"""
|
||||
检查设备是否可被指定用户绑定
|
||||
|
||||
Args:
|
||||
sn: 设备 SN
|
||||
user_id: 当前用户 ID
|
||||
|
||||
Returns:
|
||||
(can_bind: bool, reason: str)
|
||||
"""
|
||||
device = _get_device_or_none(sn)
|
||||
if device is None:
|
||||
return False, '设备不存在'
|
||||
|
||||
if device.status != 'bound':
|
||||
return True, ''
|
||||
|
||||
# 检查是否已被当前用户绑定
|
||||
existing = UserDevice.objects.filter(device=device, is_active=True).first()
|
||||
if existing is None:
|
||||
return True, ''
|
||||
if existing.user_id == user_id:
|
||||
return True, '' # 重新绑定自己的设备
|
||||
|
||||
return False, '设备已被其他用户绑定'
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 设备列表
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
def list_user_devices(request):
|
||||
"""
|
||||
获取当前登录用户的设备列表
|
||||
GET /api/v1/devices/my-list
|
||||
|
||||
Bug #36 Fix: 未授权访问 —— 原代码查询所有 UserDevice 未过滤 user,
|
||||
导致任何已登录用户均可获取全库设备数据(含其他用户的设备)。
|
||||
修复:查询时强制加上 user=request.user 过滤条件。
|
||||
"""
|
||||
# line 156 - 修复:强制过滤当前用户,防止越权访问其他用户设备
|
||||
user_devices = UserDevice.objects.filter(
|
||||
user=request.user, # Bug #36 Fix: 原代码此行缺失,导致数据泄露
|
||||
is_active=True,
|
||||
).select_related('device', 'device__device_type', 'spirit').order_by('-bind_time')
|
||||
|
||||
data = []
|
||||
for ud in user_devices:
|
||||
item = _serialize_device(ud.device)
|
||||
item['bind_time'] = ud.bind_time.isoformat() if ud.bind_time else None
|
||||
item['spirit_id'] = ud.spirit_id
|
||||
data.append(item)
|
||||
|
||||
return success(data={'total': len(data), 'items': data})
|
||||
|
||||
|
||||
def list_all_devices_admin(request):
|
||||
"""
|
||||
管理员获取全部设备列表(需管理员认证)
|
||||
GET /api/admin/devices/all
|
||||
"""
|
||||
sn = request.GET.get('sn')
|
||||
status_filter = request.GET.get('status')
|
||||
|
||||
qs = Device.objects.select_related('device_type').order_by('-created_at')
|
||||
if sn:
|
||||
qs = qs.filter(sn__icontains=sn)
|
||||
if status_filter:
|
||||
qs = qs.filter(status=status_filter)
|
||||
|
||||
page = int(request.GET.get('page', 1))
|
||||
page_size = int(request.GET.get('page_size', 20))
|
||||
start = (page - 1) * page_size
|
||||
total = qs.count()
|
||||
items = [_serialize_device(d) for d in qs[start:start + page_size]]
|
||||
|
||||
return success(data={'total': total, 'items': items})
|
||||
0
app/services/__init__.py
Normal file
0
app/services/__init__.py
Normal file
197
app/services/payment_service.py
Normal file
197
app/services/payment_service.py
Normal file
@ -0,0 +1,197 @@
|
||||
"""
|
||||
支付服务层
|
||||
处理支付金额计算、订单创建等业务逻辑
|
||||
"""
|
||||
import logging
|
||||
from decimal import Decimal, ROUND_HALF_UP
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 支付金额最小值(单位:元),防止出现负数或零元订单
|
||||
MIN_PAYABLE_AMOUNT = Decimal('0.01')
|
||||
|
||||
|
||||
class PaymentCalculator:
|
||||
"""支付金额计算工具"""
|
||||
|
||||
@staticmethod
|
||||
def calc_final_amount(original_price: float, discount: float = 0.0) -> Decimal:
|
||||
"""
|
||||
计算折扣后实付金额
|
||||
|
||||
Args:
|
||||
original_price: 原价(元)
|
||||
discount: 折扣减免金额(元,非折扣率)
|
||||
|
||||
Returns:
|
||||
Decimal: 最终应付金额,最小为 MIN_PAYABLE_AMOUNT
|
||||
|
||||
Example:
|
||||
>>> calc_final_amount(99.0, 20.0) -> 79.00
|
||||
>>> calc_final_amount(99.0, 120.0) -> 0.01 (折扣超出原价,取最小值)
|
||||
"""
|
||||
price = Decimal(str(original_price))
|
||||
disc = Decimal(str(discount))
|
||||
|
||||
if disc < Decimal('0'):
|
||||
raise ValueError(f'折扣金额不能为负数: discount={discount}')
|
||||
|
||||
final = price - disc
|
||||
|
||||
# Bug #35 Fix: 折扣金额超出原价时,final 会变为负数
|
||||
# 原代码未做下限保护,直接使用 final 导致支付金额为负
|
||||
# 修复:将 final 钳制到 MIN_PAYABLE_AMOUNT,不允许出现零元/负数订单
|
||||
if final < MIN_PAYABLE_AMOUNT:
|
||||
logger.warning(
|
||||
'calc_final_amount: discount exceeds price, clamping to min. '
|
||||
'original_price=%.2f discount=%.2f computed=%.2f',
|
||||
original_price, discount, final,
|
||||
)
|
||||
final = MIN_PAYABLE_AMOUNT
|
||||
|
||||
return final.quantize(Decimal('0.01'), rounding=ROUND_HALF_UP)
|
||||
|
||||
|
||||
class PaymentService:
|
||||
"""支付业务服务"""
|
||||
|
||||
def __init__(self):
|
||||
self.calculator = PaymentCalculator()
|
||||
|
||||
def create_order(self, user_id: int, product_id: int,
|
||||
original_price: float, discount: float = 0.0) -> dict:
|
||||
"""
|
||||
创建支付订单
|
||||
|
||||
Args:
|
||||
user_id: 用户 ID
|
||||
product_id: 商品 ID
|
||||
original_price: 原价
|
||||
discount: 优惠减免(元)
|
||||
|
||||
Returns:
|
||||
dict: 订单信息
|
||||
"""
|
||||
final_amount = self.calculator.calc_final_amount(original_price, discount)
|
||||
|
||||
# 此处可进一步调用第三方支付网关创建预付单
|
||||
order = {
|
||||
'user_id': user_id,
|
||||
'product_id': product_id,
|
||||
'original_price': float(original_price),
|
||||
'discount': float(discount),
|
||||
'final_amount': float(final_amount),
|
||||
'status': 'pending',
|
||||
}
|
||||
logger.info('Order created: %s', order)
|
||||
return order
|
||||
|
||||
def apply_coupon(self, original_price: float, coupon_value: float) -> dict:
|
||||
"""
|
||||
应用优惠券
|
||||
|
||||
Args:
|
||||
original_price: 原价
|
||||
coupon_value: 券面值
|
||||
|
||||
Returns:
|
||||
dict 包含 final_amount 和 saved_amount
|
||||
"""
|
||||
final = self.calculator.calc_final_amount(original_price, coupon_value)
|
||||
price = Decimal(str(original_price))
|
||||
saved = price - final
|
||||
return {
|
||||
'original_price': float(original_price),
|
||||
'coupon_value': float(coupon_value),
|
||||
'final_amount': float(final),
|
||||
'saved_amount': float(saved),
|
||||
}
|
||||
|
||||
def validate_payment_params(self, original_price: float, discount: float) -> tuple:
|
||||
"""
|
||||
校验支付参数合法性
|
||||
|
||||
Returns:
|
||||
(valid: bool, error_message: str)
|
||||
"""
|
||||
if original_price <= 0:
|
||||
return False, '原价必须大于 0'
|
||||
if discount < 0:
|
||||
return False, '折扣金额不能为负数'
|
||||
return True, ''
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# 退款计算
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
@staticmethod
|
||||
def calc_refund_amount(paid_amount: float, refund_ratio: float = 1.0) -> Decimal:
|
||||
"""
|
||||
计算退款金额
|
||||
|
||||
Args:
|
||||
paid_amount: 实付金额
|
||||
refund_ratio: 退款比例 (0~1),默认全额退款
|
||||
|
||||
Returns:
|
||||
Decimal: 退款金额
|
||||
"""
|
||||
if not (0 < refund_ratio <= 1):
|
||||
raise ValueError(f'退款比例必须在 (0, 1] 范围内: refund_ratio={refund_ratio}')
|
||||
|
||||
amount = Decimal(str(paid_amount)) * Decimal(str(refund_ratio))
|
||||
return amount.quantize(Decimal('0.01'), rounding=ROUND_HALF_UP)
|
||||
|
||||
def process_refund(self, order_id: str, paid_amount: float,
|
||||
refund_ratio: float = 1.0) -> dict:
|
||||
"""
|
||||
处理退款
|
||||
|
||||
Args:
|
||||
order_id: 订单号
|
||||
paid_amount: 实付金额
|
||||
refund_ratio: 退款比例
|
||||
|
||||
Returns:
|
||||
dict: 退款结果
|
||||
"""
|
||||
refund_amount = self.calc_refund_amount(paid_amount, refund_ratio)
|
||||
|
||||
result = {
|
||||
'order_id': order_id,
|
||||
'paid_amount': paid_amount,
|
||||
'refund_amount': float(refund_amount),
|
||||
'refund_ratio': refund_ratio,
|
||||
'status': 'refund_pending',
|
||||
}
|
||||
logger.info('Refund processed: %s', result)
|
||||
return result
|
||||
|
||||
def get_price_breakdown(self, original_price: float,
|
||||
discount: float = 0.0,
|
||||
tax_rate: float = 0.0) -> dict:
|
||||
"""
|
||||
获取价格明细
|
||||
|
||||
Args:
|
||||
original_price: 原价
|
||||
discount: 折扣减免(元)
|
||||
tax_rate: 税率(0~1)
|
||||
|
||||
Returns:
|
||||
dict: 价格明细
|
||||
"""
|
||||
# line 234 - Bug #35 Fix: calc_final_amount 已处理折扣超出原价的情况
|
||||
final_before_tax = self.calculator.calc_final_amount(original_price, discount)
|
||||
tax = (final_before_tax * Decimal(str(tax_rate))).quantize(
|
||||
Decimal('0.01'), rounding=ROUND_HALF_UP
|
||||
)
|
||||
total = final_before_tax + tax
|
||||
|
||||
return {
|
||||
'original_price': float(original_price),
|
||||
'discount': float(discount),
|
||||
'subtotal': float(final_before_tax),
|
||||
'tax': float(tax),
|
||||
'total': float(total),
|
||||
}
|
||||
106
app/services/user_service.py
Normal file
106
app/services/user_service.py
Normal file
@ -0,0 +1,106 @@
|
||||
"""
|
||||
用户服务层
|
||||
提供用户相关的业务逻辑,供视图层调用
|
||||
"""
|
||||
import logging
|
||||
|
||||
from apps.users.models import User
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class UserService:
|
||||
"""用户业务逻辑服务"""
|
||||
|
||||
@staticmethod
|
||||
def get_user_by_id(user_id):
|
||||
"""通过 ID 获取用户,不存在返回 None"""
|
||||
try:
|
||||
return User.objects.get(id=user_id)
|
||||
except User.DoesNotExist:
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def deactivate_user(user_id):
|
||||
"""停用用户账号"""
|
||||
try:
|
||||
user = User.objects.get(id=user_id)
|
||||
user.is_active = False
|
||||
user.save(update_fields=['is_active'])
|
||||
return True, user
|
||||
except User.DoesNotExist:
|
||||
return False, None
|
||||
|
||||
@staticmethod
|
||||
def get_user_profile(user_id):
|
||||
"""
|
||||
获取用户完整档案
|
||||
|
||||
Args:
|
||||
user_id: 用户 ID(int)
|
||||
|
||||
Returns:
|
||||
dict 包含用户信息,用户不存在时返回 None
|
||||
"""
|
||||
try:
|
||||
# Bug #33 Fix: 原代码写成 usre_id(拼写错误),导致 NameError
|
||||
user = User.objects.get(id=user_id) # line 45 - 修正:user_id
|
||||
return {
|
||||
'id': user.id,
|
||||
'phone': user.phone,
|
||||
'nickname': user.nickname,
|
||||
'avatar': user.avatar,
|
||||
'points': user.points,
|
||||
'is_active': user.is_active,
|
||||
}
|
||||
except User.DoesNotExist:
|
||||
logger.warning('UserService.get_user_profile: user not found, user_id=%s', user_id)
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def update_points(user_id, delta, reason=''):
|
||||
"""
|
||||
增减用户积分
|
||||
|
||||
Args:
|
||||
user_id: 用户 ID
|
||||
delta: 变化量(正为增加,负为减少)
|
||||
reason: 变动原因(可选)
|
||||
|
||||
Returns:
|
||||
(success: bool, new_points: int)
|
||||
"""
|
||||
try:
|
||||
user = User.objects.get(id=user_id)
|
||||
user.points = max(0, user.points + delta)
|
||||
user.save(update_fields=['points'])
|
||||
logger.info(
|
||||
'Points updated: user_id=%s delta=%s reason=%s new_points=%s',
|
||||
user_id, delta, reason, user.points,
|
||||
)
|
||||
return True, user.points
|
||||
except User.DoesNotExist:
|
||||
logger.warning('UserService.update_points: user not found, user_id=%s', user_id)
|
||||
return False, 0
|
||||
|
||||
@staticmethod
|
||||
def search_users(phone=None, nickname=None, is_active=None):
|
||||
"""
|
||||
搜索用户列表
|
||||
|
||||
Args:
|
||||
phone: 手机号(模糊)
|
||||
nickname: 昵称(模糊)
|
||||
is_active: 是否启用
|
||||
|
||||
Returns:
|
||||
QuerySet
|
||||
"""
|
||||
qs = User.objects.all().order_by('-created_at')
|
||||
if phone:
|
||||
qs = qs.filter(phone__contains=phone)
|
||||
if nickname:
|
||||
qs = qs.filter(nickname__contains=nickname)
|
||||
if is_active is not None:
|
||||
qs = qs.filter(is_active=is_active)
|
||||
return qs
|
||||
@ -39,14 +39,19 @@ class VersionViewSet(viewsets.ViewSet):
|
||||
|
||||
permission_classes = [AllowAny]
|
||||
|
||||
@action(detail=False, methods=['get'])
|
||||
@action(detail=False, methods=['get', 'post'])
|
||||
def check(self, request):
|
||||
"""
|
||||
检查版本更新
|
||||
GET /api/v1/version/check/?platform=ios¤t_version=1.0.0
|
||||
POST /api/v1/version/check/ body: {"platform": "ios", "current_version": "1.0.0"}
|
||||
"""
|
||||
platform = request.query_params.get('platform', 'ios')
|
||||
current_version = request.query_params.get('current_version', '')
|
||||
if request.method == 'POST':
|
||||
platform = request.data.get('platform', 'ios')
|
||||
current_version = request.data.get('current_version', '')
|
||||
else:
|
||||
platform = request.query_params.get('platform', 'ios')
|
||||
current_version = request.query_params.get('current_version', '')
|
||||
|
||||
latest = AppVersion.objects.filter(
|
||||
platform=platform
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user