From 01e3fbac5e809222787e353038cfdbb26cf5b7ed Mon Sep 17 00:00:00 2001
From: Julian Rother <julianr@fsmpi.rwth-aachen.de>
Date: Fri, 9 Sep 2016 23:21:02 +0200
Subject: [PATCH] Add delimiter grouping support for queries
---
db.py | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/db.py b/db.py
index 056c41b..311aef8 100644
--- a/db.py
+++ b/db.py
@@ -64,12 +64,16 @@ def query(operation, *params, delim="sep"):
rows = cur.fetchall()
res = []
for row in rows:
- res.append({})
+ res.append([{}])
for col, desc in zip(row, cur.description):
name = desc[0].split('.')[-1]
- if type(col) == str:
+ if name == delim:
+ res[-1].append({})
+ elif type(col) == str:
col = col.replace('\\n', '\n').replace('\\r', '\r')
- res[-1][name] = col
+ res[-1][-1][name] = col
+ if len(res[-1]) == 1:
+ res[-1] = res[-1][0]
return res
def modify(operation, *params):
--
GitLab