Есть ли способ узнать, как пользователь вызвал программу из bash?

См. Примечание внизу относительно первоначально предложенного сценария оболочки.

Новый более гибкий подход заключается в том, что сценарий python предоставляет новый параметр командной строки, позволяющий пользователям указывать настраиваемую строку, которую они предпочли бы видеть в сообщениях об ошибках.

Например, если пользователь предпочитает вызывать скрипт 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' и последующего вызова его из сценария bash с именем 'pytest.sh' с различными аргументами, вот результат:

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

ПРИМЕЧАНИЕ: критика оригинального сценария оболочки aliasTest.sh была верной. Хотя наличие заранее определенного псевдонима является частью спецификации вопроса и может предполагаться, что оно существует в пользовательской среде, предложение определило псевдоним (создавая вводящее в заблуждение впечатление, что это часть рекомендации, а не указанная часть среды пользователя), и не было показано, как оболочка будет взаимодействовать с вызываемым скриптом python. На практике пользователь должен либо создать оболочку, либо определить псевдоним внутри оболочки, а сценарий 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]

Невозможно различить, когда интерпретатор для сценария явно указан в командной строке, и когда он выводится ОС из строки хэшбэга.

Доказательство:

$ 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 выполняется с помощью следующего шебанга: #!/usr/bin/env python3 или python foo.py, а не варианта #!/usr/bin/python или /usr/bin/python foo.py. Таким образом, вы можете изменить переменную $PATH и добавить каталог, в котором находится ваш ложный 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