feat(python): implement generation of PUT endpoints

Part of #16

Author: php-coder

examples/python/fastapi/postgres/routes.py

@@ -144,8 +144,22 @@ def get_v1_categories_category_id(categoryId, conn=Depends(db_connection)):
 
 
 @router.put('/v1/categories/{categoryId}', status_code = status.HTTP_204_NO_CONTENT)
-def put_v1_categories_category_id():
-    pass
+def put_v1_categories_category_id(body: CreateCategoryDto, categoryId, conn=Depends(db_connection)):
+    try:
+        with conn:
+            with conn.cursor() as cur:
+                cur.execute(
+                    """
+                    UPDATE categories
+                       SET name = %(name)s
+                         , name_ru = %(name_ru)s
+                         , slug = %(slug)s
+                         , updated_at = NOW()
+                         , updated_by = %(user_id)s
+                     WHERE id = %(categoryId)s
+                    """, {"name": body.name, "name_ru": body.name_ru, "slug": body.slug, "user_id": body.user_id, "categoryId": categoryId})
+    finally:
+        conn.close()
 
 
 @router.delete('/v1/categories/{categoryId}', status_code = status.HTTP_204_NO_CONTENT)

src/templates/routes.py.ejs

@@ -44,6 +44,7 @@ function formatQueryForPython(query, indentLevel) {
     return `"${sql}"`
 }
 
+// LATER: reduce duplication with routes.go.ejs
 // {'values':
 //   [
 //     {
@@ -59,11 +60,30 @@ function extractInsertValues(queryAst) {
 	return Array.from(new Set(values));
 }
 
+// LATER: reduce duplication with routes.go.ejs
+// {'set':
+//   [
+//     {
+//       column: 'updated_by',
+//       value: { type: 'param', value: 'user_id' },
+//       table: null
+//     }
+//   ]
+// } => [ 'user_id' ]
+function extractUpdateValues(queryAst) {
+	// LATER: distinguish between b.param and q.param and extract only first
+	return queryAst.set.map(elem => elem.value.type === 'param' ? elem.value.value : null)
+		.filter(value => value) // filter out nulls
+}
+
 // LATER: reduce duplication with routes.go.ejs
 function extractProperties(queryAst) {
 	if (queryAst.type === 'insert') {
 		return extractInsertValues(queryAst);
 	}
+    if (queryAst.type === 'update') {
+		return extractUpdateValues(queryAst);
+	}
 	return [];
 }
 
@@ -259,11 +279,27 @@ def <%- pythonMethodName %>(body: <%- model %>, conn=Depends(db_connection)):
 
         }
         if (method.name === 'put') {
+            // TODO: reduce duplication with POST
+            const dto = query2dto(sqlParser, method);
+			// LATER: do we really need signature and cache?
+			const cacheKey = dto ? dto.signature : null;
+			const model = dtoInCache(dto) ? dtoCache[cacheKey] : dto.name;
+
+            const methodArgs = [`body: ${model}`, ...argsFromPath, 'conn=Depends(db_connection)']
+
+            const sql = convertToPsycopgNamedArguments(formatQueryForPython(method.query, 20))
+            const params = extractParamsFromQuery(method.query);
+            const formattedParams = formatParamsAsPythonDict(params)
 %>
 
 @router.put('<%- path %>', status_code = status.HTTP_204_NO_CONTENT)
-def <%- pythonMethodName %>():
-    pass
+def <%- pythonMethodName %>(<%- methodArgs.join(', ') %>):
+    try:
+        with conn:
+            with conn.cursor() as cur:
+                cur.execute(<%- sql %><%- formattedParams %>)
+    finally:
+        conn.close()
 <%
 
         }