มีวิธีรู้ไหมว่าผู้ใช้เรียกใช้โปรแกรมจาก bash ได้อย่างไร?

ดูหมายเหตุที่ด้านล่างเกี่ยวกับสคริปต์ wrapper ที่เสนอในตอนแรก

แนวทางใหม่ที่ยืดหยุ่นมากขึ้นคือสคริปต์ python มีตัวเลือกบรรทัดคำสั่งใหม่ โดยอนุญาตให้ผู้ใช้สามารถระบุสตริงที่กำหนดเองที่พวกเขาต้องการเห็นในข้อความแสดงข้อผิดพลาด

ตัวอย่างเช่น หากผู้ใช้ต้องการเรียกสคริปต์หลาม 'myPyScript.py' ผ่านนามแฝง พวกเขาสามารถเปลี่ยนคำจำกัดความนามแฝงได้จากสิ่งนี้:

  alias myAlias='myPyScript.py $@'

สำหรับสิ่งนี้:

  alias myAlias='myPyScript.py --caller=myAlias $@'

หากพวกเขาต้องการเรียกสคริปต์ python จากเชลล์สคริปต์ ก็สามารถใช้ตัวเลือกบรรทัดคำสั่งเพิ่มเติมได้ดังนี้:

  #!/bin/bash
  exec myPyScript.py "$@" --caller=${0##*/}

การใช้งานแนวทางนี้ที่เป็นไปได้อื่นๆ:

  bash -c myPyScript.py --caller="bash -c myPyScript.py"

  myPyScript.py --caller=myPyScript.py

สำหรับการแสดงรายการบรรทัดคำสั่งแบบขยาย นี่คือสคริปต์ 'pyTest.py' ตามคำติชมของ @CharlesDuffy ซึ่งแสดงรายการ cmdline สำหรับสคริปต์ python ที่ทำงานอยู่ รวมถึงกระบวนการหลักที่สร้างสคริปต์นั้น หากใช้อาร์กิวเมนต์ -caller ใหม่ อาร์กิวเมนต์นั้นจะปรากฏในบรรทัดคำสั่ง แม้ว่านามแฝงจะถูกขยาย ฯลฯ

#!/usr/bin/env python

import os, re

with open ("/proc/self/stat", "r") as myfile:
  data = [x.strip() for x in str.split(myfile.readlines()[0],' ')]

pid = data[0]
ppid = data[3]

def commandLine(pid):
  with open ("/proc/"+pid+"/cmdline", "r") as myfile:
    return [x.strip() for x in str.split(myfile.readlines()[0],'\x00')][0:-1]

pid_cmdline = commandLine(pid)
ppid_cmdline = commandLine(ppid)

print "%r" % pid_cmdline
print "%r" % ppid_cmdline

หลังจากบันทึกสิ่งนี้ลงในไฟล์ชื่อ 'pytest.py' แล้วเรียกมันจากสคริปต์ทุบตีชื่อ 'pytest.sh' พร้อมอาร์กิวเมนต์ต่างๆ ผลลัพธ์ที่ได้คือ:

$ ./pytest.sh a b "c d" e
['python', './pytest.py']
['/bin/bash', './pytest.sh', 'a', 'b', 'c d', 'e']

หมายเหตุ: การวิพากษ์วิจารณ์สคริปต์ตัวตัดคำต้นฉบับ aliasTest.sh นั้นถูกต้อง แม้ว่าการมีอยู่ของนามแฝงที่กำหนดไว้ล่วงหน้าจะเป็นส่วนหนึ่งของข้อกำหนดของคำถาม และอาจสันนิษฐานได้ว่ามีอยู่ในสภาพแวดล้อมของผู้ใช้ ข้อเสนอก็กำหนดนามแฝง (สร้างความประทับใจที่ทำให้เข้าใจผิดว่าเป็นส่วนหนึ่งของคำแนะนำมากกว่าที่ระบุไว้ เป็นส่วนหนึ่งของสภาพแวดล้อมของผู้ใช้) และไม่ได้แสดงว่า wrapper จะสื่อสารกับสคริปต์ python ที่เรียกว่าอย่างไร ในทางปฏิบัติ ผู้ใช้จะต้องระบุแหล่งที่มาของ wrapper หรือกำหนดนามแฝงภายใน wrapper และสคริปต์ python จะต้องมอบหมายการพิมพ์ข้อความแสดงข้อผิดพลาดให้กับสคริปต์การโทรแบบกำหนดเองหลายสคริปต์ (ซึ่งมีข้อมูลการโทรอยู่) และไคลเอนต์ก็จะมี เพื่อเรียกสคริปต์ตัวตัดคำ การแก้ปัญหาเหล่านั้นนำไปสู่แนวทางที่ง่ายขึ้น ซึ่งสามารถขยายไปยังกรณีการใช้งานเพิ่มเติมจำนวนเท่าใดก็ได้

ต่อไปนี้เป็นสคริปต์ต้นฉบับเวอร์ชันที่สร้างความสับสนน้อยกว่าสำหรับการอ้างอิง:

#!/bin/bash
shopt -s expand_aliases
alias myAlias='myPyScript.py'

# called like this:
set -o history
myAlias $@
_EXITCODE=$?
CALL_HISTORY=( `history` )
_CALLING_MODE=${CALL_HISTORY[1]}

case "$_EXITCODE" in
0) # no error message required
  ;;
