Apakah ada cara untuk mengetahui bagaimana pengguna menjalankan program dari bash?

Lihat Catatan di bagian bawah mengenai skrip pembungkus yang awalnya diusulkan.

Pendekatan baru yang lebih fleksibel adalah skrip python menyediakan opsi baris perintah baru, memungkinkan pengguna menentukan string khusus yang ingin mereka lihat dalam pesan kesalahan.

Misalnya, jika pengguna lebih suka memanggil skrip python 'myPyScript.py' melalui alias, mereka dapat mengubah definisi alias dari ini:

  alias myAlias='myPyScript.py $@'

untuk ini:

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

Jika mereka lebih suka memanggil skrip python dari skrip shell, ia dapat menggunakan opsi baris perintah tambahan seperti:

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

Kemungkinan penerapan lain dari pendekatan ini:

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

  myPyScript.py --caller=myPyScript.py

Untuk mencantumkan baris perintah yang diperluas, berikut skrip 'pyTest.py', berdasarkan masukan dari @CharlesDuffy, yang mencantumkan cmdline untuk skrip python yang sedang berjalan, serta proses induk yang memunculkannya. Jika argumen -caller baru digunakan, argumen tersebut akan muncul di baris perintah, meskipun alias akan diperluas, dll.

#!/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

Setelah menyimpannya ke file bernama 'pytest.py', dan kemudian memanggilnya dari skrip bash bernama 'pytest.sh' dengan berbagai argumen, inilah hasilnya:

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

CATATAN: kritik terhadap skrip pembungkus asli aliasTest.sh valid. Meskipun keberadaan alias yang telah ditentukan sebelumnya adalah bagian dari spesifikasi pertanyaan, dan dapat dianggap ada di lingkungan pengguna, proposal tersebut mendefinisikan alias tersebut (menciptakan kesan menyesatkan bahwa itu adalah bagian dari rekomendasi, bukan bagian yang ditentukan). bagian dari lingkungan pengguna), dan itu tidak menunjukkan bagaimana pembungkus akan berkomunikasi dengan skrip python yang disebut. Dalam praktiknya, pengguna harus mencari sumber pembungkus atau menentukan alias di dalam pembungkus, dan skrip python harus mendelegasikan pencetakan pesan kesalahan ke beberapa skrip panggilan khusus (tempat informasi panggilan berada), dan klien akan memilikinya. untuk memanggil skrip pembungkus. Pemecahan masalah tersebut menghasilkan pendekatan yang lebih sederhana, yang dapat diperluas ke sejumlah kasus penggunaan tambahan.

Berikut versi skrip asli yang tidak terlalu membingungkan, untuk referensi:

#!/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

Inilah hasilnya:

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

Tidak ada cara untuk membedakan antara saat penerjemah skrip ditentukan secara eksplisit pada baris perintah dan saat disimpulkan oleh OS dari baris hashbang.

Bukti:

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

ps -o command $$

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

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

Hal ini mencegah Anda mendeteksi perbedaan antara dua kasus pertama dalam daftar Anda.

Saya juga yakin bahwa tidak ada cara yang masuk akal untuk mengidentifikasi cara lain (yang dimediasi) dalam memanggil suatu perintah.

Saya dapat melihat dua cara untuk melakukan ini:

• Yang paling sederhana, seperti yang disarankan oleh 3sky, adalah mengurai baris perintah dari dalam skrip python. argparse dapat digunakan untuk melakukannya dengan cara yang andal. Ini hanya berfungsi jika Anda dapat mengubah skrip itu.
• Cara yang lebih rumit, sedikit lebih umum dan rumit, adalah dengan mengubah python yang dapat dieksekusi di sistem Anda.

Karena opsi pertama sudah didokumentasikan dengan baik, berikut sedikit rincian lebih lanjut tentang opsi kedua:

Terlepas dari cara skrip Anda dipanggil, python dijalankan. Tujuannya di sini adalah mengganti python yang dapat dieksekusi dengan skrip yang memeriksa apakah foo.py ada di antara argumen, dan jika ya, periksa apakah --bar juga ada. Jika tidak, cetak pesannya dan kembalikan.

Dalam setiap kasus lainnya, jalankan python asli yang dapat dieksekusi.

Sekarang, mudah-mudahan, menjalankan python dilakukan melalui shebang berikut: #!/usr/bin/env python3, atau melalui python foo.py, bukan varian #!/usr/bin/python atau /usr/bin/python foo.py. Dengan begitu, Anda dapat mengubah variabel $PATH, dan menambahkan direktori tempat python palsu Anda berada.

Dalam kasus lain, Anda dapat mengganti /usr/bin/python executable, dengan risiko tidak bermain baik dengan pembaruan.

Cara yang lebih kompleks untuk melakukan hal ini mungkin adalah dengan namespace dan mount, tetapi cara di atas mungkin sudah cukup, terutama jika Anda memiliki hak admin.

Contoh apa yang bisa berfungsi sebagai skrip:

#!/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 "$@"

Namun, setelah membaca kembali pertanyaan Anda, jelas saya salah memahaminya. Menurut pendapat saya, tidak apa-apa untuk mencetak "foo.py harus dipanggil seperti foo.py --bar", "harap tambahkan bilah ke argumen Anda" atau "silakan coba (daripada )", apa pun yang terjadi pengguna memasukkan:

• Jika itu adalah alias (git), ini adalah kesalahan satu kali, dan pengguna akan mencoba aliasnya setelah membuatnya, sehingga mereka tahu di mana harus meletakkan bagian --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
  • Jika ya, mereka seharusnya dapat memahami pesan tersebut tanpa kesulitan, dan menyesuaikan baris perintah mereka.

Saya tahu ini tugas bash, tapi menurut saya cara termudah adalah mengubah 'foo.py'. Tentu saja tergantung tingkat kerumitan skrip, tapi mungkin cocok. Berikut ini contoh kode:

#!/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'

Dalam hal ini, tidak masalah bagaimana pengguna menjalankan kode ini.

$ ./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