diff --git a/db.py b/db.py index 056c41b0e5c1244440f63803b03196a20afa784a..311aef89168e7752ca98c2d7443893a646d6ac11 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):