Add repository language metric

Signed-off-by: CreativeNick <[email protected]>
DSACMS · Aug 5, 2024 · 8b4132a · 8b4132a
1 parent 538a8b2
commit 8b4132a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/scripts/metricsLib/metrics_definitions.py b/scripts/metricsLib/metrics_definitions.py
@@ -23,6 +23,9 @@
 # Metrics that save a resource to a file
 RESOURCE_METRICS = []
 
+# Predominant Languages Endpoint (ex. https://api.github.com/repos/chaoss/augur/languages)
+LANGUAGE_ENDPOINT = "https://api.github.com/repos/chaoss/augur/languages"
+
 REPO_GITHUB_GRAPHQL_QUERY = """
 query ($repo: String!, $owner: String!) {
   repository(name: $repo, owner: $owner) {
@@ -116,6 +119,16 @@
                                  {"total_project_blank_lines": ["blank_lines"],
                                  "average_blank_lines": ["avg_blank_lines"]}))
 
+SIMPLE_METRICS.append(ListMetric("repositoryLanguages", 
+                                 ["owner", "repo"], 
+                                 LANGUAGE_ENDPOINT, 
+                                 {"languages": None}, 
+                                 token=TOKEN))
+
+SIMPLE_METRICS.append(GraphQLMetric("githubGraphqlSimpleCounts", ["repo", "owner"],
+                                    REPO_GITHUB_GRAPHQL_QUERY,
+                                    github_graphql_simple_counts_metric_map, token=TOKEN))
+
 REPOMETRICS_ENDPOINT = "https://raw.githubusercontent.com/{owner}/{repo}/main/code.json"
 repometrics_values = {"project_type": "project_type", "user_input": "user_input", "project_fisma_level": "project_fisma_level", 
                       "group": "group", "subset_in_healthcare": "subset_in_healthcare", "user_type": "user_type",