refactor(api): unify scheme query endpoints
This commit is contained in:
@@ -1312,8 +1312,34 @@ def get_scheme_schema(name: str) -> dict[str, dict[str, Any]]:
|
||||
def get_scheme(name: str, schema_name: str) -> dict[str, Any]:
|
||||
return api.get_scheme(name, schema_name)
|
||||
|
||||
def get_all_schemes(name: str) -> list[dict[str, Any]]:
|
||||
return api.get_all_schemes(name)
|
||||
def get_all_schemes(
|
||||
name: str,
|
||||
scheme_type: str | None = None,
|
||||
query_date: Any | None = None,
|
||||
) -> list[dict[str, Any]]:
|
||||
if scheme_type is None and query_date is None:
|
||||
return api.get_all_schemes(name)
|
||||
|
||||
from app.services.scheme_management import query_scheme_list
|
||||
|
||||
rows = query_scheme_list(name, scheme_type=scheme_type, query_date=query_date) or []
|
||||
columns = [
|
||||
"scheme_id",
|
||||
"scheme_name",
|
||||
"scheme_type",
|
||||
"username",
|
||||
"create_time",
|
||||
"scheme_start_time",
|
||||
"scheme_detail",
|
||||
]
|
||||
result = []
|
||||
for row in rows:
|
||||
item = dict(zip(columns, row, strict=False))
|
||||
detail = item.get("scheme_detail")
|
||||
if isinstance(detail, dict) and detail.get("network") not in (None, name):
|
||||
continue
|
||||
result.append(item)
|
||||
return result
|
||||
|
||||
############################################################
|
||||
# pipe_risk_probability 41
|
||||
@@ -1345,5 +1371,3 @@ def get_all_sensor_placements(name: str) -> list[dict[Any, Any]]:
|
||||
def get_all_burst_locate_results(name: str) -> list[dict[Any, Any]]:
|
||||
return api.get_all_burst_locate_results(name)
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user