unit test for token usage

Author: PeeachPie

ForgejoRepoAPI.py

@@ -4,9 +4,20 @@
 import isodate
 from pyforgejo import PyforgejoApi
 
-from interface_wrapper import (Branch, Comment, Commit, Contributor, Invite,
-                               IRepositoryAPI, Issue, PullRequest, Repository,
-                               User, WikiPage, logging)
+from interface_wrapper import (
+    Branch,
+    Comment,
+    Commit,
+    Contributor,
+    Invite,
+    IRepositoryAPI,
+    Issue,
+    PullRequest,
+    Repository,
+    User,
+    WikiPage,
+    logging,
+)
 
 
 class ForgejoRepoAPI(IRepositoryAPI):

GitHubRepoAPI.py

@@ -1,6 +1,17 @@
-from interface_wrapper import (Branch, Comment, Commit, Contributor, Invite,
-                               IRepositoryAPI, Issue, PullRequest, Repository,
-                               User, WikiPage, logging)
+from interface_wrapper import (
+    Branch,
+    Comment,
+    Commit,
+    Contributor,
+    Invite,
+    IRepositoryAPI,
+    Issue,
+    PullRequest,
+    Repository,
+    User,
+    WikiPage,
+    logging,
+)
 
 from github import Github, GithubException
 

contributors_parser.py

@@ -1,7 +1,6 @@
 from dataclasses import asdict, dataclass
 from time import sleep
 from typing import Generator
-from datetime import datetime
 
 from constants import EMPTY_FIELD, TIMEDELTA
 from interface_wrapper import IRepositoryAPI, Repository

git_logger.py

@@ -3,6 +3,7 @@
 from constants import TIMEDELTA
 from interface_wrapper import IRepositoryAPI, RepositoryFactory
 
+
 def login(source, token, base_url):
     client = RepositoryFactory.create_api(source, token, base_url)
     return client
@@ -21,6 +22,7 @@ def get_repos_from_file(repos_path: str) -> list[str]:
 
     return list_repos
 
+
 class Clients:
     def __init__(self, source: str, tokens: list[str], base_url: str | None = None):
         # Возможно это можно переписать покрасивее

main.py

@@ -12,6 +12,7 @@
 
 from utils import parse_time
 
+
 def parse_args():
     parser = argparse.ArgumentParser()
     parser.add_argument("--invites", help="print pending invites", action="store_true")
@@ -115,6 +116,7 @@ def parse_args():
                 action.required = True
     return parser.parse_args()
 
+
 def run(args, binded_repos, repos_for_wiki=None):
     start = parse_time(args.start.split('-'))
     finish = parse_time(args.finish.split('-'))
@@ -150,6 +152,7 @@ def run(args, binded_repos, repos_for_wiki=None):
             args.out, args.google_token, args.table_id, args.sheet_id
         )
 
+
 def main():
     args = parse_args()
 
@@ -171,5 +174,6 @@ def main():
     else:
         run(args, binded_repos)
 
+
 if __name__ == '__main__':
     main()

pull_requests_parser.py

@@ -12,6 +12,7 @@
 from interface_wrapper import IRepositoryAPI, Repository
 from utils import logger
 
+
 @dataclass(kw_only=True, frozen=True)
 class PullRequestData:
     repository_name: str = ''

test_token_usage.py

@@ -6,6 +6,7 @@
 
 import git_logger
 
+
 def parse_args(args):
     parser = argparse.ArgumentParser()
     parser.add_argument('--tt1', type=str, required=True, help='first test token')
@@ -23,6 +24,7 @@ def parse_args(args):
 
     return parser.parse_args(args)
 
+
 class TestTokenUsage(unittest.TestCase):
     def setUp(self):
         test_args = parse_args(sys.argv[1:])
@@ -53,16 +55,17 @@ def _get_rate_limit(clients: git_logger.Clients):
 
     @staticmethod
     def _is_only_one_token_used(limit_start, limit_finish):
