mirror of
https://github.com/fastapi-admin/fastapi-admin.git
synced 2025-08-15 03:04:37 +08:00
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from typing import List, Optional, Type
|
|
|
|
from fastapi import Depends
|
|
from fastapi.params import Path
|
|
from starlette.requests import Request
|
|
from tortoise import Tortoise
|
|
|
|
from fastapi_admin.exceptions import InvalidResource
|
|
from fastapi_admin.resources import Dropdown, Link, Model, Resource
|
|
|
|
|
|
def get_model(resource: Optional[str] = Path(...)):
|
|
if not resource:
|
|
return
|
|
for app, models in Tortoise.apps.items():
|
|
model = models.get(resource.title())
|
|
if model:
|
|
return model
|
|
|
|
|
|
def get_model_resource(request: Request, model=Depends(get_model)):
|
|
model_resource = request.app.get_model_resource(model) # type:Model
|
|
actions = model_resource.get_actions()
|
|
bulk_actions = model_resource.get_bulk_actions()
|
|
model_resource.actions = actions
|
|
model_resource.bulk_actions = bulk_actions
|
|
return model_resource
|
|
|
|
|
|
def _get_resources(resources: List[Type[Resource]]):
|
|
ret = []
|
|
for resource in resources:
|
|
item = {
|
|
"icon": resource.icon,
|
|
"label": resource.label,
|
|
}
|
|
if issubclass(resource, Link):
|
|
item["type"] = "link"
|
|
item["url"] = resource.url
|
|
item["target"] = resource.target
|
|
elif issubclass(resource, Model):
|
|
item["type"] = "model"
|
|
item["model"] = resource.model.__name__.lower()
|
|
elif issubclass(resource, Dropdown):
|
|
item["type"] = "dropdown"
|
|
item["resources"] = _get_resources(resource.resources)
|
|
else:
|
|
raise InvalidResource("Should be subclass of Resource")
|
|
ret.append(item)
|
|
return ret
|
|
|
|
|
|
def get_resources(request: Request) -> List[dict]:
|
|
resources = request.app.resources
|
|
return _get_resources(resources)
|
|
|
|
|
|
def get_redis(request: Request):
|
|
return request.app.redis
|
|
|
|
|
|
def get_current_admin(request: Request):
|
|
return request.state.admin
|