Есть ли хороший способ обработки параметров командной строки в сценариях, которые вы можете импортировать позже? - PullRequest
Новая
       29

Есть ли хороший способ обработки параметров командной строки в сценариях, которые вы можете импортировать позже?

1 голос
/ 07 июня 2019

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

#!/usr/bin/python

def getArgs():
    parser = argparse.ArgumentParser()
    parser.add_argument('input',type=int)
    parser.add_argument('-v','--verbose',action='store_true')
    return parser.parse_args()

def main(input,verbose):
    result = calculation(input,verbose) 
    if verbose:
        print(str(input) + " squared is " + str(result))
    else:
        print(result)

def calculation(input,verbose):
    if verbose:
        print("Doing Calculation")
    result = input * input
    return result 

if __name__ == '__main__': #checks to see if this script is being executed directly, will not run if imported into another script
    import argparse
    args=getArgs()
    if args.verbose:
        print("You have enabled verbosity")
    main(args.input,args.verbose)

Вот несколько иллюстративных вариантов выполнения 

user@machine ~ $ ./whatever.py 7
49
user@machine ~ $ ./whatever.py -v 7
You have enabled verbosity
Doing Calculation
7 squared is 49
user@machine ~ $ python
Python 3.7.3 (default, Mar 26 2019, 21:43:19) 
[GCC 8.2.1 20181127] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import whatever
>>> whatever.main(7,False)
49
>>> whatever.main(7,True)
Doing Calculation
7 squared is 49

Этот сценарий работает, но я считаю, что есть более чистый способдля обработки параметров командной строки в случае, если вы импортируете сценарий позже, например, при принудительном использовании параметра по умолчанию.Я полагаю, что одним из вариантов было бы рассматривать параметр как глобальную переменную, но я все еще подозреваю, что есть менее подробный (каламбур) способ включить эти параметры в более поздние функции.

1 Ответ

1 голос
/ 07 июня 2019

Если у вас есть несколько функций, которые имеют общие параметры, поместите параметры в объект и рассмотрите возможность создания методов функций его типа: 

class Square:
  def __init__(self,v=False): self.verb=v
  def calculate(self,x):
    if self.verb: print(…)
    return x*x
  def main(self,x):
    if self.verb: print(…)
    y=self.calculate(x)
    print("%s squared is %s"%(x,y) if self.verb else y)

if __name__=="__main__":
  args=getArgs()
  Square(args.verbose).main(args.input)

(Значение по умолчанию False обычно соответствует требованиям API-клиента.)

...