将 from app.services.tjnetwork import * 改为显式导入

2026-03-09 17:51:12 +08:00
parent c946e1b58b
commit 8c7d77e6ee
38 changed files with 845 additions and 38 deletions
@@ -1,6 +1,15 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    calculate_demand_to_network,
+    calculate_demand_to_nodes,
+    calculate_demand_to_region,
+    get_demand,
+    get_demand_schema,
+    set_demand,
+)

 router = APIRouter()

@@ -1,6 +1,42 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    delete_junction,
+    delete_pipe,
+    delete_pump,
+    delete_reservoir,
+    delete_tank,
+    delete_valve,
+    get_all_scada_info,
+    get_element_properties,
+    get_element_properties_with_type,
+    get_element_type,
+    get_element_type_value,
+    get_link_properties,
+    get_link_type,
+    get_links,
+    get_node_links,
+    get_node_properties,
+    get_node_type,
+    get_nodes,
+    get_scada_info,
+    get_status,
+    get_status_schema,
+    get_title,
+    get_title_schema,
+    is_junction,
+    is_link,
+    is_node,
+    is_pipe,
+    is_pump,
+    is_reservoir,
+    is_tank,
+    is_valve,
+    set_status,
+    set_title,
+)

 router = APIRouter()

@@ -1,6 +1,15 @@
 from fastapi import APIRouter, Request, Depends
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    get_all_scada_info,
+    get_major_node_coords,
+    get_major_pipe_nodes,
+    get_network_in_extent,
+    get_network_link_nodes,
+    get_network_node_coords,
+    get_node_coord,
+)
 from app.auth.dependencies import get_current_user as verify_token
 from app.infra.cache.redis_client import redis_client, encode_datetime, decode_datetime
 import msgpack
@@ -1,6 +1,15 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    add_junction,
+    delete_junction,
+    get_all_junctions,
+    get_junction,
+    get_junction_schema,
+    set_junction,
+)

 router = APIRouter()

@@ -1,6 +1,16 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    PIPE_STATUS_OPEN,
+    add_pipe,
+    delete_pipe,
+    get_all_pipes,
+    get_pipe,
+    get_pipe_schema,
+    set_pipe,
+)

 router = APIRouter()

@@ -1,6 +1,15 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    add_pump,
+    delete_pump,
+    get_all_pumps,
+    get_pump,
+    get_pump_schema,
+    set_pump,
+)

 router = APIRouter()

@@ -1,6 +1,42 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    add_district_metering_area,
+    add_region,
+    add_service_area,
+    add_virtual_district,
+    calculate_district_metering_area_for_network,
+    calculate_district_metering_area_for_nodes,
+    calculate_district_metering_area_for_region,
+    calculate_service_area,
+    calculate_virtual_district,
+    delete_district_metering_area,
+    delete_region,
+    delete_service_area,
+    delete_virtual_district,
+    generate_district_metering_area,
+    generate_service_area,
+    generate_sub_district_metering_area,
+    generate_virtual_district,
+    get_all_district_metering_area_ids,
+    get_all_district_metering_areas,
+    get_all_service_areas,
+    get_all_virtual_districts,
+    get_district_metering_area,
+    get_district_metering_area_schema,
+    get_region,
+    get_region_schema,
+    get_service_area,
+    get_service_area_schema,
+    get_virtual_district,
+    get_virtual_district_schema,
+    set_district_metering_area,
+    set_region,
+    set_service_area,
+    set_virtual_district,
+)

 router = APIRouter()

@@ -1,6 +1,15 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    add_reservoir,
+    delete_reservoir,
+    get_all_reservoirs,
+    get_reservoir,
+    get_reservoir_schema,
+    set_reservoir,
+)

 router = APIRouter()

@@ -1,6 +1,13 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    get_tag,
+    get_tag_schema,
+    get_tags,
+    set_tag,
+)

 router = APIRouter()

@@ -1,6 +1,15 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    add_tank,
+    delete_tank,
+    get_all_tanks,
+    get_tank,
+    get_tank_schema,
+    set_tank,
+)

 router = APIRouter()

@@ -1,6 +1,16 @@
 from fastapi import APIRouter, Request
 from typing import Any, List, Dict, Union
-from app.services.tjnetwork import *
+from app.services.tjnetwork import (
+    Any,
+    ChangeSet,
+    VALVES_TYPE_PRV,
+    add_valve,
+    delete_valve,
+    get_all_valves,
+    get_valve,
+    get_valve_schema,
+    set_valve,
+)

 router = APIRouter()