OrgTJWater/TJWaterServerBinary
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复引用错误

Browse Source
This commit is contained in:
Jiang
2026-03-09 18:11:24 +08:00
parent 0d8a7f5cb7
commit 7345210bdd
3 changed files with 176 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/native/wndb/inp_in.py
+96 -65
View File
@@ -1,6 +1,6 @@
import datetime
import os
from .project_backup import *
from .project import *
from .database import ChangeSet, write
from .sections import *
from .s0_base import get_region_type
@@ -38,11 +38,13 @@ from .s32_region_util import from_postgis_polygon,to_postgis_polygon
# DingZQ, 2024-12-28, export inp
from .inp_out import export_inp
_S = 'S'
_L = 'L'
_S = "S"
_L = "L"
def _inp_in_option(section: list[str], version: str = "3") -> str:
return inp_in_option_v3(section) if version == "3" else inp_in_option(section)
def _inp_in_option(section: list[str], version: str = '3') -> str:
return inp_in_option_v3(section) if version == '3' else inp_in_option(section)
_handler = {
TITLE: (_S, inp_in_title),
@@ -122,12 +124,14 @@ _level_4 = [
map_regiontype = {
# map the region types from desktop to server
'DISTRIBUTION':'WDA',
'DMA':'DMA',
'PMA':'PMA',
'VD':'VD',
'SA':'SA',
"DISTRIBUTION": "WDA",
"DMA": "DMA",
"PMA": "PMA",
"VD": "VD",
"SA": "SA",
}
class SQLBatch:
def __init__(self, project: str, count: int = 100) -> None:
self.batch: list[str] = []
@@ -140,13 +144,13 @@ class SQLBatch:
self.flush()
def flush(self) -> None:
write(self.project, ''.join(self.batch))
write(self.project, "".join(self.batch))
self.batch.clear()
def _print_time(desc: str) -> datetime.datetime:
now = datetime.datetime.now()
time = now.strftime('%Y-%m-%d %H:%M:%S')
time = now.strftime("%Y-%m-%d %H:%M:%S")
print(f"{time}: {desc}")
return now
@@ -154,7 +158,7 @@ def _print_time(desc: str) -> datetime.datetime:
def _get_file_offset(inp: str) -> tuple[dict[str, list[int]], bool]:
offset: dict[str, list[int]] = {}
current = ''
current = ""
demand_outside = False
with open(inp) as f:
@@ -164,22 +168,22 @@ def _get_file_offset(inp: str) -> tuple[dict[str, list[int]], bool]:
break
line = line.strip()
if line.startswith('['):
if line.startswith("["):
for s in section_name:
if line.startswith(f'[{s}'):
if line.startswith(f"[{s}"):
if s not in offset:
offset[s] = []
offset[s].append(f.tell())
current = s
break
elif line != '' and line.startswith(';') == False:
elif line != "" and line.startswith(";") == False:
if current == DEMANDS:
demand_outside = True
return (offset, demand_outside)
def parse_file(project: str, inp: str, version: str = '3') -> None:
def parse_file(project: str, inp: str, version: str = "3") -> None:
start = _print_time(f'Start reading file "{inp}"...')
_print_time("First scan...")
@@ -228,17 +232,17 @@ def parse_file(project: str, inp: str, version: str = '3') -> None:
break
line = line.strip()
if line.startswith('['):
if line.startswith("["):
break
elif line == '':
elif line == "":
continue
if is_s:
sections[s].append(line)
else:
if line.startswith(';'):
if version != '3': #v2
line = line.removeprefix(';')
if line.startswith(";"):
if version != "3": # v2
line = line.removeprefix(";")
if s == PATTERNS: # ;desc
pass
elif s == CURVES: # ;type: desc
@@ -249,29 +253,37 @@ def parse_file(project: str, inp: str, version: str = '3') -> None:
tokens = line.split()
if tokens[1].upper() in pattern_v3_types: # v3
sql_batch.add(f"insert into _pattern (id) values ('{tokens[0]}');")
sql_batch.add(
f"insert into _pattern (id) values ('{tokens[0]}');"
)
current_pattern = tokens[0]
if tokens[1].upper() == 'VARIABLE':
if tokens[1].upper() == "VARIABLE":
variable_patterns.append(tokens[0])
continue
if current_pattern != tokens[0]:
sql_batch.add(f"insert into _pattern (id) values ('{tokens[0]}');")
sql_batch.add(
f"insert into _pattern (id) values ('{tokens[0]}');"
)
current_pattern = tokens[0]
elif s == CURVES:
tokens = line.split()
if tokens[1].upper() in curve_types: # v3
sql_batch.add(f"insert into _curve (id, type) values ('{tokens[0]}', '{tokens[1].upper()}');")
sql_batch.add(
f"insert into _curve (id, type) values ('{tokens[0]}', '{tokens[1].upper()}');"
)
current_curve = tokens[0]
continue
if current_curve != tokens[0]:
type = curve_types[0]
if curve_type_desc_line != None:
type = curve_type_desc_line.split(':')[0].strip()
sql_batch.add(f"insert into _curve (id, type) values ('{tokens[0]}', '{type}');")
type = curve_type_desc_line.split(":")[0].strip()
sql_batch.add(
f"insert into _curve (id, type) values ('{tokens[0]}', '{type}');"
)
current_curve = tokens[0]
curve_type_desc_line = None
elif s == REGION:
@@ -279,12 +291,14 @@ def parse_file(project: str, inp: str, version: str = '3') -> None:
region_list[tokens[0]] = tokens[1]
elif s == BOUND:
tokens = line.split()
if(tokens[0]!=current_region and len(current_bound)>0):
if tokens[0] != current_region and len(current_bound) > 0:
# insert the previous region after get all the vertex of the attatched geometry
current_bound.append(current_bound[0])
current_geometry = to_postgis_polygon(current_bound)
region_type = map_regiontype[region_list[tokens[0]]]
sql_batch.add(f"insert into region(id, boundary,r_type) values ('{current_region}', '{current_geometry}','{region_type}');")
sql_batch.add(
f"insert into region(id, boundary,r_type) values ('{current_region}', '{current_geometry}','{region_type}');"
)
# start the new region
current_bound.clear()
vertex_point = (float(tokens[1]), float(tokens[2]))
@@ -292,9 +306,16 @@ def parse_file(project: str, inp: str, version: str = '3') -> None:
current_region = tokens[0]
elif s == REGION_NODES:
tokens = line.split()
if(tokens[0]!=current_region and len(current_region_nodes)>0):
if (
tokens[0] != current_region
and len(current_region_nodes) > 0
):
# insert the previous region after get all the vertex of the attatched geometry
sql_batch.add(get_insert_into_region_sql(current_region,current_region_nodes))
sql_batch.add(
get_insert_into_region_sql(
current_region, current_region_nodes
)
)
# start the new region
current_region_nodes.clear()
current_region_nodes.append(tokens[1])
@@ -302,7 +323,9 @@ def parse_file(project: str, inp: str, version: str = '3') -> None:
if s == JUNCTIONS:
sql_batch.add(handler(line, demand_outside))
elif s == PATTERNS:
sql_batch.add(handler(line, current_pattern not in variable_patterns))
sql_batch.add(
handler(line, current_pattern not in variable_patterns)
)
elif s == BOUND or s == REGION_NODES:
continue
else:
@@ -320,38 +343,44 @@ def parse_file(project: str, inp: str, version: str = '3') -> None:
current_bound.append(current_bound[0])
current_geometry = to_postgis_polygon(current_bound)
region_type = map_regiontype[region_list[current_region]]
sql_batch.add(f"insert into region(id, boundary,r_type) values ('{current_region}', '{current_geometry}','{region_type}');")
sql_batch.add(
f"insert into region(id, boundary,r_type) values ('{current_region}', '{current_geometry}','{region_type}');"
)
# reset the current region to none for the [REGION_NODES] session reading
# current_region=None
# need to insert the last region_nodes into database
if len(current_region_nodes) > 0:
sql_batch.add(get_insert_into_region_sql(current_region,current_region_nodes))
sql_batch.add(
get_insert_into_region_sql(current_region, current_region_nodes)
)
# current_region=None
sql_batch.flush()
end = _print_time(f'End reading file "{inp}"')
print(f"Total (in second): {(end-start).seconds}(s)")
def get_insert_into_region_sql(region: str, nodes: list[str]) -> str:
str_sql=''
str_sql = ""
str_nodes = str(nodes).replace("'", "''")
r_type=region[0:region.index('_')]
if r_type == 'DMA' or r_type == 'SA' or r_type == 'VD':
table = ''
if r_type == 'DMA':
table = 'region_dma'
elif r_type == 'SA':
table = 'region_sa'
source=region[region.index('_')+1:]
r_type = region[0 : region.index("_")]
if r_type == "DMA" or r_type == "SA" or r_type == "VD":
table = ""
if r_type == "DMA":
table = "region_dma"
elif r_type == "SA":
table = "region_sa"
source = region[region.index("_") + 1 :]
str_sql = f"insert into region_sa(id,time_index,source,nodes) values ('{region}', 0,'{source}','{str_nodes}');"
elif r_type == 'VD':
table = 'region_vd'
elif r_type == "VD":
table = "region_vd"
return str_sql
def read_inp(project: str, inp: str, version: str = '3') -> bool:
if version != '3' and version != '2':
version = '2'
def read_inp(project: str, inp: str, version: str = "3") -> bool:
if version != "3" and version != "2":
version = "2"
if is_project_open(project):
close_project(project)
@@ -364,19 +393,20 @@ def read_inp(project: str, inp: str, version: str = '3') -> bool:
parse_file(project, inp, version)
'''try:
"""try:
parse_file(project, inp, version)
except:
close_project(project)
delete_project(project)
return False'''
return False"""
close_project(project)
return True
# DingZQ, 2024-12-28, convert v3 to v2
def convert_inp_v3_to_v2(inp: str) -> ChangeSet:
project = 'v3Tov2'
project = "v3Tov2"
if is_project_open(project):
close_project(project)
@@ -387,38 +417,39 @@ def convert_inp_v3_to_v2(inp: str) -> ChangeSet:
create_project(project)
open_project(project)
filename = f'inp/{project}_temp.inp'
filename = f"inp/{project}_temp.inp"
if os.path.exists(filename):
os.remove(filename)
with open(filename, 'w') as f:
with open(filename, "w") as f:
f.write(inp)
parse_file(project, filename, '3')
parse_file(project, filename, "3")
'''try:
"""try:
parse_file(project, inp, version)
except:
close_project(project)
delete_project(project)
return False'''
return False"""
return export_inp(project, '2')
return export_inp(project, "2")
def import_inp(project: str, cs: ChangeSet, version: str = '3') -> bool:
if version != '3' and version != '2':
version = '2'
if 'inp' not in cs.operations[0]:
def import_inp(project: str, cs: ChangeSet, version: str = "3") -> bool:
if version != "3" and version != "2":
version = "2"
if "inp" not in cs.operations[0]:
return False
filename = f'inp/{project}_temp.inp'
filename = f"inp/{project}_temp.inp"
if os.path.exists(filename):
os.remove(filename)
_print_time(f'Start writing temp file "{filename}"...')
with open(filename, 'w',encoding="GBK") as f:
f.write(str(cs.operations[0]['inp']))
with open(filename, "w", encoding="GBK") as f:
f.write(str(cs.operations[0]["inp"]))
_print_time(f'End writing temp file "{filename}"...')
result = read_inp(project, filename, version)
app/native/wndb/inp_out.py
+1 -1
View File
@@ -1,5 +1,5 @@
import os
from .project_backup import *
from .project import *
from .database import ChangeSet
from .sections import *
from .s1_title import inp_out_title
app/native/wndb/s34_sa_cal.py
+1 -1
View File
@@ -1,6 +1,6 @@
import os
import ctypes
from .project_backup import have_project
from .project import have_project
from .inp_out import dump_inp
def calculate_service_area(name: str) -> list[dict[str, list[str]]]: