wbreport 2.2.1__py2.py3-none-any.whl

wbreport/pdf/charts/legend.py

@@ -0,0 +1,15 @@
+from reportlab.graphics.charts.legends import Legend
+from reportlab.graphics.shapes import Circle
+from reportlab.lib.colors import transparent
+
+
+class CustomLegend(Legend):
+    def _defaultSwatch(self, x, thisy, dx, dy, fillColor, strokeWidth, strokeColor):
+        return Circle(
+            x,
+            thisy + dx / 2,
+            dx / 2,
+            fillColor=fillColor,
+            strokeColor=transparent,
+            strokeWidth=strokeWidth,
+        )

wbreport/pdf/charts/pie.py

@@ -0,0 +1,169 @@
+from reportlab.graphics.charts.piecharts import Pie
+from reportlab.graphics.shapes import Drawing
+from reportlab.lib import colors
+from reportlab.lib.units import cm
+from wbreport.pdf.charts.legend import CustomLegend
+
+
+def get_data_and_labels_from_df(df, color_palette, percent=True):
+    data = list()
+    colornamepairs = list()
+
+    for index, row in enumerate(df.itertuples()):
+        data.append(float(row.weighting))
+
+        label = f"{row[2]*100:.1f}%" if percent else f"{row[2]:.1f}"
+        colornamepairs.append((colors.HexColor(color_palette[index]), f"{row[1]} {label}"))
+
+    return data, colornamepairs
+
+
+def get_legend_height(df, r2_legend=0.3 * cm, legend_padding=1, legend_max_cols=7):
+    return min(len(df), legend_max_cols) * (r2_legend + legend_padding)
+
+
+def get_pie_chart_vertical_height(
+    df,
+    r2=2.5 * cm,
+    padding=0.45 * cm,
+    r2_legend=0.3 * cm,
+    legend_padding=1,
+    legend_max_cols=7,
+):
+    return 3 * padding + r2 + get_legend_height(df, r2_legend, legend_padding, legend_max_cols)
+
+
+def get_pie_chart_vertical(
+    df,
+    width,
+    height,
+    color_palette,
+    r2=2.5 * cm,
+    padding=0.45 * cm,
+    r2_legend=0.3 * cm,
+    legend_padding=1,
+    legend_max_cols=7,
+):
+    drawing = Drawing(width, height)
+    pie = Pie()
+
+    pie.width = r2
+    pie.height = r2
+
+    pie.x = (width - r2) / 2
+    pie.y = height - r2 - padding
+    df = df.sort_values(by=["weighting"], ascending=False)
+    data, colornamepairs = get_data_and_labels_from_df(df, color_palette)
+
+    for i, color in enumerate(colornamepairs):
+        pie.slices[i].fillColor = color[0]
+        pie.slices[i].strokeColor = colors.transparent
+        pie.slices[i].strokeWidth = 0  # Border width for wedge
+    pie.slices.strokeWidth = 0  # Width of the border around the pie chart.
+    pie.data = data
+
+    drawing.add(pie)
+
+    legend = CustomLegend()
+
+    legend.alignment = "right"
+    legend.boxAnchor = "nw"
+    legend.x = 0
+    legend.y = height - r2 - 2 * padding
+    legend.dx = legend.dy = r2_legend
+    legend.strokeWidth = -1
+    legend.columnMaximum = legend_max_cols
+    legend.colorNamePairs = colornamepairs
+    legend.fontName = "customfont"
+    legend.fontSize = 6
+
+    legend.deltax = 0  # Here
+    legend.deltay = legend_padding
+    legend.swdx = 12
+    legend.swdy = 0
+
+    legend.dxTextSpace = r2_legend
+
+    legend.variColumn = True
+
+    drawing.add(legend)
+    return drawing
+
+
+def get_pie_chart_horizontal_height(
+    df,
+    r2=2.5 * cm,
+    padding=0.45 * cm,
+    r2_legend=0.3 * cm,
+    legend_padding=1,
+    legend_max_cols=7,
+):
+    return max(
+        2 * padding + r2,
+        2 * padding + get_legend_height(df, r2_legend, legend_padding, legend_max_cols),
+    )
+
+
+def get_pie_chart_horizontal(
+    df,
+    width,
+    height,
+    color_palette,
+    col_width,
+    r2=2.5 * cm,
+    padding=0.45 * cm,
+    r2_legend=0.3 * cm,
+    legend_padding=1,
+    legend_max_cols=7,
+    legend_x=None,
+):
+    drawing = Drawing(width, height)
+    pie = Pie()
+
+    pie.width = r2
+    pie.height = r2
+
+    pie.x = (col_width - r2) / 2
+    pie.y = height - r2 - padding
+    df = df.sort_values(by=["weighting"], ascending=False)
+    data, colornamepairs = get_data_and_labels_from_df(df, color_palette)
+
+    for i, color in enumerate(colornamepairs):
+        pie.slices[i].fillColor = color[0]
+        pie.slices[i].strokeColor = colors.transparent
+        pie.slices[i].strokeWidth = 0  # Border width for wedge.
+    pie.slices.strokeWidth = 0  # Width of the border around the pie chart.
+    pie.data = data
+
+    drawing.add(pie)
+
+    legend = CustomLegend()
+
+    # legend.x = 4.82 * cm
+    # legend.y += 0.78 * cm
+
+    legend_height = get_legend_height(df, r2_legend, legend_padding, legend_max_cols)
+
+    legend.alignment = "right"
+    legend.boxAnchor = "nw"
+
+    if legend_x:
+        legend.x = legend_x
+    else:
+        legend.x = width - col_width
+
+    legend.y = (height + legend_height) / 2
+    legend.dx = legend.dy = r2_legend
+    legend.strokeWidth = -1
+    legend.columnMaximum = legend_max_cols
+    legend.colorNamePairs = colornamepairs
+    legend.fontName = "customfont"
+    legend.fontSize = 6
+
+    legend.deltax = 0
+    legend.deltay = legend_padding
+    legend.swdx = 12
+    legend.swdy = 0
+
+    drawing.add(legend)
+    return drawing

