Skip to content

Commit 97ea854

Browse files
committed
feat: add field parameters transfer and associated metrics handling
1 parent 988cb63 commit 97ea854

2 files changed

Lines changed: 22 additions & 1 deletion

File tree

transfers/metrics.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@
3636
Asset,
3737
PermissionHistory,
3838
ThingGeologicFormationAssociation,
39+
Stratigraphy,
40+
FieldParameters,
3941
ChemistrySampleInfo,
4042
NMAHydraulicsData,
4143
NMARadionuclides,
@@ -168,6 +170,12 @@ def permissions_metrics(self, *args, **kw) -> None:
168170
def stratigraphy_metrics(self, *args, **kw) -> None:
169171
self._handle_metrics(ThingGeologicFormationAssociation, *args, **kw)
170172

173+
def nma_stratigraphy_metrics(self, *args, **kw) -> None:
174+
self._handle_metrics(Stratigraphy, name="NMA_Stratigraphy", *args, **kw)
175+
176+
def field_parameters_metrics(self, *args, **kw) -> None:
177+
self._handle_metrics(FieldParameters, name="FieldParameters", *args, **kw)
178+
171179
def associated_data_metrics(self, *args, **kw) -> None:
172180
self._handle_metrics(AssociatedData, name="AssociatedData", *args, **kw)
173181

transfers/transfer.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@
6666

6767
from transfers.asset_transfer import AssetTransferer
6868
from transfers.chemistry_sampleinfo import ChemistrySampleInfoTransferer
69+
from transfers.field_parameters_transfer import FieldParametersTransferer
6970
from transfers.hydraulicsdata import HydraulicsDataTransferer
7071
from transfers.radionuclides import RadionuclidesTransferer
7172
from transfers.major_chemistry import MajorChemistryTransferer
@@ -104,6 +105,7 @@ class TransferOptions:
104105
transfer_surface_water_data: bool
105106
transfer_hydraulics_data: bool
106107
transfer_chemistry_sampleinfo: bool
108+
transfer_field_parameters: bool
107109
transfer_major_chemistry: bool
108110
transfer_radionuclides: bool
109111
transfer_ngwmn_views: bool
@@ -137,6 +139,7 @@ def load_transfer_options() -> TransferOptions:
137139
transfer_chemistry_sampleinfo=get_bool_env(
138140
"TRANSFER_CHEMISTRY_SAMPLEINFO", True
139141
),
142+
transfer_field_parameters=get_bool_env("TRANSFER_FIELD_PARAMETERS", True),
140143
transfer_major_chemistry=get_bool_env("TRANSFER_MAJOR_CHEMISTRY", True),
141144
transfer_radionuclides=get_bool_env("TRANSFER_RADIONUCLIDES", True),
142145
transfer_ngwmn_views=get_bool_env("TRANSFER_NGWMN_VIEWS", True),
@@ -420,7 +423,7 @@ def _transfer_parallel(
420423
if opts.transfer_nma_stratigraphy:
421424
future = executor.submit(
422425
_execute_transfer_with_timing,
423-
"Stratigraphy",
426+
"StratigraphyLegacy",
424427
StratigraphyLegacyTransferer,
425428
flags,
426429
)
@@ -510,6 +513,11 @@ def _transfer_parallel(
510513
results = _execute_transfer(MinorTraceChemistryTransferer, flags=flags)
511514
metrics.minor_trace_chemistry_metrics(*results)
512515

516+
if opts.transfer_field_parameters:
517+
message("TRANSFERRING FIELD PARAMETERS")
518+
results = _execute_transfer(FieldParametersTransferer, flags=flags)
519+
metrics.field_parameters_metrics(*results)
520+
513521
# =========================================================================
514522
# PHASE 3: Sensors (Sequential - required before continuous water levels)
515523
# =========================================================================
@@ -662,6 +670,11 @@ def _transfer_sequential(
662670
results = _execute_transfer(ChemistrySampleInfoTransferer, flags=flags)
663671
metrics.chemistry_sampleinfo_metrics(*results)
664672

673+
if opts.transfer_field_parameters:
674+
message("TRANSFERRING FIELD PARAMETERS")
675+
results = _execute_transfer(FieldParametersTransferer, flags=flags)
676+
metrics.field_parameters_metrics(*results)
677+
665678
if opts.transfer_major_chemistry:
666679
message("TRANSFERRING MAJOR CHEMISTRY")
667680
results = _execute_transfer(MajorChemistryTransferer, flags=flags)

0 commit comments

Comments
 (0)