Argparse与函数的交互

时间:2019-05-22 03:03:40

标签: python python-3.x argparse

因此,对于我的任务,我必须使用argparse解析.cvs文件并对其进行计算。我很难弄清楚该怎么做。

我已经完成了必需的三个功能,但是我不知道如何将它们正确地合并到我们给出的框架代码中。

请参见在测试我的程序时运行配置。将使用适当的参数进行编辑,并指定.cvs文件。到目前为止,我所做的工作是通过手工测试cvs文件,而不是使用argparse对其进行适当调用。我所做的工作甚至没有利用argparse。

我的代码中只有argparse的唯一原因是因为它被抛给我们时几乎没有解释,我们必须与之交互才能获得正确的结果。事实证明,这对我来说非常困难(且压力很大),因此,朝正确方向的推动表示赞赏-我真的很想了解这里的逻辑。

我们给出的基本代码包括我的示例中Grand_total函数以外的所有内容:

import argparse


def Grand_total(filepath):
  """
  Calculates the total amount of avocado sales given data from  a .csv file.
  """
  avo_price = open(filepath,"r")
  avo_price.readline() # to skip the header 
  lines_list=avo_price.readlines()
  sum = 0.0
  for line in lines_list:
    columns = line.split(",")
    prices= float(columns[2])
    volume = float(columns[3])
    total = (round((prices * volume),2))
    price_list = []
    price_list.append(total)
    for num in price_list:
      sum = sum + num
  print ("Total Sales:",(sum))



def parse_args():
    """
    parse_args takes no input and returns an Namespace dictionary describing the arguments selected
    by the user.  If invalid arguments are selected, the function will print an error message and quit.
    :return: Namespace with arguments to use
    """

    parser = argparse.ArgumentParser("Read CSV input about avocados and print total amount sold")
    parser.add_argument('--input', '-i', dest='input', required=True, type=str,
                       help='input CSV file')

    parser.add_argument('--group_by_region', '-r', dest='group_by_region', action='store_true', default=False,
                       help='Calculate results per region (default: calculate for all regions)')
    parser.add_argument('--organic', '-o', dest='organic', action='store_true', default=False,
                       help='Only calculate for organic avocados (default: calculate for conventional and organic)')

    return parser.parse_args()


def main():
    """
    The main body of the program.  It parses arguments, and performs calculations on a CSV
    """

    # get arguments entered by users
    args = parse_args()

    # TODO remove these print statements
    # This code is provided as an example for how to interpret the results of parse_args()
    print("Input file:      {}".format(args.Grand_total))
    print("Group by region: {}".format(args.city_total))
    print("Only organic:    {}".format(args.organic_total))


if __name__ == "__main__":
    main()

如果我运行此命令(包括其他功能), 我得到:

AttributeError: 'Namespace' object has no attribute 'Grand_total'

1 个答案:

答案 0 :(得分:1)

除其他事项外,argparse的作用是将命令行参数收集到一个Namespace对象中,您可以轻松访问该对象。 如果一切顺利,args将看起来像这样:

Namespace(input='my_csv.csv', group_by_region=False, organic=False)

然后,您将使用args.input访问输入路径。

如何重组程序的示例:

def main():
    args = parse_args()  # Note: you don't actually need a separate function for this
    path = args.input
    if args.group_by_region:
        region_output = #  calculate for each region here
        # do something with region output

    else:
        Grand_total(path)

我昨天实际上使用argparse做了一个快速项目;您可以看看它here;请注意,参数存储在separate file中。