Makefile added; targets for local action run and image building added; workflow for image building and publishing added; image building in action replaced with container pulling

sources/loc.py

@@ -0,0 +1,108 @@
+import re
+import os
+import base64
+import requests
+from github import Github, InputGitAuthor
+import datetime
+from string import Template
+import matplotlib.pyplot as plt
+from io import StringIO, BytesIO
+from dotenv import load_dotenv
+import time
+
+from make_bar_graph import BarGraph
+
+
+class LinesOfCode:
+
+    def __init__(self, id, username, ghtoken, repositoryData, ignored_repos):
+        self.id = id
+        self.username = username
+
+        self.g = Github(ghtoken)
+        self.headers = {"Authorization": "Bearer " + ghtoken}
+        self.repositoryData = repositoryData
+        self.ignored_repos = ignored_repos
+
+    def calculateLoc(self):
+        result = self.repositoryData
+        yearly_data = {}
+        for repo in result['data']['user']['repositories']['edges']:
+            if repo['node']['name'] not in self.ignored_repos:
+                self.getCommitStat(repo['node'], yearly_data)
+                time.sleep(0.7)
+        return yearly_data
+
+    def plotLoc(self, yearly_data):
+        graph = BarGraph(yearly_data)
+        graph.build_graph()
+        self.pushChart()
+
+    def run_query_v3(self, endPoint):
+        # print(endPoint)
+        request = requests.get(endPoint, headers=self.headers)
+        if request.status_code == 401:
+            raise Exception("Invalid token {}.".format(request.status_code))
+        elif request.status_code == 204:
+            return []
+        else:
+            return request.json()
+
+    def getQuarter(self, timeStamp):
+        month = datetime.datetime.fromisoformat(timeStamp).month
+        if month >= 1 and month <= 3:
+            return 1
+        elif month >= 4 and month <= 6:
+            return 2
+        elif month >= 7 and month <= 9:
+            return 3
+        elif month >= 10 and month <= 12:
+            return 4
+
+    def getCommitStat(self, repoDetails, yearly_data):
+        commitsURL = 'https://api.github.com/repos/' + repoDetails['nameWithOwner'] + '/commits'
+        filteredCommitsEndPoint = commitsURL + '?author=' + self.username
+        filteredCommitsResult = self.run_query_v3(filteredCommitsEndPoint)
+        # This ignores the error message you get when you try to list commits for an empty repository
+        if not type(filteredCommitsResult) == list:
+            return
+        this_year = datetime.datetime.utcnow().year
+
+        for i in range(len(filteredCommitsResult)):
+            iso_date = filteredCommitsResult[i]["commit"]["author"]["date"]
+            date = re.search(r'\d+-\d+-\d+', iso_date).group(0)
+            curr_year = datetime.datetime.fromisoformat(date).year
+            # if  curr_year != this_year:
+
+            individualCommitEndPoint = commitsURL + '/' + filteredCommitsResult[i]["sha"]
+            individualCommitResult = self.run_query_v3(individualCommitEndPoint)
+
+            quarter = self.getQuarter(date)
+            if repoDetails['primaryLanguage'] is not None:
+
+                if curr_year not in yearly_data:
+                    yearly_data[curr_year] = {}
+                if quarter not in yearly_data[curr_year]:
+                    yearly_data[curr_year][quarter] = {}
+                if repoDetails['primaryLanguage']['name'] not in yearly_data[curr_year][quarter]:
+                    yearly_data[curr_year][quarter][repoDetails['primaryLanguage']['name']] = 0
+                yearly_data[curr_year][quarter][repoDetails['primaryLanguage']['name']] += (individualCommitResult["stats"]["additions"] - individualCommitResult["stats"]['deletions'])
+
+                # to find total
+
+                # if 'total' not in yearly_data[curr_year]:
+                #   yearly_data[curr_year]['total']={}
+                # if repoDetails['primaryLanguage']['name'] not in yearly_data[curr_year]['total']:
+                #   yearly_data[curr_year]['total'][repoDetails['primaryLanguage']['name']]=0
+                # yearly_data[curr_year]['total'][repoDetails['primaryLanguage']['name']]+=(result[i][1]+result[i][2])
+
+    def pushChart(self):
+        repo = self.g.get_repo(f"{self.username}/{self.username}")
+        committer = InputGitAuthor('readme-bot', '41898282+github-actions[bot]@users.noreply.github.com')
+        with open('bar_graph.png', 'rb') as input_file:
+            data = input_file.read()
+        try:
+            contents = repo.get_contents("charts/bar_graph.png")
+            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)