feat(python): implement generation of DELETE endpoints

php-coder · php-coder · commit cb5b9412db0b · 2024-04-07T11:55:37.000+07:00
Part of #16
diff --git a/examples/python/fastapi/postgres/routes.py b/examples/python/fastapi/postgres/routes.py
@@ -163,5 +163,15 @@ def put_v1_categories_category_id(body: CreateCategoryDto, categoryId, conn=Depe
 
 
 @router.delete('/v1/categories/{categoryId}', status_code=status.HTTP_204_NO_CONTENT)
-def delete_v1_categories_category_id():
-    pass
+def delete_v1_categories_category_id(categoryId, conn=Depends(db_connection)):
+    try:
+        with conn:
+            with conn.cursor() as cur:
+                cur.execute(
+                    """
+                    DELETE
+                      FROM categories
+                     WHERE id = %(categoryId)s
+                    """, {"categoryId": categoryId})
+    finally:
+        conn.close()
diff --git a/src/templates/routes.py.ejs b/src/templates/routes.py.ejs
@@ -304,11 +304,21 @@ def <%- pythonMethodName %>(<%- methodArgs.join(', ') %>):
 
         }
         if (method.name === 'delete') {
+            const methodArgs = [ ...argsFromPath, 'conn=Depends(db_connection)' ]
+
+            const sql = convertToPsycopgNamedArguments(formatQueryForPython(method.query, 20))
+            const params = extractParamsFromQuery(method.query)
+            const formattedParams = formatParamsAsPythonDict(params)
 %>
 
 @router.delete('<%- 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()
 <%
 
         }
