Included summary stats

Author: technige

neo4j/__main__.py

@@ -83,6 +83,7 @@ def main():
     parser.add_argument("-s", "--secure", action="store_true")
     parser.add_argument("-v", "--verbose", action="count")
     parser.add_argument("-x", "--times", type=int, default=1)
+    parser.add_argument("-z", "--summarize", action="store_true")
     args = parser.parse_args()
 
     if args.verbose:
@@ -102,24 +103,30 @@ def main():
 
     driver = GraphDatabase.driver(args.url, secure=args.secure)
     session = driver.session()
-    if session:
-        for _ in range(args.times):
-            for statement in args.statement:
-                try:
-                    records = session.run(statement, parameters)
-                except CypherError as error:
-                    stderr.write("%s: %s\r\n" % (error.code, error.message))
-                else:
-                    if not args.quiet:
-                        has_results = False
-                        for i, record in enumerate(records):
-                            has_results = True
-                            if i == 0:
-                                stdout.write("%s\r\n" % "\t".join(record.__keys__))
-                            stdout.write("%s\r\n" % "\t".join(map(repr, record)))
-                        if has_results:
-                            stdout.write("\r\n")
-        session.close()
+    for _ in range(args.times):
+        for statement in args.statement:
+            try:
+                result = session.run(statement, parameters)
+            except CypherError as error:
+                stderr.write("%s: %s\r\n" % (error.code, error.message))
+            else:
+                if not args.quiet:
+                    has_results = False
+                    for i, record in enumerate(result):
+                        has_results = True
+                        if i == 0:
+                            stdout.write("%s\r\n" % "\t".join(record.__keys__))
+                        stdout.write("%s\r\n" % "\t".join(map(repr, record)))
+                    if has_results:
+                        stdout.write("\r\n")
+                    if args.summarize:
+                        summary = result.summarize()
+                        stdout.write("Statement      : %r\r\n" % summary.statement)
+                        stdout.write("Parameters     : %r\r\n" % summary.parameters)
+                        stdout.write("Statement Type : %r\r\n" % summary.statement_type)
+                        stdout.write("Statistics     : %r\r\n" % summary.statistics)
+                        stdout.write("\r\n")
+    session.close()
 
 
 if __name__ == "__main__":

neo4j/session.py

@@ -36,6 +36,12 @@ class which can be used to obtain `Driver` instances that are used for
 from .typesystem import hydrated
 
 
+STATEMENT_TYPE_READ_ONLY = "r"
+STATEMENT_TYPE_READ_WRITE = "rw"
+STATEMENT_TYPE_WRITE_ONLY = "w"
+STATEMENT_TYPE_SCHEMA_WRITE = "sw"
+
+
 Latency = namedtuple("Latency", ["overall", "network", "wait"])
 
 
@@ -98,11 +104,14 @@ def session(self, **config):
 
 class Result(list):
 
-    def __init__(self, session):
-        self.session = session
+    def __init__(self, session, statement, parameters):
         super(Result, self).__init__()
+        self.session = session
+        self.statement = statement
+        self.parameters = parameters
         self.keys = None
         self.complete = False
+        self.summary = None
         self.bench_test = None
 
     def on_header(self, metadata):
@@ -115,6 +124,8 @@ def on_record(self, values):
 
     def on_footer(self, metadata):
         self.complete = True
+        self.summary = ResultSummary(self.statement, self.parameters,
+                                     metadata.get("type"), metadata.get("stats"))
         if self.bench_test:
             self.bench_test.end_recv = perf_counter()
 
@@ -126,6 +137,42 @@ def consume(self):
         while not self.complete:
             fetch_next()
 
+    def summarize(self):
+        self.consume()
+        return self.summary
+
+
+class ResultSummary(object):
+
+    def __init__(self, statement, parameters, statement_type, statistics):
+        self.statement = statement
+        self.parameters = parameters
+        self.statement_type = statement_type
+        self.statistics = StatementStatistics(statistics or {})
+
+
+class StatementStatistics(object):
+    contains_updates = False
+    nodes_created = 0
+    nodes_deleted = 0
+    relationships_created = 0
+    relationships_deleted = 0
+    properties_set = 0
+    labels_added = 0
+    labels_removed = 0
+    indexes_added = 0
+    indexes_removed = 0
+    constraints_added = 0
+    constraints_removed = 0
+
+    def __init__(self, statistics):
+        for key, value in dict(statistics).items():
+            key = key.replace("-", "_")
+            setattr(self, key, value)
+
+    def __repr__(self):
+        return repr(vars(self))
+
 
 class Session(object):
     """ Logical session carried out over an established TCP connection.
@@ -162,7 +209,7 @@ def run(self, statement, parameters=None):
         t = BenchTest()
         t.init = perf_counter()
 
-        result = Result(self)
+        result = Result(self, statement, parameters)
         result.bench_test = t
 
         run_response = Response(self.connection)