code style applied to main

2023-02-17 15:34:26 +01:00
parent e8a1770feb
commit 8e675eaafd
12 changed files with 442 additions and 473 deletions
--- a/sources/graphics_chart_drawer.py
+++ b/sources/graphics_chart_drawer.py
@@ -0,0 +1,66 @@
+from typing import Dict
+
+from numpy import arange, array, add, amax
+import matplotlib.patches as mpatches
+import matplotlib.pyplot as plt
+
+from manager_download import DownloadManager as DM
+
+
+MAX_LANGUAGES = 5
+
+
+async def create_loc_graph(yearly_data: Dict, save_path: str):
+    """
+    Draws graph of lines of code written by user by quarters of years.
+    Picks top `MAX_LANGUAGES` languages from each quarter only.
+
+    :param yearly_data: GitHub user yearly data.
+    :param save_path: Path to save the graph file.
+    """
+    colors = await DM.get_remote_yaml("linguist")
+
+    years = len(yearly_data.keys())
+    year_indexes = arange(years)
+
+    languages_all_loc = dict()
+    for i, y in enumerate(sorted(yearly_data.keys())):
+        for q in yearly_data[y].keys():
+            langs = sorted(yearly_data[y][q].keys(), key=lambda l: yearly_data[y][q][l], reverse=True)[0:MAX_LANGUAGES]
+
+            for lang in langs:
+                if lang not in languages_all_loc:
+                    languages_all_loc[lang] = array([[0] * years] * 4)
+                languages_all_loc[lang][q - 1][i] = yearly_data[y][q][lang]
+
+    fig = plt.figure()
+    ax = fig.add_axes([0, 0, 1.5, 1])
+
+    language_handles = []
+    cumulative = array([[0] * years] * 4)
+
+    for key, value in languages_all_loc.items():
+        color = colors[key]["color"] if colors[key]["color"] is not None else "w"
+        language_handles += [mpatches.Patch(color=color, label=key)]
+
+        for quarter in range(4):
+            ax.bar(year_indexes + quarter * 0.21, value[quarter], 0.2, bottom=cumulative[quarter], color=color)
+            cumulative[quarter] = add(cumulative[quarter], value[quarter])
+
+    ax.set_ylabel("LOC added", fontdict=dict(weight="bold"))
+    ax.set_xticks(array([arange(i, i + 0.84, step=0.21) for i in year_indexes]).flatten(), labels=["Q1", "Q2", "Q3", "Q4"] * years)
+
+    sax = ax.secondary_xaxis("top")
+    sax.set_xticks(year_indexes + 0.42, labels=sorted(yearly_data.keys()))
+    sax.spines["top"].set_visible(False)
+
+    ax.legend(title="Language", handles=language_handles, loc="upper left", bbox_to_anchor=(1, 1), framealpha=0, title_fontproperties=dict(weight="bold"))
+
+    sax.tick_params(axis="both", length=0)
+    sax.spines["top"].set_visible(False)
+    ax.spines["top"].set_visible(False)
+    ax.spines["right"].set_visible(False)
+
+    plt.ylim(0, 1.05 * amax(cumulative))
+    plt.savefig(save_path, bbox_inches="tight")
+    plt.close(fig)