1)
  echo "customized error message #1 [$_CALLING_MODE]" 1>&2
  ;;
2)
  echo "customized error message #2 [$_CALLING_MODE]" 1>&2
  ;;
esac

นี่คือผลลัพธ์:

$ aliasTest.sh 1 2 3
['./myPyScript.py', '1', '2', '3']
customized error message #2 [myAlias]

ไม่มีทางที่จะแยกความแตกต่างระหว่างเวลาที่มีการระบุล่ามสำหรับสคริปต์อย่างชัดเจนในบรรทัดคำสั่ง และเมื่อระบบปฏิบัติการอนุมานจากบรรทัด hashbang

การพิสูจน์:

$ cat test.sh 
#!/usr/bin/env bash

ps -o command $$

$ bash ./test.sh 
COMMAND
bash ./test.sh

$ ./test.sh 
COMMAND
bash ./test.sh

การทำเช่นนี้จะป้องกันไม่ให้คุณตรวจพบความแตกต่างระหว่างสองกรณีแรกในรายการของคุณ

ฉันมั่นใจด้วยว่าไม่มีวิธีที่สมเหตุสมผลในการระบุวิธีอื่น ๆ (เป็นสื่อกลาง) ในการเรียกคำสั่ง

ฉันเห็นสองวิธีในการทำเช่นนี้:

• วิธีที่ง่ายที่สุด ตามที่ 3sky แนะนำ คือการแยกวิเคราะห์บรรทัดคำสั่งจากภายในสคริปต์ python argparse สามารถใช้เพื่อดำเนินการดังกล่าวได้อย่างน่าเชื่อถือ ใช้งานได้เฉพาะในกรณีที่คุณสามารถเปลี่ยนสคริปต์นั้นได้
• วิธีที่ซับซ้อนกว่า ทั่วไปกว่าเล็กน้อยและเกี่ยวข้องคือการเปลี่ยนไฟล์ปฏิบัติการ python บนระบบของคุณ

เนื่องจากตัวเลือกแรกได้รับการบันทึกไว้อย่างดี ต่อไปนี้เป็นรายละเอียดเพิ่มเติมเล็กน้อยเกี่ยวกับตัวเลือกที่สอง:

ไม่ว่าสคริปต์ของคุณจะถูกเรียกด้วยวิธีใดก็ตาม python จะถูกรัน เป้าหมายคือการแทนที่ไฟล์ปฏิบัติการ python ด้วยสคริปต์ที่ตรวจสอบว่า foo.py อยู่ในอาร์กิวเมนต์หรือไม่ และหากเป็นเช่นนั้น ให้ตรวจสอบว่า --bar อยู่ด้วยหรือไม่ ถ้าไม่เช่นนั้น ให้พิมพ์ข้อความแล้วส่งคืน

ในกรณีอื่น ๆ ให้ดำเนินการปฏิบัติการ python จริง

