Skip to content

Commit

Permalink
Merge pull request #32 from wegank/trivial-3
Browse files Browse the repository at this point in the history
Add support for reading token from gh
  • Loading branch information
Artturin authored Oct 28, 2024
2 parents 3180716 + 9c47756 commit 3a1484d
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions src/label_approved/cli.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import argparse
import logging
import os
import shutil
import subprocess
import sys
from dataclasses import dataclass
from datetime import datetime
Expand All @@ -17,11 +19,15 @@ def ghtoken() -> Optional[str]:
for env_key in ("INPUT_GITHUB_TOKEN", "GITHUB_BOT_TOKEN", "GITHUB_TOKEN"):
token = os.getenv(env_key)
if token:
break
else:
print("need a github token")
sys.exit(1)
return token
return token

if shutil.which("gh"):
r = subprocess.run(["gh", "auth", "token"], stdout=subprocess.PIPE, text=True)
if r.returncode == 0:
return r.stdout.strip()

print("need a github token")
sys.exit(1)


label_dict: dict[int, str] = {
Expand Down

0 comments on commit 3a1484d

Please sign in to comment.