Merge pull request #13 from patterninc/feat/add-is-prod-tag

perimeter tag correct and add "is_production" tag

Author: phitoduck

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ds-platform-utils"
-version = "0.2.2"
+version = "0.2.3"
 description = "Utility library for Pattern Data Science."
 readme = "README.md"
 authors = [

src/ds_platform_utils/metaflow/write_audit_publish.py

@@ -1,4 +1,5 @@
 import json
+import os
 import warnings
 from pathlib import Path
 from textwrap import dedent
@@ -119,7 +120,10 @@ def extract(prefix: str, default: str = "unknown") -> str:
         "user": current.username,  # username of user who triggered the run (argo-workflows if its a deployed flow)
         "domain": extract("ds.domain"),  # business unit (domain) of the flow, same as app
         "namespace": current.namespace,  # namespace of the flow
-        "perimeter": "PROD" if current.is_production else "Default",  # perimeter of the flow
+        "perimeter": str(os.environ.get("OB_CURRENT_PERIMETER") or os.environ.get("OBP_PERIMETER")),
+        "is_production": str(
+            current.is_production
+        ),  # True, if the flow is deployed with the --production flag else false
         "team": "data-science",  # team name, hardcoded as data-science
     }