[Python Cheatsheet] 18. subprocess - 외부 프로세스 실행

1
2
3
4
5

import subprocess

# 기본 실행
result = subprocess.run(["echo", "Hello, World!"])
print(result.returncode)  # 0 (성공)

1
2
3
4
5
6
7
8
9

# 출력 캡처
result = subprocess.run(
    ["ls", "-la"],
    capture_output=True,  # stdout, stderr 캡처
    text=True,            # 문자열로 반환 (bytes 대신)
)
print(result.stdout)
print(result.stderr)
print(result.returncode)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

# check=True: 실패 시 예외
try:
    subprocess.run(
        ["ls", "nonexistent_file"],
        capture_output=True,
        text=True,
        check=True,  # returncode != 0이면 예외
    )
except subprocess.CalledProcessError as e:
    print(f"Command failed with code {e.returncode}")
    print(f"stderr: {e.stderr}")

1
2
3
4
5
6
7
8

# 타임아웃
try:
    subprocess.run(
        ["sleep", "10"],
        timeout=2,  # 2초 후 타임아웃
    )
except subprocess.TimeoutExpired:
    print("Command timed out!")

1
2
3
4
5
6
7
8

# 입력 전달 (stdin)
result = subprocess.run(
    ["cat"],
    input="Hello from Python\n",
    capture_output=True,
    text=True,
)
print(result.stdout)  # Hello from Python

1
2
3
4
5
6

result = subprocess.run(["echo", "test"], capture_output=True, text=True)

result.args        # ['echo', 'test']
result.returncode  # 0
result.stdout      # 'test\n'
result.stderr      # ''

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15

# shell=True: 셸을 통해 실행
# ⚠️ 보안 위험: 사용자 입력이 포함되면 인젝션 가능
result = subprocess.run(
    "echo $HOME",  # 문자열로 전달
    shell=True,
    capture_output=True,
    text=True,
)

# 🔴 위험한 예시 (절대 하지 말 것)
user_input = "file.txt; rm -rf /"  # 악의적 입력
subprocess.run(f"cat {user_input}", shell=True)  # 인젝션!

# ✅ 안전한 방법: 리스트로 전달
subprocess.run(["cat", user_input])  # 인자가 이스케이프됨

1
2
3
4
5
6
7
8
9

# 파이프: cmd1 | cmd2
from subprocess import Popen, PIPE

# ls | grep .py
p1 = Popen(["ls", "-la"], stdout=PIPE)
p2 = Popen(["grep", ".py"], stdin=p1.stdout, stdout=PIPE, text=True)
p1.stdout.close()  # SIGPIPE 허용
output = p2.communicate()[0]
print(output)

1
2
3
4
5
6
7

# shell=True로 파이프 (간단하지만 덜 안전)
result = subprocess.run(
    "ls -la | grep .py",
    shell=True,
    capture_output=True,
    text=True,
)

1
2
3
4
5
6
7
8

from subprocess import Popen, PIPE

# 실시간 출력 읽기
with Popen(["ping", "-c", "3", "google.com"], stdout=PIPE, text=True) as proc:
    for line in proc.stdout:
        print(f">> {line.strip()}")

print(f"Exit code: {proc.returncode}")

1
2
3
4
5
6
7
8
9

# communicate()로 입출력 처리
proc = Popen(
    ["python", "-c", "import sys; print(sys.stdin.read().upper())"],
    stdin=PIPE,
    stdout=PIPE,
    text=True,
)
stdout, stderr = proc.communicate(input="hello world")
print(stdout)  # HELLO WORLD

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

# Git 명령 실행
def git_commit(message: str) -> bool:
    try:
        subprocess.run(
            ["git", "commit", "-m", message],
            check=True,
            capture_output=True,
            text=True,
        )
        return True
    except subprocess.CalledProcessError as e:
        print(f"Git commit failed: {e.stderr}")
        return False

1
2
3
4
5
6
7

# 환경변수 설정
import os

env = os.environ.copy()
env["MY_VAR"] = "custom_value"

subprocess.run(["printenv", "MY_VAR"], env=env)

1
2
3
4
5

# 작업 디렉토리 변경
subprocess.run(
    ["ls", "-la"],
    cwd="/tmp",  # /tmp에서 실행
)

1
2
3
4
5
6

# os.system(): 레거시, 출력 캡처 불가
import os
os.system("echo hello")  # 사용 지양

# subprocess.run(): 권장
subprocess.run(["echo", "hello"])