типы пути к каталогу с argparse

Мой скрипт Python должен читать файлы из каталога, переданного в командной строке. Я определил читаемый тип каталога _, как показано ниже, для использования с argparse для проверки того, что каталог, переданный в командной строке, существует и доступен для чтения. Кроме того, значение по умолчанию (/tmp/non _существующий _каталог в приведенном ниже примере )также было указано для аргумента каталога. Проблема здесь в том, что argparse вызывает readable _dir ()со значением по умолчанию даже в ситуации, когда аргумент каталога явно передается в командной строке.Это приводит к тому, что сценарий вылетает из строя, так как путь по умолчанию /tmp/non_existent_dir не существует в контексте, где каталог явно передается в командной строке. Я мог бы обойти это, не указав значение по умолчанию и сделав этот аргумент обязательным, или отложив проверку на более позднее время в сценарии, но это более элегантное решение, о котором все знают?

#!/usr/bin/python
import argparse
import os

def readable_dir(prospective_dir):
  if not os.path.isdir(prospective_dir):
    raise Exception("readable_dir:{0} is not a valid path".format(prospective_dir))
  if os.access(prospective_dir, os.R_OK):
    return prospective_dir
  else:
    raise Exception("readable_dir:{0} is not a readable dir".format(prospective_dir))

parser = argparse.ArgumentParser(description='test', fromfile_prefix_chars="@")
parser.add_argument('-l', '--launch_directory', type=readable_dir, default='/tmp/non_existent_dir')
args = parser.parse_args()
55
задан iruvar 10 July 2012 в 03:35
поделиться