FilesExpand file tree

main

/

ci_utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

59 lines (52 loc) · 1.59 KB

main

/

ci_utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

59 lines (52 loc) · 1.59 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

import os

import subprocess

import sys

from contextlib import contextmanager

from pathlib import Path

from typing import Dict, List, Optional, Union

@contextmanager

def group(name: str):

"""Group output in GitHub Actions."""

# Only print group markers if running in GitHub Actions or if forced

# But for now, we'll always print them as they are harmless in local terminals

print(f"::group::{name}")

try:

yield

finally:

print("::endgroup::")

def run(

cmd: List[str],

env: Optional[Dict[str, str]] = None,

cwd: Optional[Union[str, Path]] = None,

check: bool = True,

capture_output: bool = False,

) -> subprocess.CompletedProcess:

"""Run a command with optional grouping and error handling."""

cmd_str = " ".join(str(c) for c in cmd)

# Don't group if capturing output, as it's likely an internal check

should_group = not capture_output

if should_group:

print(f"::group::{cmd_str}")

try:

run_env = os.environ.copy()

if env:

run_env.update(env)

result = subprocess.run(

cmd,

env=run_env,

cwd=cwd,

check=check,

text=True,

capture_output=capture_output

)

return result

except subprocess.CalledProcessError as e:

if should_group:

print(f"Command failed with exit code {e.returncode}")

raise

finally:

if should_group:

print("::endgroup::")

def fail(msg: str) -> None:

"""Exit with an error message."""

sys.exit(f"Error: {msg}")