หวังว่าตอนนี้การรัน python จะเสร็จสิ้นโดยใช้ shebang ต่อไปนี้: #!/usr/bin/env python3 หรือ trough python foo.py แทนที่จะเป็นตัวแปร #!/usr/bin/python หรือ /usr/bin/python foo.py ด้วยวิธีนี้ คุณสามารถเปลี่ยนตัวแปร $PATH และเติมไดเร็กทอรีที่มี false python อยู่ข้างหน้าได้

ในอีกกรณีหนึ่ง คุณสามารถแทนที่ /usr/bin/python executable ได้ โดยมีความเสี่ยงที่การอัปเดตจะเล่นได้ไม่ดี

วิธีที่ซับซ้อนกว่าในการทำเช่นนี้อาจเป็นการใช้เนมสเปซและการเมานต์ แต่วิธีการข้างต้นก็อาจเพียงพอแล้ว โดยเฉพาะอย่างยิ่งหากคุณมีสิทธิ์ของผู้ดูแลระบบ

ตัวอย่างสิ่งที่สามารถใช้เป็นสคริปต์ได้:

#!/usr/bin/env bash

function checkbar
{
    for i in "$@"
    do
            if [ "$i" = "--bar" ]
            then
                    echo "Well done, you added --bar!"
                    return 0
            fi
    done
    return 1
}

command=$(basename ${1:-none})
if [ $command = "foo.py" ]
then
    if ! checkbar "$@"
    then
        echo "Please add --bar to the command line, like so:"
        printf "%q " $0
        printf "%q " "$@"
        printf -- "--bar\n"
        exit 1
    fi
fi
/path/to/real/python "$@"

อย่างไรก็ตาม หลังจากอ่านคำถามของคุณอีกครั้ง เห็นได้ชัดว่าฉันเข้าใจผิด ในความคิดของฉัน มันเป็นเรื่องปกติที่จะพิมพ์ "foo.py ต้องถูกเรียกว่า foo.py --bar", "โปรดเพิ่มแถบในอาร์กิวเมนต์ของคุณ" หรือ "โปรดลอง (แทน )" โดยไม่คำนึงว่า ผู้ใช้ป้อน:

• หากนั่นคือนามแฝง (git) นี่เป็นข้อผิดพลาดครั้งเดียว และผู้ใช้จะลองใช้นามแฝงของตนหลังจากสร้างมันขึ้นมา เพื่อให้พวกเขารู้ว่าจะต้องใส่ส่วน --bar ไว้ที่ไหน
• with either with /usr/bin/foo.py or python foo.py:
  • If the user is not really command line-savvy, they can just paste the working command that is displayed, even if they don't know the difference
  • หากเป็นเช่นนั้น พวกเขาควรจะสามารถเข้าใจข้อความได้โดยไม่มีปัญหา และปรับบรรทัดคำสั่งได้

ฉันรู้ว่ามันเป็นงาน bash แต่ฉันคิดว่าวิธีที่ง่ายที่สุดคือแก้ไข 'foo.py' แน่นอนว่ามันขึ้นอยู่กับระดับของสคริปต์ที่ซับซ้อน แต่บางทีมันอาจจะพอดีก็ได้ นี่คือโค้ดตัวอย่าง:

#!/usr/bin/python

import sys

if len(sys.argv) > 1 and sys.argv[1] == '--bar':
    print 'make magic'
else:
    print 'Please add the --bar option to your command, like so:'
    print '    python foo.py --bar'

ในกรณีนี้ ไม่ว่าผู้ใช้จะเรียกใช้โค้ดนี้อย่างไร

$ ./a.py
Please add the --bar option to your command, like so:
    python foo.py --bar

$ ./a.py -dua
Please add the --bar option to your command, like so:
    python foo.py --bar

$ ./a.py --bar
make magic

$ python a.py --t
Please add the --bar option to your command, like so:
    python foo.py --bar

$ /home/3sky/test/a.py
Please add the --bar option to your command, like so:
    python foo.py --bar

$ alias a='python a.py'
$ a
Please add the --bar option to your command, like so:
    python foo.py --bar

$ a --bar
make magic