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.
"""
def decorator(func):
def inner(*args, **kwargs):
attempt = 0
@@ -55,9 +54,7 @@ def retry(attempts=5, timeout=30):
time.sleep(sleep)
attempt += 1
return func(*args, **kwargs)
return inner
return decorator
@@ -97,7 +94,8 @@ def parse_into_list(values):
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:
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))
@retry
@retry(attempts=5, timeout=60)
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,
@@ -145,9 +143,10 @@ def open_pull_request(title, body, target, source, is_draft=False, can_modify=Tr
return response
@retry
@retry()
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:
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):
"""Given a response for an open or updated PR, set metadata
"""
Given a response for an open or updated PR, set metadata
Parameters:
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)
@retry
@retry(attempts=5, timeout=30)
def list_pull_requests(target, source):
"""Given a target and source, return a list of pull requests that match
(or simply exit given some kind of error code)
@@ -219,9 +219,10 @@ def list_pull_requests(target, source):
return response.json()
@retry
@retry(attempts=5, timeout=30)
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:
entry (dict) : the pull request metadata
@@ -250,7 +251,8 @@ def add_assignees(entry, assignees):
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).
Parameters:
@@ -264,7 +266,7 @@ def find_pull_request(listing, source):
return entry
@retry
@retry(attempts=5, timeout=30)
def find_default_branch():
"""
Find default branch for a repo (only called if branch not provided)
@@ -282,7 +284,7 @@ def find_default_branch():
return default_branch
@retry
@retry(attempts=5, timeout=30)
def add_reviewers(entry, reviewers, team_reviewers):
"""
Given regular or team reviewers, add them to a PR.