acryl-datahub-dagster-plugin 1.3.1.7rc1__py3-none-any.whl → 1.3.1.8rc1__py3-none-any.whl

{acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info → acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acryl-datahub-dagster-plugin
-Version: 1.3.1.7rc1
+Version: 1.3.1.8rc1
 Summary: Datahub Dagster plugin to capture executions and send to Datahub
 Home-page: https://docs.datahub.com/
 License: Apache-2.0
@@ -19,82 +19,82 @@ Classifier: Operating System :: POSIX :: Linux
 Classifier: Environment :: Console
 Classifier: Environment :: MacOS X
 Classifier: Topic :: Software Development
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: dagit>=1.10.0
+Requires-Dist: acryl-datahub[datahub-rest,sql-parser]==1.3.1.8rc1
 Requires-Dist: dagster>=1.10.0
-Requires-Dist: acryl-datahub[datahub-rest,sql-parser]==1.3.1.7rc1
 Provides-Extra: ignore
 Provides-Extra: dev
-Requires-Dist: types-six; extra == "dev"
-Requires-Dist: coverage>=5.1; extra == "dev"
+Requires-Dist: mypy==1.17.1; extra == "dev"
+Requires-Dist: types-python-dateutil; extra == "dev"
+Requires-Dist: dagster-snowflake-pandas>=0.11.0; extra == "dev"
+Requires-Dist: packaging; extra == "dev"
+Requires-Dist: types-PyYAML; extra == "dev"
+Requires-Dist: types-tabulate; extra == "dev"
+Requires-Dist: dagster-snowflake>=0.11.0; extra == "dev"
 Requires-Dist: build; extra == "dev"
-Requires-Dist: sqlalchemy-stubs; extra == "dev"
-Requires-Dist: ruff==0.11.7; extra == "dev"
-Requires-Dist: types-setuptools; extra == "dev"
+Requires-Dist: requests-mock; extra == "dev"
 Requires-Dist: types-requests; extra == "dev"
-Requires-Dist: dagit>=1.10.0; extra == "dev"
-Requires-Dist: tox; extra == "dev"
-Requires-Dist: types-pytz; extra == "dev"
-Requires-Dist: types-python-dateutil; extra == "dev"
+Requires-Dist: sqlalchemy-stubs; extra == "dev"
 Requires-Dist: dagster>=1.10.0; extra == "dev"
 Requires-Dist: twine; extra == "dev"
-Requires-Dist: types-freezegun; extra == "dev"
-Requires-Dist: requests-mock; extra == "dev"
-Requires-Dist: dagster-snowflake>=0.11.0; extra == "dev"
-Requires-Dist: types-toml; extra == "dev"
 Requires-Dist: pytest>=6.2.2; extra == "dev"
+Requires-Dist: acryl-datahub[datahub-rest,sql-parser]==1.3.1.8rc1; extra == "dev"
 Requires-Dist: freezegun; extra == "dev"
-Requires-Dist: pytest-cov>=2.8.1; extra == "dev"
-Requires-Dist: types-PyYAML; extra == "dev"
-Requires-Dist: mypy==1.17.1; extra == "dev"
 Requires-Dist: types-cachetools; extra == "dev"
-Requires-Dist: jsonpickle; extra == "dev"
-Requires-Dist: types-tabulate; extra == "dev"
-Requires-Dist: acryl-datahub[datahub-rest,sql-parser]==1.3.1.7rc1; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.16.0; extra == "dev"
 Requires-Dist: types-dataclasses; extra == "dev"
+Requires-Dist: types-pytz; extra == "dev"
+Requires-Dist: types-freezegun; extra == "dev"
+Requires-Dist: dagit>=1.10.0; extra == "dev"
+Requires-Dist: pytest-cov>=2.8.1; extra == "dev"
+Requires-Dist: tox; extra == "dev"
+Requires-Dist: types-click==0.1.12; extra == "dev"
+Requires-Dist: types-toml; extra == "dev"
 Requires-Dist: dagster-aws>=0.11.0; extra == "dev"
+Requires-Dist: coverage>=5.1; extra == "dev"
+Requires-Dist: types-setuptools; extra == "dev"
 Requires-Dist: pydantic!=1.10.3,>=1.10.0; extra == "dev"
-Requires-Dist: dagster-snowflake-pandas>=0.11.0; extra == "dev"
-Requires-Dist: types-click==0.1.12; extra == "dev"
+Requires-Dist: jsonpickle; extra == "dev"
 Requires-Dist: deepdiff!=8.0.0; extra == "dev"
-Requires-Dist: packaging; extra == "dev"
-Requires-Dist: pytest-asyncio>=0.16.0; extra == "dev"
+Requires-Dist: types-six; extra == "dev"
+Requires-Dist: ruff==0.11.7; extra == "dev"
 Provides-Extra: integration-tests
-Requires-Dist: types-six; extra == "integration-tests"
-Requires-Dist: coverage>=5.1; extra == "integration-tests"
+Requires-Dist: mypy==1.17.1; extra == "integration-tests"
+Requires-Dist: types-python-dateutil; extra == "integration-tests"
+Requires-Dist: dagster-snowflake-pandas>=0.11.0; extra == "integration-tests"
+Requires-Dist: packaging; extra == "integration-tests"
+Requires-Dist: types-PyYAML; extra == "integration-tests"
+Requires-Dist: types-tabulate; extra == "integration-tests"
+Requires-Dist: dagster-snowflake>=0.11.0; extra == "integration-tests"
 Requires-Dist: build; extra == "integration-tests"
-Requires-Dist: sqlalchemy-stubs; extra == "integration-tests"
-Requires-Dist: ruff==0.11.7; extra == "integration-tests"
-Requires-Dist: types-setuptools; extra == "integration-tests"
+Requires-Dist: requests-mock; extra == "integration-tests"
 Requires-Dist: types-requests; extra == "integration-tests"
-Requires-Dist: dagit>=1.10.0; extra == "integration-tests"
-Requires-Dist: tox; extra == "integration-tests"
-Requires-Dist: types-pytz; extra == "integration-tests"
-Requires-Dist: types-python-dateutil; extra == "integration-tests"
+Requires-Dist: sqlalchemy-stubs; extra == "integration-tests"
 Requires-Dist: dagster>=1.10.0; extra == "integration-tests"
 Requires-Dist: twine; extra == "integration-tests"
-Requires-Dist: types-freezegun; extra == "integration-tests"
-Requires-Dist: requests-mock; extra == "integration-tests"
-Requires-Dist: dagster-snowflake>=0.11.0; extra == "integration-tests"
-Requires-Dist: types-toml; extra == "integration-tests"
 Requires-Dist: pytest>=6.2.2; extra == "integration-tests"
+Requires-Dist: acryl-datahub[datahub-rest,sql-parser]==1.3.1.8rc1; extra == "integration-tests"
 Requires-Dist: freezegun; extra == "integration-tests"
-Requires-Dist: pytest-cov>=2.8.1; extra == "integration-tests"
-Requires-Dist: types-PyYAML; extra == "integration-tests"
-Requires-Dist: mypy==1.17.1; extra == "integration-tests"
 Requires-Dist: types-cachetools; extra == "integration-tests"
-Requires-Dist: jsonpickle; extra == "integration-tests"
-Requires-Dist: types-tabulate; extra == "integration-tests"
-Requires-Dist: acryl-datahub[datahub-rest,sql-parser]==1.3.1.7rc1; extra == "integration-tests"
+Requires-Dist: pytest-asyncio>=0.16.0; extra == "integration-tests"
 Requires-Dist: types-dataclasses; extra == "integration-tests"
+Requires-Dist: types-pytz; extra == "integration-tests"
+Requires-Dist: types-freezegun; extra == "integration-tests"
+Requires-Dist: dagit>=1.10.0; extra == "integration-tests"
+Requires-Dist: pytest-cov>=2.8.1; extra == "integration-tests"
+Requires-Dist: tox; extra == "integration-tests"
+Requires-Dist: types-click==0.1.12; extra == "integration-tests"
+Requires-Dist: types-toml; extra == "integration-tests"
 Requires-Dist: dagster-aws>=0.11.0; extra == "integration-tests"
+Requires-Dist: coverage>=5.1; extra == "integration-tests"
+Requires-Dist: types-setuptools; extra == "integration-tests"
 Requires-Dist: pydantic!=1.10.3,>=1.10.0; extra == "integration-tests"
-Requires-Dist: dagster-snowflake-pandas>=0.11.0; extra == "integration-tests"
-Requires-Dist: types-click==0.1.12; extra == "integration-tests"
+Requires-Dist: jsonpickle; extra == "integration-tests"
 Requires-Dist: deepdiff!=8.0.0; extra == "integration-tests"
-Requires-Dist: packaging; extra == "integration-tests"
-Requires-Dist: pytest-asyncio>=0.16.0; extra == "integration-tests"
+Requires-Dist: types-six; extra == "integration-tests"
+Requires-Dist: ruff==0.11.7; extra == "integration-tests"
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type

{acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info → acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info}/RECORD

@@ -1,17 +1,17 @@
 datahub_dagster_plugin/__init__.py,sha256=v460GAGTiTOr3Up8iep5w1DSNe2tWI-N3pf_dQiCR4I,74
-datahub_dagster_plugin/_version.py,sha256=DUGS-14-5lhjkgYXZpd9_l-qwMVxmb5AJUszUCLJ4SU,148
+datahub_dagster_plugin/_version.py,sha256=inK60CmqWkiP0Is5uM6q1TB6NbdluR7sUowYOYroKXQ,148
 datahub_dagster_plugin/datahub_dagster_plugin.py,sha256=szsT7ddxs16eE56mhCdNXDtlakbNLmuhd2wUbkba5Xc,64
 datahub_dagster_plugin/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datahub_dagster_plugin/client/dagster_generator.py,sha256=pZygCcJ_2OlmPCHFvxdnKuOxZbZSXUAIdGQqaXsz0mc,35214
+datahub_dagster_plugin/client/dagster_generator.py,sha256=fGx76tZR1YgU4gno7lisIdZS0ygqsWMW3KoCvIEaAeQ,35268
 datahub_dagster_plugin/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datahub_dagster_plugin/modules/snowflake_pandas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datahub_dagster_plugin/modules/snowflake_pandas/datahub_snowflake_pandas_io_manager.py,sha256=N280oLxNKyunZlnuc8_W9JLQW9anO1NURBSPF9Qfo1E,1229
 datahub_dagster_plugin/modules/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datahub_dagster_plugin/modules/storage/datahub_db_io_manager.py,sha256=jxzqYRM0rOL2BVtx1BjHViN22ertwDtLmaKUiuTBPSY,2592
 datahub_dagster_plugin/sensors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datahub_dagster_plugin/sensors/datahub_sensors.py,sha256=8HLFAJiqD5hj3074oz53_mUd_Hhji9nuW5G9j9OsLsk,32965
-acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info/METADATA,sha256=S-OMU0C-2JfpfnZ_uvgr_6JgaNzV_ZjW4BdVx3Kwbbc,5271
-acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info/entry_points.txt,sha256=-CtPxtYb1u-zR36QnUQrvJJ6qbf1eDw9SruA22XBPZw,116
-acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info/top_level.txt,sha256=JS8QiBAj3eMTcMNcstm_EXGAcziiXVNT2nzOcfhdEMc,23
-acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info/RECORD,,
+datahub_dagster_plugin/sensors/datahub_sensors.py,sha256=xJiIGMx8JRfenEnnh9eLd0vFUFSC8qLSvJo9mVs64nY,33193
+acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info/METADATA,sha256=vhTD_DQqj8uNd8vAZSDPpLwPukoPIB0BR5WcCEQ0btE,5272
+acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info/entry_points.txt,sha256=-CtPxtYb1u-zR36QnUQrvJJ6qbf1eDw9SruA22XBPZw,116
+acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info/top_level.txt,sha256=JS8QiBAj3eMTcMNcstm_EXGAcziiXVNT2nzOcfhdEMc,23
+acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info/RECORD,,

{acryl_datahub_dagster_plugin-1.3.1.7rc1.dist-info → acryl_datahub_dagster_plugin-1.3.1.8rc1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

datahub_dagster_plugin/_version.py

@@ -1,3 +1,3 @@
 # Published at https://pypi.org/project/acryl-datahub-dagster-plugin/.
 __package_name__ = "acryl-datahub-dagster-plugin"
-__version__ = "1.3.1.7rc1"
+__version__ = "1.3.1.8rc1"

datahub_dagster_plugin/client/dagster_generator.py

@@ -253,6 +253,7 @@ class DagsterEnvironment:
     is_branch_deployment: bool = False
     branch: Optional[str] = "prod"
     module: Optional[str] = None
+    location_name: Optional[str] = None
 
 
 def job_url_generator(dagster_url: str, dagster_environment: DagsterEnvironment) -> str:
@@ -261,8 +262,8 @@ def job_url_generator(dagster_url: str, dagster_environment: DagsterEnvironment)
     else:
         base_url = dagster_url
 
-    if dagster_environment.module:
-        base_url = f"{base_url}/locations/{dagster_environment.module}"
+    if dagster_environment.location_name:
+        base_url = f"{base_url}/locations/{dagster_environment.location_name}"
 
     return base_url
 

datahub_dagster_plugin/sensors/datahub_sensors.py

@@ -259,6 +259,7 @@ class DatahubSensors:
     ) -> Optional[DagsterEnvironment]:
         module: Optional[str] = None
         repository: Optional[str] = None
+        location_name: Optional[str] = None
         if (
             context
             and context.dagster_run.job_code_origin
@@ -280,6 +281,9 @@ class DatahubSensors:
             else:
                 context.log.error("Unable to get Module")
 
+        if context and context.dagster_run.remote_job_origin:
+            location_name = context.dagster_run.remote_job_origin.location_name
+
         dagster_environment = DagsterEnvironment(
             is_cloud=os.getenv("DAGSTER_CLOUD_IS_BRANCH_DEPLOYMENT", None) is not None,
             is_branch_deployment=(
@@ -290,6 +294,7 @@ class DatahubSensors:
             branch=os.getenv("DAGSTER_CLOUD_DEPLOYMENT_NAME", "prod"),
             module=module,
             repository=repository,
+            location_name=location_name,
         )
         return dagster_environment