From 70ea4b25e8d781e27fd16c8f52f732240f26fa2b Mon Sep 17 00:00:00 2001
From: Charlie Marsh <charlie.r.marsh@gmail.com>
Date: Tue, 17 Jan 2023 11:14:11 -0500
Subject: [PATCH] Allow duplicate enum values for enum.auto() (#1933)

Closes #1932.
---
 resources/test/fixtures/flake8_pie/PIE796.py | 8 +++++++-
 src/rules/flake8_pie/rules.rs                | 9 +++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/resources/test/fixtures/flake8_pie/PIE796.py b/resources/test/fixtures/flake8_pie/PIE796.py
index e8ea7c3724..227ae8acc2 100644
--- a/resources/test/fixtures/flake8_pie/PIE796.py
+++ b/resources/test/fixtures/flake8_pie/PIE796.py
@@ -2,7 +2,7 @@ import enum
 from enum import Enum, unique
 
 
-class FakeEnum(enum.Enum):
+class FakeEnum1(enum.Enum):
     A = "A"
     B = "B"
     C = "B"  # PIE796
@@ -58,3 +58,9 @@ class FakeEnum9(enum.Enum):
     A = "A"
     B = "B"
     C = "C"
+
+
+class FakeEnum10(enum.Enum):
+    A = enum.auto()
+    B = enum.auto()
+    C = enum.auto()
diff --git a/src/rules/flake8_pie/rules.rs b/src/rules/flake8_pie/rules.rs
index cbe2a5f368..4def9afc38 100644
--- a/src/rules/flake8_pie/rules.rs
+++ b/src/rules/flake8_pie/rules.rs
@@ -131,6 +131,15 @@ where
             continue;
         };
 
+        if let ExprKind::Call { func, .. } = &value.node {
+            if checker
+                .resolve_call_path(func)
+                .map_or(false, |call_path| call_path == ["enum", "auto"])
+            {
+                continue;
+            }
+        }
+
         if !seen_targets.insert(ComparableExpr::from(value)) {
             let diagnostic = Diagnostic::new(
                 violations::PreferUniqueEnums {