new project

fastapi_admin/routes.py

@@ -0,0 +1,182 @@
+from typing import Optional
+
+from fastapi import APIRouter, Depends, Path
+from starlette.requests import Request
+from starlette.responses import RedirectResponse
+from tortoise import Model
+
+from fastapi_admin.depends import get_model, get_model_resource, get_resources
+from fastapi_admin.resources import Model as ModelResource
+from fastapi_admin.responses import redirect
+from fastapi_admin.template import render_values, templates
+
+router = APIRouter()
+
+
+@router.get("/maintenance")
+async def maintenance(request: Request):
+    return templates.TemplateResponse("errors/maintenance.html", context={"request": request})
+
+
+@router.get("/list/{resource}")
+async def list_view(
+    request: Request,
+    model: Model = Depends(get_model),
+    resources=Depends(get_resources),
+    model_resource: ModelResource = Depends(get_model_resource),
+    resource: str = Path(...),
+    page_size: Optional[int] = None,
+    page_num: int = 1,
+):
+    fields_name = model_resource.get_fields_name()
+    fields_label = model_resource.get_fields_label()
+    fields = model_resource.get_fields()
+    params = await model_resource.resolve_query_params(dict(request.query_params))
+    filters = await model_resource.get_filters(params)
+    qs = model.filter(**params)
+    total = await qs.count()
+    if page_size:
+        qs = qs.limit(page_size)
+    else:
+        page_size = model_resource.page_size
+    qs = qs.offset((page_num - 1) * page_size)
+    values = await qs.values_list(*fields_name)
+    values = await render_values(fields, values)
+    return templates.TemplateResponse(
+        "list.html",
+        context={
+            "request": request,
+            "resources": resources,
+            "fields_label": fields_label,
+            "fields": fields,
+            "values": values,
+            "filters": filters,
+            "resource": resource,
+            "model_resource": model_resource,
+            "resource_label": model_resource.label,
+            "page_size": page_size,
+            "page_num": page_num,
+            "total": total,
+            "from": page_size * (page_num - 1) + 1,
+            "to": page_size * page_num,
+            "page_title": model_resource.page_title,
+            "page_pre_title": model_resource.page_pre_title,
+        },
+    )
+
+
+@router.post("/edit/{resource}/{pk}")
+async def edit(
+    request: Request,
+    resource: str = Path(...),
+    pk: int = Path(...),
+    model_resource: ModelResource = Depends(get_model_resource),
+    resources=Depends(get_resources),
+    model=Depends(get_model),
+):
+    form = await request.form()
+    data = await model_resource.resolve_data(dict(form))
+    obj = await model.get(pk=pk)
+    await obj.update_from_dict(data).save()
+    inputs = await model_resource.get_inputs(obj)
+    if "save" in form.keys():
+        return templates.TemplateResponse(
+            "edit.html",
+            context={
+                "request": request,
+                "resources": resources,
+                "resource_label": model_resource.label,
+                "resource": resource,
+                "model_resource": model_resource,
+                "inputs": inputs,
+                "pk": pk,
+                "page_title": model_resource.page_title,
+                "page_pre_title": model_resource.page_pre_title,
+            },
+        )
+    return redirect(request, "list_view", resource=resource)
+
+
+@router.get("/edit/{resource}/{pk}")
+async def edit_view(
+    request: Request,
+    resource: str = Path(...),
+    pk: int = Path(...),
+    model_resource: ModelResource = Depends(get_model_resource),
+    resources=Depends(get_resources),
+    model=Depends(get_model),
+):
+    obj = await model.get(pk=pk)
+    inputs = await model_resource.get_inputs(obj)
+    return templates.TemplateResponse(
+        "edit.html",
+        context={
+            "request": request,
+            "resources": resources,
+            "resource_label": model_resource.label,
+            "resource": resource,
+            "inputs": inputs,
+            "pk": pk,
+            "model_resource": model_resource,
+            "page_title": model_resource.page_title,
+            "page_pre_title": model_resource.page_pre_title,
+        },
+    )
+
+
+@router.get("/create/{resource}")
+async def create_view(
+    request: Request,
+    resource: str = Path(...),
+    resources=Depends(get_resources),
+    model_resource: ModelResource = Depends(get_model_resource),
+):
+    inputs = await model_resource.get_inputs()
+    return templates.TemplateResponse(
+        "create.html",
+        context={
+            "request": request,
+            "resources": resources,
+            "resource_label": model_resource.label,
+            "resource": resource,
+            "inputs": inputs,
+            "model_resource": model_resource,
+            "page_title": model_resource.page_title,
+            "page_pre_title": model_resource.page_pre_title,
+        },
+    )
+
+
+@router.post("/create/{resource}")
+async def create(
+    request: Request,
+    resource: str = Path(...),
+    resources=Depends(get_resources),
+    model_resource: ModelResource = Depends(get_model_resource),
+    model=Depends(get_model),
+):
+    inputs = await model_resource.get_inputs()
+    form = await request.form()
+    data = await model_resource.resolve_data(dict(form))
+    await model.create(**data)
+    if "save" in form.keys():
+        return redirect(request, "list_view", resource=resource)
+    return templates.TemplateResponse(
+        "create.html",
+        context={
+            "request": request,
+            "resources": resources,
+            "resource_label": model_resource.label,
+            "resource": resource,
+            "inputs": inputs,
+            "model_resource": model_resource,
+            "page_title": model_resource.page_title,
+            "page_pre_title": model_resource.page_pre_title,
+        },
+    )
+
+
+@router.get("/delete/{resource}/{pk}")
+async def delete_view(request: Request, pk: int, model: Model = Depends(get_model)):
+    await model.filter(pk=pk).delete()
+    return RedirectResponse(url=request.headers.get("referer"))