lint: tidying up code in block tasks
This commit is contained in:
parent
a0da4d4641
commit
ac5a604587
8 changed files with 86 additions and 49 deletions
|
@ -11,21 +11,28 @@ from app.terraform import BaseAutomation
|
|||
|
||||
|
||||
class BlockBridgeGitHubAutomation(BaseAutomation):
|
||||
"""
|
||||
Automation task to import bridge reachability results from GitHub.
|
||||
"""
|
||||
|
||||
short_name = "block_bridge_github"
|
||||
description = "Import bridge reachability results from GitHub"
|
||||
frequency = 30
|
||||
|
||||
def automate(self, full: bool = False) -> Tuple[bool, str]:
|
||||
g = Github(app.config['GITHUB_API_KEY'])
|
||||
repo = g.get_repo(app.config['GITHUB_BRIDGE_REPO'])
|
||||
for vp in app.config['GITHUB_BRIDGE_VANTAGE_POINTS']:
|
||||
contents = repo.get_contents(f"recentResult_{vp}")
|
||||
github = Github(app.config['GITHUB_API_KEY'])
|
||||
repo = github.get_repo(app.config['GITHUB_BRIDGE_REPO'])
|
||||
for vantage_point in app.config['GITHUB_BRIDGE_VANTAGE_POINTS']:
|
||||
contents = repo.get_contents(f"recentResult_{vantage_point}")
|
||||
if isinstance(contents, list):
|
||||
return False, f"Expected a file at recentResult_{vp} but got a directory."
|
||||
return (False,
|
||||
f"Expected a file at recentResult_{vantage_point}"
|
||||
" but got a directory.")
|
||||
results = contents.decoded_content.decode('utf-8').splitlines()
|
||||
for result in results:
|
||||
parts = result.split("\t")
|
||||
if isoparse(parts[2]) < (datetime.datetime.now(datetime.timezone.utc) - datetime.timedelta(days=3)):
|
||||
if isoparse(parts[2]) < (datetime.datetime.now(datetime.timezone.utc)
|
||||
- datetime.timedelta(days=3)):
|
||||
continue
|
||||
if int(parts[1]) < 40:
|
||||
bridge: Bridge = Bridge.query.filter(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue