This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 7e6b7e1667 Fix D105 for mongo provider (#38016)
7e6b7e1667 is described below
commit 7e6b7e1667e0c5311039e8bda8e6cb3e3de3c529
Author: Poorvi Rohidekar <[email protected]>
AuthorDate: Sun Mar 10 20:08:29 2024 +0530
Fix D105 for mongo provider (#38016)
---
airflow/providers/mongo/hooks/mongo.py | 2 ++
pyproject.toml | 1 -
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/airflow/providers/mongo/hooks/mongo.py
b/airflow/providers/mongo/hooks/mongo.py
index 7a968ca9b2..95d141aaec 100644
--- a/airflow/providers/mongo/hooks/mongo.py
+++ b/airflow/providers/mongo/hooks/mongo.py
@@ -118,6 +118,7 @@ class MongoHook(BaseHook):
self.allow_insecure = False
def __enter__(self):
+ """Return the object when a context manager is created."""
return self
def __exit__(
@@ -126,6 +127,7 @@ class MongoHook(BaseHook):
exc_val: BaseException | None,
exc_tb: TracebackType | None,
) -> None:
+ """Close mongo connection when exiting the context manager."""
if self.client is not None:
self.client.close()
self.client = None
diff --git a/pyproject.toml b/pyproject.toml
index e3173e256e..eb83fedb50 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1468,7 +1468,6 @@ combine-as-imports = true
"airflow/providers/google/cloud/links/dataproc.py" = ["D105"]
"airflow/providers/imap/hooks/imap.py" = ["D105"]
"airflow/providers/microsoft/psrp/hooks/psrp.py" = ["D105"]
-"airflow/providers/mongo/hooks/mongo.py" = ["D105"]
"airflow/providers/samba/hooks/samba.py" = ["D105"]
"airflow/providers/smtp/hooks/smtp.py" = ["D105"]
"airflow/providers/ssh/hooks/ssh.py" = ["D105"]
