Mã lai trong các kịch bản shell. Chia sẻ biến

Câu trả lời này thảo luận về cách chạy đoạn mã Python nhiều dòng từ dòng lệnh trong thiết bị đầu cuối. Tôi nhận thấy rằng câu trả lời hoạt động rất tốt trong các kịch bản shell, ngay cả với thụt lề lồng nhau, rất hay, ví dụ:

#!/bin/bash
some_text="Hello world"
echo $some_text

cat <<EOF | python -
import sys;
for r in range(3):
  print r
  for a in range(2):
    print "hello"
EOF

in:

0 
hello
hello
1
hello
hello
2
hello
hello

Tuy nhiên, tôi gặp khó khăn khi chia sẻ các biến giữa tập lệnh shell và đoạn mã Python.

1. Làm cách nào tôi có thể thu thập đầu ra của chỉ mục python trong tập lệnh bash? (ví dụ trong một biến như $output).

2. Làm cách nào tôi có thể chuyển một biến bash (ví dụ $some_text) cho tập lệnh Python?

Bắt một biến thành Python

Vì (khi EOFđiểm đánh dấu không được trích dẫn) thay thế biến xảy ra trước khi văn bản được chuyển từ pythonđầu vào tiêu chuẩn sang đầu vào tiêu chuẩn, bạn có thể ném biến ngay trong tập lệnh.

python - <<EOF
some_text = "$some_text"
EOF

Nếu some_textlà test, trăn sẽ thấy some_text = "test". Tuy nhiên, lưu ý rằng nó có thể được coi là một lỗ hổng tiêm mã. Nếu some_textlà "; import os; os.system("evil-command"); x = ", ví dụ, python sẽ thấy:

some_text = ""; import os; os.system("evil-command"); x = ""

và chạy lệnh ác đó.

Nếu bạn muốn có thể kéo mã Python của mình ngay vào tập lệnh mà không cần sửa đổi gì, bạn có thể xuất biến của mình.

export some_text

và sử dụng os.environđể lấy nó.

some_text = os.environ['some_text']

Đó là một cách tiếp cận an toàn hơn / an toàn hơn.

Nhận đầu ra từ Python

Bạn có thể sử dụng thay thế lệnh để thu thập đầu ra của tập lệnh.

output=$(
python - <<EOF
import sys;
for r in range(3):
  print r
  for a in range(2):
    print "hello"
EOF
)

(lưu ý rằng tất cả các ký tự dòng mới được loại bỏ)

Vấn đề với cách tiếp cận của bạn là tập lệnh python nhúng không còn có quyền truy cập vào stdin gốc (vì stdin của nó là ... chính nó).

Nếu đó là một vấn đề bạn có thể viết:

python -c '
import sys;
for r in range(3):
  print r
  for a in range(2):
    print "hello"
'

Hoặc nếu tập lệnh python có thể chứa dấu ngoặc đơn:

python -c "$(cat << 'EOF'
import sys;
for r in range(3):
  print r
  for a in range(2):
    print "hello"
EOF
)"

Hoặc là:

python <(cat << 'EOF'
import sys;
for r in range(3):
  print r
  for a in range(2):
    print "hello"
EOF
)

Sử dụng dấu gạch ngang làm tên tệp:

ruby - a b <<'END'
puts ARGV.join(",")
END

python - a b <<'END'
import sys
print ",".join(sys.argv[1:])
END

Tôi không biết liệu sys.argv[1:]có đúng cách để làm điều này trong Python không. Đối với -e / -c, bạn có thể chỉ định kết thúc đối số bằng -:

set -- -a -b -c
ruby -e 'puts ARGV.join(",")' -- "$@"
python -c 'import sys; print ",".join(sys.argv[2:])' -- "$@"

Chụp đầu ra và chuyển hướng STDERR:

x=$(ruby <<'END' 2> /dev/null
puts "a"
abort "b"
END
)

1) Bạn có thể viết các phép gán biến cho một tệp bằng python và sau đó lấy nguồn đó trong tập lệnh bash của bạn.

2) Vì từ của bạn (EOF) không được trích dẫn, tất cả các dòng của tài liệu ở đây phải chịu sự mở rộng tham số. Bạn có thể sử dụng điều này để truyền công cụ cho tập lệnh python.