Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
import os
import json
from typing import Optional, IO, Tuple, Literal
from .common import request, Result
DateTimeTuple = Tuple[str, str]
def find(project_id: Optional[str] = None,
obs_id: Optional[str] = None,
module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'] = 'MSC',
detector_no: Optional[str] = None,
file_type: Optional[str] = None,
filter: Optional[str] = None,
obs_time: Optional[DateTimeTuple] = None,
create_time: Optional[DateTimeTuple] = None,
qc0_status: Optional[int] = None,
prc_status: Optional[int] = None,
file_name: Optional[str] = None,
ra_obj: Optional[int] = None,
dec_obj: Optional[int] = None,
radius: Optional[float] = None,
object_name: Optional[str] = None,
page: int = 1,
limit: int = 0) -> Result:
"""
根据给定的参数在搜索0级数据。
Args:
project_id (Optional[str], optional): 项目ID. Defaults to None.
obs_id (Optional[str], optional): 观测ID. Defaults to None.
module_id (Optional[str], optional): 模块ID，如'MSC', 'IFS'. Defaults to None.
detector_no (Optional[str], optional): 探测器编号. Defaults to None.
file_type (Optional[str], optional): 文件类型，如'SCI'. Defaults to None.
filter (Optional[str], optional): 滤光片. Defaults to None.
obs_time (Optional[DateTimeTuple], optional): 观测时间范围. Defaults to None.
create_time (Optional[DateTimeTuple], optional): 创建时间范围. Defaults to None.
qc0_status (Optional[int], optional): QC0状态. Defaults to None.
prc_status (Optional[int], optional): 处理状态. Defaults to None.
file_name (Optional[str], optional): 文件名. Defaults to None.
ra_obj (Optional[int], optional): 目标赤经. Defaults to None.
dec_obj (Optional[int], optional): 目标赤纬. Defaults to None.
radius (Optional[float], optional): 搜索半径. Defaults to None.
object_name (Optional[str], optional): 目标名称. Defaults to None.
page (int, optional): 页码. Defaults to 1.
limit (int, optional): 每页数量. Defaults to 0，不限制.
Returns:
Result: 搜索结果对象.
"""
params = {
'project_id': project_id,
'obs_id': obs_id,
'module_id': module_id,
'detector_no': detector_no,
'file_type': file_type,
'filter': filter,
'qc0_status': qc0_status,
'prc_status': prc_status,
'file_name': file_name,
'ra_obj': ra_obj,
'dec_obj': dec_obj,
'radius': radius,
'object_name': object_name,
'obs_time_start': None,
'obs_time_end': None,
'create_time_start': None,
'create_time_end': None,
'page': page,
'limit': limit,
}
if obs_time is not None:
params['obs_time_start'], params['obs_time_end'] = obs_time
if create_time is not None:
params['create_time_start'], params['create_time_end'] = create_time
return request.post("/api/level0", params)
def get_by_id(_id: str) -> Result:
return request.get(f"/api/level0/_id/{_id}")
def find_by_level0_id(level0_id: str) -> Result:
"""
通过 level0 的 ID 查询0级数据。
Args:
level0_id (str): 0级数据的ID。
Returns:
Result: 查询结果。
"""
return request.get(f"/api/level0/{level0_id}")
def update_qc0_status(level0_id: str, file_type: str, qc0_status: int) -> Result:
"""
更新0级数据的QC0状态
Args:
level0_id (str): 0级数据的ID
file_type (str): 文件类型
qc0_status (int): QC0状态
Returns:
Result: 更新结果
"""
return request.put(f"/api/level0/qc0_status/{level0_id}", {'file_type': file_type, 'qc0_status': qc0_status})
def update_prc_status(level0_id: str, file_type: str, prc_status: int) -> Result:
"""
更新0级数据的处理状态。
Args:
level0_id (str): 0级数据的ID。
file_type (str): 文件类型。
prc_status (int): 处理状态。
Returns:
Result: 操作结果。
"""
return request.put(f"/api/level0/prc_status/{level0_id}", {'file_type': file_type, 'prc_status': prc_status})
def write(local_file: str, **kwargs) -> Result:
"""
将本地文件写入DFS中。
Args:
local_file (str]): 文件路径。
**kwargs: 额外的关键字参数，这些参数将传递给DFS。
Returns:
Result: 操作的结果对象，包含操作是否成功以及相关的错误信息，成功返回数据对象。
"""
if not os.path.exists(local_file):
raise FileNotFoundError(local_file)
return request.post_file("/api/level0/file", local_file, kwargs)
def write_cat(local_file: str, **kwargs) -> Result:
"""
主巡天仿真数据的星表本地文件写入DFS中。
Args:
local_file (str]): 文件路径。
**kwargs: 额外的关键字参数，这些参数将传递给DFS。
Returns:
Result: 操作的结果对象，包含操作是否成功以及相关的错误信息，成功返回数据对象。
"""
if not os.path.exists(local_file):
raise FileNotFoundError(local_file)
return request.post_file("/api/level0/cat/file", local_file, kwargs)
def generate_prc_msg(module_id: Literal['MSC', 'IFS', 'MCI', 'HSTDM', 'CPIC'],
obs_id: str,
detector_no: str,
pipeline_id: Optional[str] = '') -> Result:
"""
生成流水线的处理消息。
Args:
module_id (str): 模块ID。
obs_id (str): 观测ID。
detector_no (str): 探测器编号。
pipeline_id (str): 流水管线ID，默认为空字符串。
Returns:
Result: 处理消息生成的结果，是否成功以及相关的错误信息。
"""
return request.put(f"/api/level0/prc/{module_id}/{obs_id}/{detector_no}", {'pipeline_id': pipeline_id})
def process_list(level0_id: str) -> Result:
"""
通过 level0 的 ID 查询0级数据处理过程。
Args:
level0_id (str): 0级数据的ID。
Returns:
Result: 成功后，Result.data为数据列表，失败message为失败原因。
"""
return request.get(f"/api/level0/prc/{level0_id}")
def add_process(level0_id: str,
pipeline_id: str,
run_id: str,
prc_time: str,
prc_status: int = -1024,
prc_module: str = "",
message: str = "") -> Result:
"""
添加0级数据处理过程。
Args:
level0_id (str): 0级数据的ID。
pipeline_id (str): 管线ID。
run_id (str): 运行ID。
prc_time (str): 处理时间,格式为"YYYY-MM-DD HH:MM:SS"。
prc_status (int): 处理状态。
prc_module (str): 处理模块。
message (str): 处理消息。
Returns:
Result: 成功后，Result.data为写入记录，失败message为失败原因。
"""
params = {
'level0_id': level0_id,
'pipeline_id': pipeline_id,
'run_id': run_id,
'prc_time': prc_time,
'prc_status': prc_status,
'prc_module': prc_module,
'message': message,
}
return request.post("/api/level0/prc", params)
def new(data: dict) -> Result:
"""
新建0级数据，用于仿真数据测试。
Args:
data (dict): 0级数据的字典表示。
Returns:
Result: 成功后，Result.data为写入记录，失败message为失败原因。
"""
return request.post("/api/level0/new", data)