观测点接口逻辑
This commit is contained in:
@@ -2,10 +2,12 @@ from sqlalchemy.orm import Session
|
||||
from typing import List, Optional, Dict, Any
|
||||
from ..models.checkpoint import Checkpoint
|
||||
from .base import BaseService
|
||||
from .section_data import SectionDataService
|
||||
|
||||
class CheckpointService(BaseService[Checkpoint]):
|
||||
def __init__(self):
|
||||
super().__init__(Checkpoint)
|
||||
self.section_service = SectionDataService()
|
||||
|
||||
def get_by_section_id(self, db: Session, section_id: str) -> List[Checkpoint]:
|
||||
"""根据断面ID获取观测点"""
|
||||
@@ -34,6 +36,7 @@ class CheckpointService(BaseService[Checkpoint]):
|
||||
def batch_import_checkpoints(self, db: Session, data: List) -> Dict[str, Any]:
|
||||
"""
|
||||
批量导入观测点数据,根据观测点ID判断是否重复,重复数据改为更新操作
|
||||
判断断面id是否存在,不存在则全部不导入
|
||||
支持事务回滚,失败时重试一次
|
||||
"""
|
||||
import logging
|
||||
@@ -55,12 +58,11 @@ class CheckpointService(BaseService[Checkpoint]):
|
||||
try:
|
||||
|
||||
# 判断断面id是否存在
|
||||
section = self.get_by_section_id(db, item_data.get('section_id'))
|
||||
section = self.section_service.get_by_section_id(db, item_data.get('section_id'))
|
||||
if not section:
|
||||
logger.error(f"Section {item_data.get('section_id')} not found")
|
||||
raise Exception(f"Section {item_data.get('section_id')} not found")
|
||||
|
||||
|
||||
checkpoint = self.get_by_point_id(db, item_data.get('point_id'))
|
||||
if checkpoint:
|
||||
# 更新操作
|
||||
|
||||
Reference in New Issue
Block a user