From ab46727c569203a8a583e09be83627ec36b7a0a5 Mon Sep 17 00:00:00 2001
From: Doug Blank <dblank@cs.brynmawr.edu>
Date: Thu, 19 Dec 2013 22:52:04 -0500
Subject: [PATCH] Changes in Struct to allow SQL-like query interface

---
 gramps/gen/merge/diff.py | 62 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 59 insertions(+), 3 deletions(-)

diff --git a/gramps/gen/merge/diff.py b/gramps/gen/merge/diff.py
index 26980e793..3147b26dd 100644
--- a/gramps/gen/merge/diff.py
+++ b/gramps/gen/merge/diff.py
@@ -159,8 +159,9 @@ def parse(string):
                 current += c
         elif stack and stack[-1] in ["'", '"', '(']: # in quote or args
             current += c
-        elif c == ".":
-            retval.append(current)
+        elif c in [".", "[", "]"]:
+            if current:
+                retval.append(current)
             current = ""
         else:
             current += c
@@ -383,12 +384,67 @@ class Struct(object):
         else:
             self.transaction = None
 
+    def __eq__(self, other):
+        if isinstance(other, Struct):
+            return self.struct == other.struct
+        else:
+            return self.struct == other
+
+    def __lt__(self, other):
+        if isinstance(other, Struct):
+            return self.struct < other.struct
+        else:
+            return self.struct < other
+
+    def __gt__(self, other):
+        if isinstance(other, Struct):
+            return self.struct > other.struct
+        else:
+            return self.struct > other
+
+    def __le__(self, other):
+        if isinstance(other, Struct):
+            return self.struct <= other.struct
+        else:
+            return self.struct <= other
+
+    def __ge__(self, other):
+        if isinstance(other, Struct):
+            return self.struct >= other.struct
+        else:
+            return self.struct >= other
+
+    def __ne__(self, other):
+        if isinstance(other, Struct):
+            return self.struct != other.struct
+        else:
+            return self.struct != other
+
+    def __len__(self):
+        return len(self.struct)
+
+    def __getattr__(self, attr):
+        """
+        Called when getattr fails. Lookup attr in struct; returns Struct
+        if more struct.  This is used only in eval for where clause. 
+        self.setitem_from_path(path, v) should be used to set value of 
+        item.
+        """
+        # for where eval:
+        if attr in self.struct:
+            return Struct(self.getitem(attr), self.db)
+        else:
+            raise AttributeError("no such attribute '%s'" % attr)
+
     def __getitem__(self, path): 
         """
         Given a path to a struct part, return the part, or None.
 
         >>> Struct(struct)["primary_name.surname_list.0.surname"]
         """
+        # For where eval:
+        if isinstance(path, int):
+            return Struct(self.struct[path], self.db)
         # Work way down to last part:
         return self.getitem_from_path(parse(path))
 
@@ -438,7 +494,7 @@ class Struct(object):
         elif hasattr(struct, item):
             return getattr(struct, item)
         elif item.startswith("("):
-            args = eval(item[1:-1] + ",")
+            args = eval(item[1:-1] + ",") # tuple of args
             return struct(*args)
         else:
             return None