Update close_issue.py

.github/close_issue.py

@@ -18,24 +18,19 @@ def get_stargazers(repo):
             'per_page': 100,
             'page': page,
         }
-        url = 'https://api.github.com/repos/{}/stargazers?'.format(repo)
-
+        url = f'https://api.github.com/repos/{repo}/stargazers'
         resp = requests.get(url, headers=headers, params=queries)
         if resp.status_code != 200:
             raise Exception('Error get stargazers: ' + resp.text)
-
         data = resp.json()
         if not data:
             break
-
         for stargazer in data:
             _stargazers[stargazer['login']] = True
         page += 1
-
     print('list stargazers done, total: ' + str(len(_stargazers)))
     return _stargazers
 
-
 def get_issues(repo):
     page = 1
     _issues = []
@@ -47,25 +42,20 @@ def get_issues(repo):
             'per_page': 100,
             'page': page,
         }
-        url = 'https://api.github.com/repos/{}/issues?'.format(repo)
-
+        url = f'https://api.github.com/repos/{repo}/issues'
         resp = requests.get(url, headers=headers, params=queries)
         if resp.status_code != 200:
             raise Exception('Error get issues: ' + resp.text)
-
         data = resp.json()
         if not data:
             break
-
         _issues += data
         page += 1
-
     print('list issues done, total: ' + str(len(_issues)))
     return _issues
 
-
 def close_issue(repo, issue_number):
-    url = 'https://api.github.com/repos/{}/issues/{}'.format(repo, issue_number)
+    url = f'https://api.github.com/repos/{repo}/issues/{issue_number}'
     data = {
         'state': 'closed',
         'state_reason': 'not_planned',
@@ -74,31 +64,29 @@ def close_issue(repo, issue_number):
     resp = requests.patch(url, headers=headers, json=data)
     if resp.status_code != 200:
         raise Exception('Error close issue: ' + resp.text)
-
     print('issue: {} closed'.format(issue_number))
 
-
 def lock_issue(repo, issue_number):
-    url = 'https://api.github.com/repos/{}/issues/{}/lock'.format(repo, issue_number)
+    url = f'https://api.github.com/repos/{repo}/issues/{issue_number}/lock'
     data = {
         'lock_reason': 'spam',
     }
     resp = requests.put(url, headers=headers, json=data)
     if resp.status_code != 204:
         raise Exception('Error lock issue: ' + resp.text)
-
     print('issue: {} locked'.format(issue_number))
 
-
 if '__main__' == __name__:
-    stargazers = get_stargazers(github_repo)
-
-    issues = get_issues(github_repo)
-    for issue in issues:
-        login = issue['user']['login']
-        if login not in stargazers:
-            print('issue: {}, login: {} not in stargazers'.format(issue['number'], login))
-            close_issue(github_repo, issue['number'])
-            lock_issue(github_repo, issue['number'])
-
-    print('done')
+    try:
+        stargazers = get_stargazers(github_repo)
+        issues = get_issues(github_repo)
+        for issue in issues:
+            login = issue['user']['login']
+            if login not in stargazers:
+                print('issue: {}, login: {} not in stargazers'.format(issue['number'], login))
+                close_issue(github_repo, issue['number'])
+                lock_issue(github_repo, issue['number'])
+        print('done')
+    except Exception as e:
+        print(f"Error occurred: {str(e)}")
+        raise