wbreport/pdf/charts/timeseries.py

@@ -0,0 +1,77 @@
+from enum import Enum
+
+from reportlab.graphics.charts.axes import LogYValueAxis, NormalDateXValueAxis
+from reportlab.graphics.charts.lineplots import LinePlot, SimpleTimeSeriesPlot
+from reportlab.graphics.shapes import Drawing
+from reportlab.lib import colors
+from reportlab.lib.units import cm
+
+
+class Scale(Enum):
+    ARITHMETIC = 1
+    LOGARITHMIC = 2
+
+
+class LogScaleTimeSeriesPlot(LinePlot):
+    def __init__(self):
+        super().__init__()
+
+        class CustomYAxis(LogYValueAxis):
+            def _calcTickPositions(self):
+                return self._calcStepAndTickPositions()[1]
+
+        self.xValueAxis = NormalDateXValueAxis()
+        self.yValueAxis = CustomYAxis()
+
+
+def get_timeseries_chart(
+    data, width, height, color, fill_color, grid_color, scale, x_label_format="{mm}/{yy}", **chart_attributes
+):
+    width -= 0.65 * cm
+    chart_map = {
+        Scale.ARITHMETIC.value: SimpleTimeSeriesPlot,
+        Scale.LOGARITHMIC.value: LogScaleTimeSeriesPlot,
+    }
+    data = [list(filter(lambda x: x[1] > 0, data[0]))]
+    max_x_ticks = 20
+
+    # ensure that if the data count is too low (e.g. less than a month range), we don't overcrowds the X axis
+    if len(data[0]) < max_x_ticks:
+        x_label_format = "{dd}/{mm}/{yy}"
+        max_x_ticks = 10
+
+    drawing = Drawing(width, height)
+
+    chart = chart_map[scale]()
+    chart.width = width
+    chart.height = height
+
+    for key, value in chart_attributes.items():
+        setattr(chart, key, value)
+
+    chart.data = data
+
+    chart.lines[0].strokeWidth = 0.5
+    chart.lines[0].strokeColor = color
+    chart.lines[0].fillColor = fill_color
+    chart.lines[0].inFill = True
+
+    chart.yValueAxis.strokeWidth = -1
+    chart.yValueAxis.strokeColor = colors.white
+    chart.yValueAxis.labels.fontSize = 7
+    chart.yValueAxis.labels.fontName = "customfont"
+    chart.yValueAxis.labels.dx = width + 0.55 * cm
+    chart.yValueAxis.maximumTicks = 100
+    chart.yValueAxis.labelTextFormat = "%d"
+
+    chart.xValueAxis.strokeWidth = 0
+    chart.xValueAxis.labels.fontSize = 7
+    chart.xValueAxis.labels.fontName = "customfont"
+    chart.xValueAxis.maximumTicks = max_x_ticks
+    chart.xValueAxis.visibleGrid = 1
+    chart.xValueAxis.gridStrokeDashArray = (0.2, 0, 0.2)
+    chart.xValueAxis.gridStrokeColor = grid_color
+    chart.xValueAxis.xLabelFormat = x_label_format
+
+    drawing.add(chart)
+    return drawing

wbreport/pdf/sandbox/run.py

@@ -0,0 +1,17 @@
+import importlib
+import sys
+from datetime import datetime
+
+from wbportfolio.models import Product
+
+report = sys.argv[1]
+
+module = importlib.import_module("wbreport.pdf.sandbox.templates.{report}")
+product = Product.objects.get(id=sys.argv[2])
+
+start = datetime.strptime(sys.argv[3], "%Y-%m-%d").date()
+end = datetime.strptime(sys.argv[4], "%Y-%m-%d").date()
+context = product.report._get_file_context(start=start, end=end)
+result = module.generate_report(context)
+with open("portfolio/report/pdf/sandbox/templates/testfile.pdf", "wb") as test_file:
+    test_file.write(result.read())