From 570a446d16c56f591765029229f0371ff398c1aa Mon Sep 17 00:00:00 2001 From: Arjun Anandkumar Date: Wed, 4 Oct 2023 19:35:27 +0200 Subject: [PATCH] Exclude venv --- airflow/providers/microsoft/psrp/hooks/psrp.py | 11 ++++++++++- airflow/providers/microsoft/psrp/provider.yaml | 4 ++++ 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/airflow/providers/microsoft/psrp/hooks/psrp.py b/airflow/providers/microsoft/psrp/hooks/psrp.py index d77e73c278bdb..3408ac07d2b3b 100644 --- a/airflow/providers/microsoft/psrp/hooks/psrp.py +++ b/airflow/providers/microsoft/psrp/hooks/psrp.py @@ -74,10 +74,14 @@ class PsrpHook(BaseHook): You can provide an alternative `configuration_name` using either `runspace_options` or by setting this key as the extra fields of your connection. """ + conn_name_attr = "psrp_conn_id" + default_conn_name = "psrp_default" + conn_type = "psrp" + hook_name = "PSRP" _conn: RunspacePool | None = None _wsman_ref: WeakKeyDictionary[RunspacePool, WSMan] = WeakKeyDictionary() - + def __init__( self, psrp_conn_id: str, @@ -281,3 +285,8 @@ def _log_record(self, log, record): log(INFO, "Progress: %s (%s)", record.activity, record.description) else: log(WARNING, "Unsupported message type: %s", message_type) + + def test_connection(self): + """Test PSRP Connection.""" + # To be implemented. + pass diff --git a/airflow/providers/microsoft/psrp/provider.yaml b/airflow/providers/microsoft/psrp/provider.yaml index cd9c953a7d287..710fc97ba248c 100644 --- a/airflow/providers/microsoft/psrp/provider.yaml +++ b/airflow/providers/microsoft/psrp/provider.yaml @@ -57,3 +57,7 @@ hooks: - integration-name: Windows PowerShell Remoting Protocol python-modules: - airflow.providers.microsoft.psrp.hooks.psrp + +connection-types: + - hook-class-name: airflow.providers.microsoft.psrp.hooks.PsrpHook + connection-type: psrp \ No newline at end of file