From 4e2f6f2685c1b6c4dd96789574b997e535dea008 Mon Sep 17 00:00:00 2001
From: Petr <petr.golubev@jetbrains.com>
Date: Thu, 28 Mar 2024 19:00:28 +0100
Subject: [PATCH] PY-28076 Type of assignment target in parentheses is not
 inferred

GitOrigin-RevId: 457adb65fea311a169ae8fd4ca78a6915518b378
---
 .../jetbrains/python/psi/impl/PyTargetExpressionImpl.java    | 2 +-
 ...notationLocalSimpleAssignmentTargetInParentheses_after.py | 2 +-
 ...eCommentLocalSimpleAssignmentTargetInParentheses_after.py | 2 +-
 python/testSrc/com/jetbrains/python/PyTypeTest.java          | 5 +++++
 4 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyTargetExpressionImpl.java b/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyTargetExpressionImpl.java
index c8b46df01e55..976b1e4c97bd 100644
--- a/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyTargetExpressionImpl.java
+++ b/python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyTargetExpressionImpl.java
@@ -121,7 +121,7 @@ public class PyTargetExpressionImpl extends PyBaseElementImpl<PyTargetExpression
     if (type != null) {
       return type;
     }
-    final PsiElement parent = getParent();
+    final PsiElement parent = PsiTreeUtil.skipParentsOfType(this, PyParenthesizedExpression.class);
     if (parent instanceof PyAssignmentStatement assignmentStatement) {
       PyExpression assignedValue = assignmentStatement.getAssignedValue();
       if (assignedValue instanceof PyParenthesizedExpression) {
diff --git a/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/annotationLocalSimpleAssignmentTargetInParentheses_after.py b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/annotationLocalSimpleAssignmentTargetInParentheses_after.py
index d1af1053d3d4..e21bbfbe07fb 100644
--- a/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/annotationLocalSimpleAssignmentTargetInParentheses_after.py
+++ b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/annotationLocalSimpleAssignmentTargetInParentheses_after.py
@@ -1,4 +1,4 @@
 def func():
-    var: [object]
+    var: [str]
     (var) = 'spam'
     var
diff --git a/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentLocalSimpleAssignmentTargetInParentheses_after.py b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentLocalSimpleAssignmentTargetInParentheses_after.py
index 69ec8af6b458..f40ee03ce7cb 100644
--- a/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentLocalSimpleAssignmentTargetInParentheses_after.py
+++ b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentLocalSimpleAssignmentTargetInParentheses_after.py
@@ -1,3 +1,3 @@
 def func():
-    (var) = 'spam'  # type: [object]
+    (var) = 'spam'  # type: [str]
     var
diff --git a/python/testSrc/com/jetbrains/python/PyTypeTest.java b/python/testSrc/com/jetbrains/python/PyTypeTest.java
index e7dd1ac2a379..d4b072c598a9 100644
--- a/python/testSrc/com/jetbrains/python/PyTypeTest.java
+++ b/python/testSrc/com/jetbrains/python/PyTypeTest.java
@@ -4566,6 +4566,11 @@ public class PyTypeTest extends PyTestCase {
              expr = c.get()""");
   }
 
+  // PY-28076
+  public void testAssignmentParens() {
+    doTest("int", "((expr)) = 42");
+  }
+
   private static List<TypeEvalContext> getTypeEvalContexts(@NotNull PyExpression element) {
     return ImmutableList.of(TypeEvalContext.codeAnalysis(element.getProject(), element.getContainingFile()).withTracing(),
                             TypeEvalContext.userInitiated(element.getProject(), element.getContainingFile()).withTracing());