From f2a6c09007d40f3058eefe16987749656a0c352d Mon Sep 17 00:00:00 2001
From: ShaharNaveh <50263213+ShaharNaveh@users.noreply.github.com>
Date: Fri, 20 Jun 2025 16:26:11 +0300
Subject: [PATCH] CLN: Loop with idx check location_tup len fields

---
 vm/src/exceptions.rs | 35 +++++++++++++++++------------------
 1 file changed, 17 insertions(+), 18 deletions(-)

diff --git a/vm/src/exceptions.rs b/vm/src/exceptions.rs
index 3c822c430..2f73ea5ae 100644
--- a/vm/src/exceptions.rs
+++ b/vm/src/exceptions.rs
@@ -1619,24 +1619,23 @@ pub(super) mod types {
                     .clone()
                     .downcast::<crate::builtins::PyTuple>()
                 {
-                    #[allow(clippy::len_zero)]
-                    if location_tuple.len() >= 1 {
-                        zelf.set_attr("filename", location_tuple.fast_getitem(0).clone(), vm)?;
-                    }
-                    if location_tuple.len() >= 2 {
-                        zelf.set_attr("lineno", location_tuple.fast_getitem(1).clone(), vm)?;
-                    }
-                    if location_tuple.len() >= 3 {
-                        zelf.set_attr("offset", location_tuple.fast_getitem(2).clone(), vm)?;
-                    }
-                    if location_tuple.len() >= 4 {
-                        zelf.set_attr("text", location_tuple.fast_getitem(3).clone(), vm)?;
-                    }
-                    if location_tuple.len() >= 5 {
-                        zelf.set_attr("end_lineno", location_tuple.fast_getitem(4).clone(), vm)?;
-                    }
-                    if location_tuple.len() >= 6 {
-                        zelf.set_attr("end_offset", location_tuple.fast_getitem(5).clone(), vm)?;
+                    let location_tup_len = location_tuple.len();
+                    for (i, &attr) in [
+                        "filename",
+                        "lineno",
+                        "offset",
+                        "text",
+                        "end_lineno",
+                        "end_offset",
+                    ]
+                    .iter()
+                    .enumerate()
+                    {
+                        if location_tup_len > i {
+                            zelf.set_attr(attr, location_tuple.fast_getitem(i).clone(), vm)?;
+                        } else {
+                            break;
+                        }
                     }
                 }
             }