Skip to content

Commit cfd23dc

Browse files
authored
Merge pull request #80 from karosc/dev
This PR does the following: - fixes call add_dll_directory() running in conda python - fixes nrtest-swmm import statements to use swmm-toolkit
2 parents 22daa1e + 779dff6 commit cfd23dc

File tree

2 files changed

+11
-8
lines changed

2 files changed

+11
-8
lines changed

nrtest-swmm/nrtest_swmm/output_reader.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from itertools import islice
1919

2020
# project import
21-
from swmm.toolkit import output, output_enum
21+
from swmm.toolkit import output, shared_enum
2222

2323

2424
def output_generator(path_ref):
@@ -40,7 +40,7 @@ def output_generator(path_ref):
4040
with OutputReader(path_ref) as reader:
4141

4242
for period_index in range(0, reader.report_periods()):
43-
for element_type in islice(output_enum.ElementType, 4):
43+
for element_type in islice(shared_enum.ElementType, 4):
4444
for element_index in range(0, reader.element_count(element_type)):
4545

4646
yield (reader.element_result(element_type, period_index, element_index),
@@ -57,10 +57,10 @@ def __init__(self, filename):
5757
self.handle = None
5858
self.count = None
5959
self.get_element_result = {
60-
output_enum.ElementType.SUBCATCH: output.get_subcatch_result,
61-
output_enum.ElementType.NODE: output.get_node_result,
62-
output_enum.ElementType.LINK: output.get_link_result,
63-
output_enum.ElementType.SYSTEM: output.get_system_result
60+
shared_enum.ElementType.SUBCATCH: output.get_subcatch_result,
61+
shared_enum.ElementType.NODE: output.get_node_result,
62+
shared_enum.ElementType.LINK: output.get_link_result,
63+
shared_enum.ElementType.SYSTEM: output.get_system_result
6464
}
6565

6666
def __enter__(self):
@@ -73,7 +73,7 @@ def __exit__(self, type, value, traceback):
7373
output.close(self.handle)
7474

7575
def report_periods(self):
76-
return output.get_times(self.handle, output_enum.Time.NUM_PERIODS)
76+
return output.get_times(self.handle, shared_enum.Time.NUM_PERIODS)
7777

7878
def element_count(self, element_type):
7979
return self.count[element_type]

swmm-toolkit/src/swmm/toolkit/__init__.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,16 @@
2929

3030
import os
3131
import platform
32+
import sys
3233

3334

3435
# Adds directory containing swmm libraries to path
3536
if platform.system() == "Windows":
3637
libdir = os.path.join(os.path.dirname(__file__), "../../swmm_toolkit")
3738

3839
if hasattr(os, 'add_dll_directory'):
39-
os.add_dll_directory(libdir)
40+
if 'conda' in sys.version:
41+
os.environ['CONDA_DLL_SEARCH_MODIFICATION_ENABLE']="1"
42+
os.add_dll_directory(libdir)
4043
else:
4144
os.environ["PATH"] = libdir + ";" + os.environ["PATH"]

0 commit comments

Comments
 (0)