FilesExpand file tree

main

/

util.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

32 lines (23 loc) · 846 Bytes

main

/

util.py

Copy path

Top

File metadata and controls

• Code
• Blame

32 lines (23 loc) · 846 Bytes

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

from __future__ import annotations

import subprocess

from typing import Any

class CalledProcessError(RuntimeError):

pass

def added_files() -> set[str]:

cmd = ('git', 'diff', '--staged', '--name-only', '--diff-filter=A')

return set(cmd_output(*cmd).splitlines())

def cmd_output(*cmd: str, retcode: int | None = 0, **kwargs: Any) -> str:

kwargs.setdefault('stdout', subprocess.PIPE)

kwargs.setdefault('stderr', subprocess.PIPE)

proc = subprocess.Popen(cmd, **kwargs)

stdout, stderr = proc.communicate()

stdout = stdout.decode()

if retcode is not None and proc.returncode != retcode:

raise CalledProcessError(cmd, retcode, proc.returncode, stdout, stderr)

return stdout

def zsplit(s: str) -> list[str]:

s = s.strip('\0')

if s:

return s.split('\0')

else:

return []