generated files moved to 'assets' dir, graph drawing fixed

.gitignore

@@ -2,19 +2,14 @@
 *.env
 
 # Generated graph images:
-*.png
+assets/
 
 # Library roots:
-node_modules/
 venv/
 
 # Python caches:
 __pycache__/
 
-# Package manager configuration files:
-package.json
-package-lock.json
-
 # IDE configuration files:
 .vscode
 .idea

Makefile

@@ -31,13 +31,14 @@ run-locally: venv
 run-container:
 	@ # Run action in container
 	docker build -t waka-readme-stats -f Dockerfile .
-	docker run --env-file $(ENV) waka-readme-stats
+	docker run --env-file $(ENV) -v ./assets/:/waka-readme-stats/assets/ waka-readme-stats
 .PHONY: run-container
 
 
 clean:
 	@ # Clean all build files, including: libraries, package manager configs, docker images and containers
 	rm -rf venv
+	rm -rf assets
 	rm -f package*.json
 	docker rm -f waka-readme-stats 2>/dev/null || true
 	docker rmi $(docker images | grep "waka-readme-stats") 2> /dev/null || true

sources/graph_drawer.py

@@ -1,6 +1,4 @@
 from typing import Dict
-from os.path import join, dirname
-from json import load
 
 import numpy as np
 import matplotlib.patches as mpatches
@@ -12,28 +10,31 @@ from download_manager import DownloadManager
 MAX_LANGUAGES = 5
 
 
-async def build_graph(yearly_data: Dict) -> str:
+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.
-    :return: String, path to graph file.
+    :param save_path: Path to save the graph file.
     """
     colors = await DownloadManager.get_remote_yaml("linguist")
 
-    languages_all_loc = dict()
     years = len(yearly_data.keys())
     year_indexes = np.arange(years)
 
-    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]
+    all_languages = dict()
+    for year in yearly_data.values():
+        for quarter in year.values():
+            for language, loc in quarter.items():
+                all_languages[language] = all_languages.get(language, 0) + loc
 
-            for lang in langs:
-                if lang not in languages_all_loc:
-                    languages_all_loc[lang] = np.array([[0] * years] * 4)
-                languages_all_loc[lang][q - 1][i] = yearly_data[y][q][lang]
+    top_languages_names = sorted(all_languages.keys(), key=lambda l: all_languages[l], reverse=True)[0:MAX_LANGUAGES]
+    top_languages = {language: np.array([[0] * years] * 4) for language in top_languages_names}
+    for index, year in enumerate(sorted(yearly_data.keys())):
+        for quarter, languages in yearly_data[year].items():
+            for language, loc in {(lang, loc) for lang, loc in languages.items() if lang in top_languages}:
+                top_languages[language][quarter - 1][index] = yearly_data[year][quarter][language]
 
     fig = plt.figure()
     ax = fig.add_axes([0, 0, 1.5, 1])
@@ -41,7 +42,7 @@ async def build_graph(yearly_data: Dict) -> str:
     language_handles = []
     cumulative = np.array([[0] * years] * 4)
 
-    for key, value in languages_all_loc.items():
+    for key, value in top_languages.items():
         color = colors[key]["color"] if colors[key]["color"] is not None else "w"
         language_handles += [mpatches.Patch(color=color, label=key)]
 
@@ -64,6 +65,5 @@ async def build_graph(yearly_data: Dict) -> str:
     ax.spines["right"].set_visible(False)
 
     plt.ylim(0, 1.05 * np.amax(cumulative))
-    plt.savefig("bar_graph.png", bbox_inches="tight")
+    plt.savefig(save_path, bbox_inches="tight")
     plt.close(fig)
-    return "bar_graph.png"

sources/loc.py

@@ -5,10 +5,11 @@ from github import Github, InputGitAuthor, AuthenticatedUser
 import datetime
 
 from download_manager import DownloadManager
-from make_bar_graph import build_graph
+from graph_drawer import create_loc_graph
 
 
 class LinesOfCode:
+    GRAPH_PATH = "assets/bar_graph.png"
 
     def __init__(self, user: AuthenticatedUser, ghtoken, repositoryData, ignored_repos):
         self.g = Github(ghtoken)
@@ -28,7 +29,7 @@ class LinesOfCode:
         return yearly_data
 
     async def plotLoc(self, yearly_data):
-        await build_graph(yearly_data)
+        await create_loc_graph(yearly_data, LinesOfCode.GRAPH_PATH)
         self.pushChart()
 
     def getQuarter(self, timeStamp):
@@ -67,10 +68,10 @@ class LinesOfCode:
     def pushChart(self):
         repo = self.g.get_repo(f"{self.user.login}/{self.user.login}")
         committer = InputGitAuthor('readme-bot', '41898282+github-actions[bot]@users.noreply.github.com')
-        with open('bar_graph.png', 'rb') as input_file:
+        with open(LinesOfCode.GRAPH_PATH, 'rb') as input_file:
             data = input_file.read()
         try:
-            contents = repo.get_contents("charts/bar_graph.png")
+            contents = repo.get_contents(LinesOfCode.GRAPH_PATH)
             repo.update_file(contents.path, "Charts Updated", data, contents.sha, committer=committer)
         except Exception as e:
-            repo.create_file("charts/bar_graph.png", "Charts Added", data, committer=committer)
+            repo.create_file(LinesOfCode.GRAPH_PATH, "Charts Added", data, committer=committer)

sources/main.py

@@ -383,7 +383,7 @@ async def get_stats(github) -> str:
     if showLocChart.lower() in truthy:
         stats += '**' + translate['Timeline'] + '**\n\n'
         branch_name = github.get_repo(f'{user.login}/{user.login}').default_branch
-        stats = stats + '![Chart not found](https://raw.githubusercontent.com/' + user.login + '/' + user.login + '/' + branch_name + '/charts/bar_graph.png) \n\n'
+        stats = stats + '![Chart not found](https://raw.githubusercontent.com/' + user.login + '/' + user.login + '/' + branch_name + '/' + LinesOfCode.GRAPH_PATH + ') \n\n'
 
     if show_updated_date.lower() in truthy:
         now = datetime.datetime.utcnow()