DRF 格式化 XLSX 内容

Question

我正在尝试在 XLSX 文件的每一行设置不同的颜色。从文档中我看到我可以使用 body 属性或 get_body() 方法传递一些条件，但这只允许我设置一些 "static" 条件。这是负责呈现 XLSX 文件的 ViewSet 配置：

class MyViewSet(XLSXFileMixin, ModelViewSet):
    def get_renderers(self) -> List[BaseRenderer]:
        if self.action == "export":
            return [XLSXRenderer()]
        else:
            return super().get_renderers()

    @action(methods=["GET"], detail=False)
    def export(self, request: Request) -> Response:
        serializer = self.get_serializer(self.get_queryset(), many=True)
        return Response(serializer.data)

    # Properties for XLSX
    column_header = {
        "titles": [
            "Hostname", "Operating System", "OS name", "OS family", "OS version", "Domain", "Serial number",
            "Available patches",
        ],
        "tab_title": "Endpoints",
        "style": {
            "font": {
                "size": 14,
                "color": "FFFFFF",
            },
            "fill": {
                "start_color": "3F803F",
                "fill_type": "solid",
            }
        }
    }

    body = {
        "style": {
            "font": {
                "size": 12,
                "color": "FFFFFF"
            },
            "fill": {
                "fill_type": "solid",
                "start_color": "2B2B2B"
            },
        }
    }

Answer 1

好的。在对源代码进行一些挖掘后，我得到了答案。 XLSXRenderer的render方法有这段代码：

for row in results:
    column_count = 0
    row_count += 1
    flatten_row = self._flatten(row)
    for column_name, value in flatten_row.items():
        if column_name == "row_color":
            continue
        column_count += 1
        cell = ws.cell(
            row=row_count, column=column_count, value=value,
        )
        cell.style = body_style
    ws.row_dimensions[row_count].height = body.get("height", 40)
    if "row_color" in row:
        last_letter = get_column_letter(column_count)
        cell_range = ws[
            "A{}".format(row_count): "{}{}".format(last_letter, row_count)
        ]
        fill = PatternFill(fill_type="solid", start_color=row["row_color"])
        for r in cell_range:
            for c in r:
                c.fill = fill

因此，当我在我的序列化程序中添加一个字段 row_color 作为 SerializerMethodField 时，我能够定义一个为行着色的函数：

def get_row_color(self, obj: Endpoint) -> str:
    """
    This method returns color value for row in XLSX sheet.
    (*self.instance,) extends queryset to a list (it must be a queryset, not a single Endpoint).
    .index(obj) gets index of currently serialized object in that list.
    As the last step one out of two values from the list is chosen using modulo 2 operation on the index.
    """
    return ["353535", "2B2B2B"][(*self.instance,).index(obj) % 2]

DRF 格式化 XLSX 内容

DRF formatting XLSX content

django

django-rest-framework

django-drf-renderer