-        return (bool(limit_start[0] - limit_finish[0])
-            != bool(limit_start[1] - limit_finish[1]))
-    
+        return bool(limit_start[0] - limit_finish[0]) != bool(
+            limit_start[1] - limit_finish[1]
+        )
+
     @staticmethod
     def _is_max_token_used(limit_start, limit_finish):
         if limit_start[0] - limit_finish[0]:
             return limit_start[0] == max(limit_start)
         else:
             return limit_start[1] == max(limit_start)
-    
+
     @staticmethod
     def _change_tokens_order(tokens, key):
         key %= len(tokens)
@@ -80,7 +83,9 @@ def _get_usage(self, binded_repos, clients):
     def test_commits_parser(self):
         self.args.commits = True
         for i in range(2):
-            clients = git_logger.Clients("github", self._change_tokens_order(self.tokens, i))
+            clients = git_logger.Clients(
+                "github", self._change_tokens_order(self.tokens, i)
+            )
             binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
 
             limit_start, limit_finish = self._get_usage(binded_repos, clients)
@@ -91,7 +96,9 @@ def test_commits_parser(self):
     def test_contributors_parser(self):
         self.args.contributors = True
         for i in range(2):
-            clients = git_logger.Clients("github", self._change_tokens_order(self.tokens, i))
+            clients = git_logger.Clients(
+                "github", self._change_tokens_order(self.tokens, i)
+            )
             binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
 
             limit_start, limit_finish = self._get_usage(binded_repos, clients)
@@ -102,7 +109,9 @@ def test_contributors_parser(self):
     def test_issues_parser(self):
         self.args.issues = True
         for i in range(2):
-            clients = git_logger.Clients("github", self._change_tokens_order(self.tokens, i))
+            clients = git_logger.Clients(
+                "github", self._change_tokens_order(self.tokens, i)
+            )
             binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
 
             limit_start, limit_finish = self._get_usage(binded_repos, clients)
@@ -113,7 +122,9 @@ def test_issues_parser(self):
     def test_invites_parser(self):
         self.args.invites = True
         for i in range(2):
-            clients = git_logger.Clients("github", self._change_tokens_order(self.tokens, i))
+            clients = git_logger.Clients(
+                "github", self._change_tokens_order(self.tokens, i)
+            )
             binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
 
             limit_start, limit_finish = self._get_usage(binded_repos, clients)
@@ -124,13 +135,16 @@ def test_invites_parser(self):
     def test_pull_requests_parser(self):
         self.args.pull_requests = True
         for i in range(2):
-            clients = git_logger.Clients("github", self._change_tokens_order(self.tokens, i))
+            clients = git_logger.Clients(
+                "github", self._change_tokens_order(self.tokens, i)
+            )
             binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
 
             limit_start, limit_finish = self._get_usage(binded_repos, clients)
 
             self.assertTrue(self._is_only_one_token_used(limit_start, limit_finish))
             self.assertTrue(self._is_max_token_used(limit_start, limit_finish))
 
+
 if __name__ == '__main__':
     unittest.main(argv=[sys.argv[0]])

utils.py

@@ -4,8 +4,9 @@
 
 from constants import MIN_SIDE_PADDING, SIDE_WHITE_SPACES, TITLE_LEN, TIMEZONE
 
+
 class logger:
-    #TODO: отключение вывода в stdout
+    # TODO: отключение вывода в stdout
     @staticmethod
     def log_title(title: str, title_len: int = TITLE_LEN):
         final_len = max(
@@ -46,6 +47,7 @@ def log_error(error: str):
     def log_warning(warning: str):
         pass
 
+
 def parse_time(datetime_str) -> datetime:
     start = (
         datetime_str[0].split('/') + datetime_str[1].split(':')

wikipars.py

@@ -1,4 +1,3 @@
-import csv
 import os
 import time
 
@@ -8,6 +7,7 @@
 
 from utils import logger
 
+
 def wikiparser(repositories: list[str], path_drepo: str, csv_name: str):
     logger.log_to_csv(csv_name, WIKI_FIELDNAMES)
 
@@ -87,7 +87,7 @@ def wikiparser(repositories: list[str], path_drepo: str, csv_name: str):
 
             for fieldname in data_commit:
                 print(fieldname, data_commit[fieldname], sep=': ')
-            
+
             logger.log_sep()
             logger.log_to_csv(csv_name, data_commit)