59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from datetime import datetime
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
from .lightning import LightningPolarity
|
||
|
|
|
||
|
|
|
||
|
|
class LightningImportBatchSummary(BaseModel):
|
||
|
|
"""文件导入批次摘要"""
|
||
|
|
batch_id: str # 使用 source_file_name + create_date 的哈希作为批次ID
|
||
|
|
source_file_name: str | None = None
|
||
|
|
import_time: datetime
|
||
|
|
event_count: int
|
||
|
|
region_id: str | None = None
|
||
|
|
location_tag: str | None = None
|
||
|
|
city: str | None = None
|
||
|
|
event_year: int | None = None
|
||
|
|
is_synthetic: bool = False
|
||
|
|
notes: str | None = None
|
||
|
|
|
||
|
|
# 统计信息
|
||
|
|
max_abs_current_ka: float | None = None
|
||
|
|
avg_abs_current_ka: float | None = None
|
||
|
|
positive_count: int = 0
|
||
|
|
negative_count: int = 0
|
||
|
|
|
||
|
|
# 导入元数据
|
||
|
|
create_user: str | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class LightningImportBatchListResponse(BaseModel):
|
||
|
|
items: list[LightningImportBatchSummary]
|
||
|
|
total: int
|
||
|
|
limit: int
|
||
|
|
offset: int
|
||
|
|
|
||
|
|
|
||
|
|
class LightningImportBatchEventItem(BaseModel):
|
||
|
|
"""导入批次中的单个事件"""
|
||
|
|
id: str
|
||
|
|
event_id: str
|
||
|
|
longitude: float | None = None
|
||
|
|
latitude: float | None = None
|
||
|
|
current_ka: float | None = None
|
||
|
|
abs_current_ka: float | None = None
|
||
|
|
polarity: LightningPolarity
|
||
|
|
event_time: datetime | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class LightningImportBatchEventsResponse(BaseModel):
|
||
|
|
batch_id: str
|
||
|
|
source_file_name: str | None = None
|
||
|
|
import_time: datetime
|
||
|
|
events: list[LightningImportBatchEventItem]
|
||
|
|
total: int
|