Extend QList()

1 files changed, 19 insertions, 2 deletions

diff --git a/pycrctrl.py b/pycrctrl.py
index 6935512..c54d147 100644
--- a/pycrctrl.py
+++ b/pycrctrl.py
@@ -74,13 +74,27 @@ class QList(QObject):
         if type(name) != int:
             raise ValueError("Wrong datatype for name!")
         else:
-            return self.list[name]
+            try:
+                return self.list[name]
+            except IndexError as e:
+                raise IndexError(e.args[0]) from None
     
     def __lshift__(self, other):
         if self.typeclass and not isinstance(other, self.typeclass):
             raise ValueError("Wrong datatype")
         else:
             self.list.append(other)
+            return self
+
+    def __rshift__(self, other):
+        if isinstance(other, list) or isinstance(other, QList):
+            item = self.list[-1]
+            try:
+                other.append(item)
+                self.list.pop()
+            except Exception:
+                raise TypeError("Cannot pass item to {}!".format(other)) from None
+                
     
     def __call__(self):
         return self.list
@@ -89,13 +103,16 @@ class QList(QObject):
         return len(self.list)
     
     def __repr__(self):
-        return "{}({})".format(self.__name__, self.list)
+        return "{}({})".format(self.__class__.__name__, self.list)
     
     # Methods
     
     def isEmpty(self):
         return len(self.list) == 0
 
+    def append(self, item):
+        return self.__lshift__(item)
+
 class QStringList(QList):
     typeclass = str