mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-07-10 08:38:07 +00:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import click
|
|
|
|
from github import update_cache, yield_recently_updated_gh_pr_branches
|
|
from gitlab import get_latest_infos_for_branch
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
pass
|
|
|
|
|
|
@cli.command(name="branch")
|
|
@click.argument("branch", default="master")
|
|
@click.option("--no-status", is_flag=True, default=False)
|
|
def get_branch(branch: str, no_status: bool):
|
|
print(f"Getting links for branch: {branch}")
|
|
tests_info = get_latest_infos_for_branch(branch, not no_status)
|
|
|
|
for name, info in tests_info.items():
|
|
print(
|
|
f"{name}\n - LINK: {info.link}\n - STATUS: {info.status}\n - DIFF SCREENS: {info.diff_screens}"
|
|
)
|
|
|
|
|
|
def do_update_pulls():
|
|
new_branch_infos = list(yield_recently_updated_gh_pr_branches())
|
|
print(80 * "*")
|
|
print(f"Found {len(new_branch_infos)} new branches")
|
|
for branch in new_branch_infos:
|
|
print(f"Getting links for branch: {branch}")
|
|
try:
|
|
tests_info = get_latest_infos_for_branch(branch.name, True)
|
|
branch.job_infos = tests_info
|
|
except Exception as e:
|
|
print(f"Failed to get links for branch: {branch.name}")
|
|
print(e)
|
|
|
|
branch_dict = {branch.name: branch for branch in new_branch_infos}
|
|
update_cache(branch_dict)
|
|
|
|
|
|
@cli.command(name="pulls")
|
|
def update_pulls():
|
|
do_update_pulls()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|