discord-results.py (2178B)
1 #!/usr/bin/env python3 2 3 import argparse 4 import os 5 import re 6 import sys 7 from pathlib import Path 8 from discord_webhook import DiscordWebhook, DiscordEmbed 9 10 parser = argparse.ArgumentParser(prog='discord-results.py', description='Sends a Discord webhook notification at the end of a CI run.') 11 parser.add_argument('-b', '--branch') 12 parser.add_argument('-k', '--keymap') 13 parser.add_argument('-u', '--url') 14 parser.add_argument('-s', '--sha') 15 args = parser.parse_args() 16 17 qmk_dir = Path(__file__).resolve().parents[2].resolve() 18 19 keyboard_re = re.compile(r'CI Metadata: KEYBOARD=(.*)$', re.MULTILINE) 20 keymap_re = re.compile(r'CI Metadata: KEYMAP=(.*)$', re.MULTILINE) 21 22 successful_builds = sum([len(list(qmk_dir.glob(f'*.{extension}'))) for extension in ['uf2', 'bin', 'hex']]) 23 failures = list(sorted([f.resolve() for f in (qmk_dir / '.build/').glob('failed.log.*')])) 24 failed_builds = [] 25 for f in failures: 26 with open(f) as fh: 27 data = fh.read() 28 kb = keyboard_re.search(data).group(1) 29 km = keymap_re.search(data).group(1) 30 failed_builds.append(f'{kb}:{km}') 31 32 webhook = DiscordWebhook(url=os.getenv('DISCORD_WEBHOOK'), username="QMK GitHub CI") 33 if len(failed_builds) > 0: 34 failstr = '' 35 for f in failed_builds: 36 if len(failstr) >= 1800: 37 failstr += '<<snip>>' 38 break 39 failstr += f'{f}\n' 40 41 embed = DiscordEmbed(title=f':infinity: CI Build Failure ({args.branch}, {args.keymap})', description=f'**{successful_builds}** builds succeeded, **{len(failed_builds)}** builds failed:```{failstr}```', color='ff9999') 42 else: 43 embed = DiscordEmbed(title=f':infinity: CI Build Success ({args.branch}, {args.keymap})', description=f'**{successful_builds}** builds succeeded.', color='99ff99') 44 45 embed.add_embed_field(name='Build Target', value=f'[**{args.branch}**](https://github.com/qmk/qmk_firmware/tree/{args.branch}) / **{args.keymap}** keymap') 46 embed.add_embed_field(name='Workflow Run', value=f'[**Link**]({args.url})') 47 embed.add_embed_field(name='Firmware Binaries', value=f'[**ci.qmk.fm**](https://ci.qmk.fm/{args.branch}/{args.sha}/index.html)') 48 embed.set_timestamp() 49 50 webhook.add_embed(embed) 51 webhook.execute()