File tree 1 file changed +10
-14
lines changed
1 file changed +10
-14
lines changed Original file line number Diff line number Diff line change @@ -201,37 +201,33 @@ def <%- pythonMethodName %>(<%- methodArgs.join(', ') %>):
201
201
https: // stackoverflow.com/questions/45399347/dictcursor-vs-realdictcursor
202
202
-% >
203
203
with conn:
204
- < % if (hasGetOne) {
205
- if (queries .length > 1 ) { /* we can omit cursor_factory but in this case we might get an unused import */ -% >
204
+ < % if (hasGetOne && queries .length > 1 ) { /* we can omit cursor_factory but in this case we might get an unused import */ -% >
206
205
with conn .cursor (cursor_factory= psycopg2 .extras .DictCursor ) as cur:
207
206
result = {}
208
- < % queries .forEach (queryInfo => {
209
- for (const [name , query ] of Object .entries (queryInfo)) {
207
+ < % queries .forEach (queryInfo => {
208
+ for (const [name , query ] of Object .entries (queryInfo)) {
210
209
-% >
211
210
cur .execute (< %- query .sql % >< %- query .formattedParams % > )
212
211
result[' <%- name %>' ] = cur .fetchone ()[0 ]
213
212
< % }
214
- })
213
+ })
215
214
-% >
216
215
return result
217
216
< %
218
- } else {
219
- const query = queries[0 ].result
217
+ } else {
218
+ const query = queries[0 ].result
220
219
-% >
221
220
with conn .cursor (cursor_factory= psycopg2 .extras .RealDictCursor ) as cur:
222
221
cur .execute (< %- query .sql % >< %- query .formattedParams % > )
222
+ < % if (hasGetMany) { -% >
223
+ return cur .fetchall ()
224
+ < % } else { /* GET with a single result */ -% >
223
225
result = cur .fetchone ()
224
226
if result is None:
225
227
raise HTTPException (status_code = status .HTTP_404_NOT_FOUND )
226
228
return result
227
- < % }
228
- } else {
229
- const query = queries[0 ].result
230
- -% >
231
- with conn .cursor (cursor_factory= psycopg2 .extras .RealDictCursor ) as cur:
232
- cur .execute (< %- query .sql % >< %- query .formattedParams % > )
233
- return cur .fetchall ()
234
229
< %
230
+ }
235
231
}
236
232
-% >
237
233
finally:
You can’t perform that action at this time.
0 commit comments