Merge pull request #22 from OSLL/8_assignee_problem

issue 8 assignee problem

Author: zmm

git_logger.py

@@ -1,8 +1,10 @@
-from github import Github, Repository, GithubException
 import csv
 
+from github import Github, Repository, GithubException, PullRequest
+
 EMPTY_FIELD = 'Empty field'
 
+
 def login(token):
     client = Github(login_or_token=token)
     try:
@@ -30,8 +32,27 @@ def get_next_repo(client: Github, repositories):
             yield repo
 
 
+def get_assignee_story(github_object):
+    assignee_result = ""
+    events = github_object.get_issue_events() if type(
+        github_object) is PullRequest.PullRequest else github_object.get_events()
+    for event in events:
+        if event.event == "assigned" or event.event == "unassigned":
+            date = event.created_at
+            if event.event == "assigned":
+                assigner = github_object.user.login
+                assignee = event.assignee.login
+                assignee_result += f"{date}: {assigner} -> {assignee}; "
+            else:
+                assigner = github_object.user.login
+                assignee = event.assignee.login
+                assignee_result += f"{date}: {assigner} -/> {assignee}; "
+    return assignee_result
+
+
 def log_commit_to_csv(info, csv_name):
-    fieldnames = ['repository name', 'commit id', 'author name', 'author login', 'author email', 'date and time', 'changed files']
+    fieldnames = ['repository name', 'commit id', 'author name', 'author login', 'author email', 'date and time',
+                  'changed files']
     with open(csv_name, 'a', newline='') as file:
         writer = csv.DictWriter(file, fieldnames=fieldnames)
         writer.writerow(info)
@@ -65,7 +86,8 @@ def log_repository_commits(repository: Repository, csv_name):
 def log_issue_to_csv(info, csv_name):
     fieldnames = ['repository name', 'number', 'title', 'state', 'task', 'created at', 'creator name', 'creator login',
                   'creator email', 'closer name', 'closer login', 'closer email', 'closed at', 'comment body',
-                  'comment created at', 'comment author name', 'comment author login', 'comment author email']
+                  'comment created at', 'comment author name', 'comment author login', 'comment author email',
+                  'assignee story', ]
     with open(csv_name, 'a', newline='') as file:
         writer = csv.DictWriter(file, fieldnames=fieldnames)
         writer.writerow(info)
@@ -93,8 +115,11 @@ def log_repository_issues(repository: Repository, csv_name):
             'comment author name': EMPTY_FIELD,
             'comment author login': EMPTY_FIELD,
             'comment author email': EMPTY_FIELD,
+            'assignee story': EMPTY_FIELD,
         }
 
+        info_tmp['assignee story'] = get_assignee_story(issue)
+
         if issue.user is not None:
             info_tmp['creator name'] = issue.user.name
             info_tmp['creator login'] = issue.user.login
@@ -123,7 +148,8 @@ def log_pr_to_csv(info, csv_name):
     fieldnames = ['repository name', 'title', 'state', 'commit into', 'commit from', 'created at', 'creator name',
                   'creator login', 'creator email',
                   'changed files', 'comment body', 'comment created at', 'comment author name', 'comment author login',
-                  'comment author email', 'merger name', 'merger login', 'merger email', 'source branch', 'target branch']
+                  'comment author email', 'merger name', 'merger login', 'merger email', 'source branch',
+                  'target branch', 'assignee story', ]
     with open(csv_name, 'a', newline='') as file:
         writer = csv.DictWriter(file, fieldnames=fieldnames)
         writer.writerow(info)
@@ -156,13 +182,16 @@ def log_repositories_pr(repository: Repository, csv_name):
             'merger email': EMPTY_FIELD,
             'source branch': pull.head.ref,
             'target branch': pull.base.ref,
+            'assignee story': EMPTY_FIELD,
         }
 
         if pull.merged_by is not None:
             info_tmp['merger name'] = pull.merged_by.name
             info_tmp['merger login'] = pull.merged_by.login
             info_tmp['merger email'] = pull.merged_by.email
 
+        info_tmp['assignee story'] = get_assignee_story(pull)
+
         if pull.get_comments().totalCount > 0:
             for comment in pull.get_comments():
                 info = info_tmp
@@ -203,6 +232,7 @@ def log_pull_requests(client: Github, repositories, csv_name):
                 'merger email',
                 'source branch',
                 'target branch',
+                'assignee story',
             )
         )
 
@@ -233,6 +263,7 @@ def log_issues(client: Github, repositories, csv_name):
                 'comment author name',
                 'comment author login',
                 'comment author email',
+                'assignee story',
             )
         )