Public/pull-request-action
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

customize args for decorator

Browse Source 
Signed-off-by: vsoch <vsoch@users.noreply.github.com>
This commit is contained in:
vsoch
2023-02-14 18:13:51 -07:00
parent b57393d6ea
commit 89160a5742
1 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
pull-request.py
View File

@@ -42,7 +42,6 @@ def retry(attempts=5, timeout=30):
We assume rate limiting is in minutes, so we set timeout to 30. We assume rate limiting is in minutes, so we set timeout to 30.
""" """
def decorator(func): def decorator(func):
def inner(*args, **kwargs): def inner(*args, **kwargs):
attempt = 0 attempt = 0
@@ -55,9 +54,7 @@ def retry(attempts=5, timeout=30):
time.sleep(sleep) time.sleep(sleep)
attempt += 1 attempt += 1
return func(*args, **kwargs) return func(*args, **kwargs)
return inner return inner
return decorator return decorator
@@ -97,7 +94,8 @@ def parse_into_list(values):
def set_env_and_output(name, value): def set_env_and_output(name, value):
"""helper function to echo a key/value pair to the environement file """
Helper function to echo a key/value pair to the environement file
Parameters: Parameters:
name (str) : the name of the environment variable name (str) : the name of the environment variable
@@ -111,7 +109,7 @@ def set_env_and_output(name, value):
environment_file.write("%s=%s\n" % (name, value)) environment_file.write("%s=%s\n" % (name, value))
@retry @retry(attempts=5, timeout=60)
def open_pull_request(title, body, target, source, is_draft=False, can_modify=True): def open_pull_request(title, body, target, source, is_draft=False, can_modify=True):
""" """
Open pull request opens a pull request with a given body and content, Open pull request opens a pull request with a given body and content,
@@ -145,9 +143,10 @@ def open_pull_request(title, body, target, source, is_draft=False, can_modify=Tr
return response return response
@retry @retry()
def update_pull_request(entry, title, body, target, state=None): def update_pull_request(entry, title, body, target, state=None):
"""Given an existing pull request, update it. """
Given an existing pull request, update it.
Parameters: Parameters:
entry (dict) : the pull request metadata entry (dict) : the pull request metadata
@@ -175,7 +174,8 @@ def update_pull_request(entry, title, body, target, state=None):
def set_pull_request_groups(response): def set_pull_request_groups(response):
"""Given a response for an open or updated PR, set metadata """
Given a response for an open or updated PR, set metadata
Parameters: Parameters:
response (requests.Response) : a requests response, unparsed response (requests.Response) : a requests response, unparsed
@@ -196,7 +196,7 @@ def set_pull_request_groups(response):
set_env_and_output("PULL_REQUEST_URL", html_url) set_env_and_output("PULL_REQUEST_URL", html_url)
@retry @retry(attempts=5, timeout=30)
def list_pull_requests(target, source): def list_pull_requests(target, source):
"""Given a target and source, return a list of pull requests that match """Given a target and source, return a list of pull requests that match
(or simply exit given some kind of error code) (or simply exit given some kind of error code)
@@ -219,9 +219,10 @@ def list_pull_requests(target, source):
return response.json() return response.json()
@retry @retry(attempts=5, timeout=30)
def add_assignees(entry, assignees): def add_assignees(entry, assignees):
"""Given a pull request metadata (from create or update) add assignees """
Given a pull request metadata (from create or update) add assignees
Parameters: Parameters:
entry (dict) : the pull request metadata entry (dict) : the pull request metadata
@@ -250,7 +251,8 @@ def add_assignees(entry, assignees):
def find_pull_request(listing, source): def find_pull_request(listing, source):
"""Given a listing and a source, find a pull request based on the source """
Given a listing and a source, find a pull request based on the source
(the branch name). (the branch name).
Parameters: Parameters:
@@ -264,7 +266,7 @@ def find_pull_request(listing, source):
return entry return entry
@retry @retry(attempts=5, timeout=30)
def find_default_branch(): def find_default_branch():
""" """
Find default branch for a repo (only called if branch not provided) Find default branch for a repo (only called if branch not provided)
@@ -282,7 +284,7 @@ def find_default_branch():
return default_branch return default_branch
@retry @retry(attempts=5, timeout=30)
def add_reviewers(entry, reviewers, team_reviewers): def add_reviewers(entry, reviewers, team_reviewers):
""" """
Given regular or team reviewers, add them to a PR. Given regular or team reviewers, add them to a PR.