Update query.py

oribarilan · Jan 22, 2024 · c87a931 · c87a931
1 parent ff54a92
commit c87a931
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/fliq/query.py b/fliq/query.py
@@ -1013,6 +1013,15 @@ def at(self,
             ElementNotFoundException: In case the query is too short.
             QueryIsUnexpectedlyEmptyException: In case the query is empty.
         """
+        # if items can be accessed using index, do that
+        if isinstance(self._items, collections.abc.Sequence):
+            try:
+                return self._items[index]  # type: ignore # (item is of type T)
+            except IndexError:
+                if default is MISSING:
+                    raise ElementNotFoundException()
+                else:
+                    return default  # type: ignore # default is not MISSING
         try:
             return self.skip(index).first(default=default)
         except QueryIsUnexpectedlyEmptyException: