Skip to content

Commit fb3e32d

Browse files
authored
Merge pull request #1786 from elementary-data/ele-3971-allow-exposure-logic-overwrite
Extract exposure logic
2 parents 9657b63 + b2892dd commit fb3e32d

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

elementary/monitor/api/report/report.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,11 @@ def _get_groups(
6363
]
6464
)
6565

66+
def _get_exposures(
67+
self, models_api: ModelsAPI, upstream_node_ids: Optional[List[str]] = None
68+
) -> Dict[str, NormalizedExposureSchema]:
69+
return models_api.get_exposures(upstream_node_ids=upstream_node_ids)
70+
6671
def get_report_data(
6772
self,
6873
days_back: int = 7,
@@ -101,7 +106,9 @@ def get_report_data(
101106
lineage_node_ids.extend(models.keys())
102107
sources = models_api.get_sources()
103108
lineage_node_ids.extend(sources.keys())
104-
exposures = models_api.get_exposures(upstream_node_ids=lineage_node_ids)
109+
exposures = self._get_exposures(
110+
models_api, upstream_node_ids=lineage_node_ids
111+
)
105112
lineage_node_ids.extend(exposures.keys())
106113
singular_tests = tests_api.get_singular_tests()
107114

0 commit comments

Comments
 (0)