add row_attributes and cell_attributes

This commit is contained in:
long2ice
2021-05-05 23:14:00 +08:00
parent 9b14227d24
commit e17d6a38c6
8 changed files with 63 additions and 23 deletions

View File

@ -26,10 +26,18 @@ General `a` tag menu.
`target` attr of `a` tag, default is `_self`.
## Field
## Action
## Model
### Field
### get_actions
### Action
### get_bulk_actions
### row_attributes
### cell_attributes
## Dropdown

View File

@ -28,7 +28,10 @@ class AdminResource(Model):
page_title = "admin model"
filters = [
filters.Search(
name="username", label="Name", search_mode="contains", placeholder="Search for username"
name="username",
label="Name",
search_mode="contains",
placeholder="Search for username",
),
filters.Date(name="created_at", label="CreatedAt"),
]
@ -52,6 +55,11 @@ class AdminResource(Model):
]
can_create = False
def cell_attributes(self, obj: dict, field: Field) -> dict:
if field.name == "id":
return {"class": "bg-danger"}
return super().cell_attributes(obj, field)
def get_actions(self) -> List[Action]:
return []
@ -111,6 +119,11 @@ class ConfigResource(Model):
),
]
def row_attributes(self, obj: dict) -> dict:
if obj.get("status") == enums.Status.on:
return {"class": "bg-green text-white"}
return super(ConfigResource, self).row_attributes(obj)
@app.register
class GithubLink(Link):

View File

@ -68,7 +68,8 @@ class FastAPIAdmin(FastAPI):
self.resources.append(resource)
def get_model_resource(self, model: Type[Model]):
return self.model_resources[model]()
r = self.model_resources.get(model)
return r() if r else None
app = FastAPIAdmin(

View File

@ -21,6 +21,8 @@ def get_model(resource: Optional[str] = Path(...)):
def get_model_resource(request: Request, model=Depends(get_model)):
model_resource = request.app.get_model_resource(model) # type:Model
if not model_resource:
raise HTTPException(status_code=HTTP_404_NOT_FOUND)
actions = model_resource.get_actions()
bulk_actions = model_resource.get_bulk_actions()
model_resource.actions = actions

View File

@ -70,6 +70,12 @@ class Model(Resource):
can_create: bool = True
enctype = "application/x-www-form-urlencoded"
def row_attributes(self, obj: dict) -> dict:
return {}
def cell_attributes(self, obj: dict, field: Field) -> dict:
return {}
def get_actions(self) -> List[Action]:
return [
Action(label=_("update"), icon="ti ti-edit", name="update", ajax=False),
@ -78,7 +84,12 @@ class Model(Resource):
def get_bulk_actions(self) -> List[Action]:
return [
Action(label=_("delete_selected"), icon="ti ti-trash", name="delete", method="DELETE"),
Action(
label=_("delete_selected"),
icon="ti ti-trash",
name="delete",
method="DELETE",
),
]
@classmethod

View File

@ -37,14 +37,15 @@ async def list_view(
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)
values = await qs.values(*fields_name)
rendered_values = await render_values(fields, values)
context = {
"request": request,
"resources": resources,
"fields_label": fields_label,
"fields": fields,
"values": values,
"rendered_values": rendered_values,
"filters": filters,
"resource": resource,
"model_resource": model_resource,

View File

@ -58,10 +58,10 @@ async def render_values(
ret = []
for value in values:
item = []
for i, v in enumerate(value):
for i, k in enumerate(value):
if display:
item.append(await fields[i].display.render(v))
item.append(await fields[i].display.render(value[k]))
else:
item.append(await fields[i].input.render(v))
item.append(await fields[i].input.render(value[k]))
ret.append(item)
return ret

View File

@ -107,8 +107,8 @@
</tr>
</thead>
<tbody>
{% for value in values %}
<tr>
{% for value in rendered_values %}
<tr {% for k,v in model_resource.row_attributes(values[loop.index0]).items() %}{{ k }}="{{ v }}"{% endfor %}>
{% if model_resource.bulk_actions %}
<td>
<input
@ -118,9 +118,13 @@
/>
</td>
{% endif %}
{% for v in value %}
<td>{{ v|safe }}</td>
{% with outer_index = loop.index0 %}
{% for x in value %}
<td
{% for k,v in model_resource.cell_attributes(values[outer_index],fields[loop.index0]).items() %}
{{ k }}="{{ v }}"{% endfor %}>{{ x|safe }}</td>
{% endfor %}
{% endwith %}
{% if model_resource.actions %}
<td class="text-end">
<span class="dropdown">