From da35e0f027b357fa0b69bf77eb7334e79ad6cde8 Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Tue, 9 Feb 2016 03:44:50 -0500
Subject: [PATCH] better slicing (and dicing!)

---
 records.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/records.py b/records.py
index 39c1515..ee25319 100644
--- a/records.py
+++ b/records.py
@@ -50,7 +50,10 @@ class ResultSet(object):
 
     def __getitem__(self, key):
         # Convert ResultSet[1] into slice.
+        is_int = False
+
         if isinstance(key, int):
+            is_int = True
             key = slice(key, key + 1, None)
 
         while len(self._all_rows) < key.stop:
@@ -60,7 +63,7 @@ class ResultSet(object):
                 break
 
         item = self._all_rows[key]
-        item = item[0] if len(item) == 1 else item
+        item = item[0] if is_int else item
 
         return item