From 07c7c8906c6f9ab6870c68fffdf588f500f50f3e Mon Sep 17 00:00:00 2001 From: clavedeluna Date: Mon, 19 Aug 2024 11:02:38 -0300 Subject: [PATCH 1/2] update pygoat findings --- ci_tests/test_pygoat_findings.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ci_tests/test_pygoat_findings.py b/ci_tests/test_pygoat_findings.py index 7602758f..ea3c6dfb 100644 --- a/ci_tests/test_pygoat_findings.py +++ b/ci_tests/test_pygoat_findings.py @@ -14,6 +14,7 @@ "pixee:python/url-sandbox", "pixee:python/use-defusedxml", "pixee:python/use-walrus-if", + "pixee:python/timezone-aware-datetime", ] From fc2a63457ebd671e93f89de27b4632640bb0a76e Mon Sep 17 00:00:00 2001 From: "pixeebot[bot]" <104101892+pixeebot[bot]@users.noreply.github.com> Date: Fri, 30 Aug 2024 17:30:45 +0000 Subject: [PATCH 2/2] Use Assignment Expression (Walrus) In Conditional --- src/core_codemods/timezone_aware_datetime.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/core_codemods/timezone_aware_datetime.py b/src/core_codemods/timezone_aware_datetime.py index d3d84592..bca4103a 100644 --- a/src/core_codemods/timezone_aware_datetime.py +++ b/src/core_codemods/timezone_aware_datetime.py @@ -74,9 +74,8 @@ def leave_Call(self, original_node: cst.Call, updated_node: cst.Call): return updated_node def _determine_module_and_kwarg(self, original_node: cst.Call): - maybe_name = self.get_aliased_prefix_name(original_node, self._module_name) - if maybe_name: + if maybe_name := self.get_aliased_prefix_name(original_node, self._module_name): # it's a regular import OR alias import if maybe_name == self._module_name: module = "datetime.datetime"