issue_15_done (time filter)

Author: Toropigeon

git_logger.py

@@ -41,8 +41,10 @@ def log_commit_to_stdout(info):
     print(info)
 
 
-def log_repository_commits(repository: Repository, csv_name):
+def log_repository_commits(repository: Repository, csv_name, start, finish):
     for commit in repository.get_commits():
+        if commit.commit.author.date < start or commit.commit.author.date > finish:
+            continue
         if commit.commit is not None:
             info = {'repository name': repository.full_name,
                     'commit id': commit.commit.sha,
@@ -75,8 +77,10 @@ def log_issue_to_stdout(info):
     print(info)
 
 
-def log_repository_issues(repository: Repository, csv_name):
+def log_repository_issues(repository: Repository, csv_name, start, finish):
     for issue in repository.get_issues(state='all'):
+        if issue.created_at < start or issue.created_at > finish:
+            continue
         info_tmp = {
             'repository name': repository.full_name, 'number': issue.number, 'title': issue.title,
             'state': issue.state, 'task': issue.body,
@@ -133,8 +137,10 @@ def log_pr_to_stdout(info):
     print(info)
 
 
-def log_repositories_pr(repository: Repository, csv_name):
+def log_repositories_pr(repository: Repository, csv_name, start, finish):
     for pull in repository.get_pulls(state='all'):
+        if pull.created_at < start or pull.created_at > finish:
+            continue
         info_tmp = {
             'repository name': repository.full_name,
             'title': pull.title,
@@ -178,7 +184,7 @@ def log_repositories_pr(repository: Repository, csv_name):
             log_pr_to_stdout(info_tmp)
 
 
-def log_pull_requests(client: Github, repositories, csv_name):
+def log_pull_requests(client: Github, repositories, csv_name, start, finish):
     with open(csv_name, 'w', newline='') as file:
         writer = csv.writer(file)
         writer.writerow(
@@ -207,10 +213,10 @@ def log_pull_requests(client: Github, repositories, csv_name):
         )
 
     for repo in get_next_repo(client, repositories):
-        log_repositories_pr(repo, csv_name)
+        log_repositories_pr(repo, csv_name, start, finish)
 
 
-def log_issues(client: Github, repositories, csv_name):
+def log_issues(client: Github, repositories, csv_name, start, finish):
     with open(csv_name, 'w', newline='') as file:
         writer = csv.writer(file)
         writer.writerow(
@@ -235,10 +241,10 @@ def log_issues(client: Github, repositories, csv_name):
         )
 
     for repo in get_next_repo(client, repositories):
-        log_repository_issues(repo, csv_name)
+        log_repository_issues(repo, csv_name, start, finish)
 
 
-def log_commits(client: Github, repositories, csv_name):
+def log_commits(client: Github, repositories, csv_name, start, finish):
     with open(csv_name, 'w', newline='') as file:
         writer = csv.writer(file)
         writer.writerow(
@@ -254,4 +260,4 @@ def log_commits(client: Github, repositories, csv_name):
         )
 
     for repo in get_next_repo(client, repositories):
-        log_repository_commits(repo, csv_name)
+        log_repository_commits(repo, csv_name, start, finish)

main.py

@@ -1,4 +1,6 @@
 import argparse
+from datetime import datetime
+
 import git_logger
 
 
@@ -9,9 +11,20 @@ def parse_args():
     parser.add_argument('-t', '--token', type=str, required=True, help='token github account')
     parser.add_argument('-l', '--list', type=str, required=True, help='repos names file')
     parser.add_argument('-o', '--out', type=str, required=True, help='output filename')
+    parser.add_argument('-s', '--start', type=str, required=False, help='start time', default='2000/01/01-00:00:00')
+    parser.add_argument('-f', '--finish', type=str, required=False, help='finish time', default='2400/01/01-00:00:00')
     return parser.parse_args()
 
 
+def parse_time(datetime_str):
+    start = datetime_str[0].split('/') + datetime_str[1].split(':') if len(datetime_str) == 2 \
+        else datetime_str[0].split('/') + ['00', '00', '00']
+    start = [int(i) for i in start]
+    start_datetime = datetime(year=start[0], month=start[1], day=start[2], hour=start[3], minute=start[4],
+                              second=start[5])
+    return start_datetime
+
+
 def main():
     args = parse_args()
     token = args.token
@@ -23,12 +36,16 @@ def main():
     except Exception as e:
         print(e)
     else:
+        if args.start:
+            start = parse_time(args.start.split('-'))
+        if args.finish:
+            finish = parse_time(args.finish.split('-'))
         if not args.p and not args.i:
-            git_logger.log_commits(client, repositories, csv_name)
+            git_logger.log_commits(client, repositories, csv_name, start, finish)
         if args.p:
-            git_logger.log_pull_requests(client, repositories, csv_name)
+            git_logger.log_pull_requests(client, repositories, csv_name, start, finish)
         if args.i:
-            git_logger.log_issues(client, repositories, csv_name)
+            git_logger.log_issues(client, repositories, csv_name, start, finish)
 
 
 if __name__ == '__main__':