Question

基本上我试图检查一些用户输入。用户必须输入正确的日期

YYYYMMDD

，第二个参数必须是正确的系统

MIDRANGE_PROD或MIDRANGE_TEST

这是我的代码：

def check_user_args(user_args):
    print(user_args[1])
    if(len(user_args[1]) == 8):
        try:
            time.strptime(user_args[1], '%Y%M%d')
        except ValueError:
            print('Invalid date! ' + user_args[1])
            sys.exit(1)
    else:
        print('Invalid date!')
        sys.exit(1)
    print(user_args[2])
    if(user_args[2] != 'MIDRANGE_PROD' and user_args[2] != 'MIDRANGE_TEST'):
        print('Invalid system! ' + user_args[2])
        sys.exit(1)
    if(len(user_args) > 3):
        print('to many arguments')
        sys.exit(1)

这不是最优雅的方式吗？我虽然使用正则表达式，但在处理日期时这很快变得复杂。此外，我似乎无法让time.striptime功能正常工作，它仍然需要无效日期

D:\oppssup\old_job>python queue_cleaner_main.py 20151929 MIDRANGE_PROD
['queue_cleaner_main.py', '20151929', 'MIDRANGE_PROD']
20151929
MIDRANGE_PROD

更新

感谢所有回复。下面是我为python3实现的最终解决方案

# Check command line arguments are correct
def check_user_args(user_args):
    known_systems = {'prod', 'test'}
    try:
        date_string, system_string = sys.argv[1:] #check they only enter 2 arguments
    except ValueError:
        sys.exit('Usage: queue_cleaner <date> <system>')
    try:
        input_date = datetime.strptime(date_string, '%Y%m%d').date() #Check date format ok
    except ValueError:
        sys.exit('expected date in YYYYMMDD format, got: {}'.format(date_string)) #
    if system_string not in known_systems:
        sys.exit('wrong system, expected: {}, got: {}'.format(' or '
                 .join(known_systems), system_string)) # check correct system entered

Answer 1

您的时间格式不正确。您应该使用%m（月）而不是%M（分钟）。请参阅文档中的strftime() and strptime() behavior。

如果您想稍后使用这些数据，那么只需解析它并捕获例外：

#!/usr/bin/env python
import sys
from datetime import datetime

known_systems = {'MIDRANGE_PROD', 'MIDRANGE_TEST'}

try:
    date_string, system_string = sys.argv[1:]
except ValueError:
    sys.exit('Usage: queue_cleaner <date> <system>')

if system_string not in known_systems:
   sys.exit('wrong system, expected: %s, got: %r' % (
       ' or '.join(known_systems), system_string))

try:
    input_date = datetime.strptime(date_string, '%Y%m%d').date()
except ValueError:
    sys.exit('expected date in YYYYMMDD format, got: %r' % date_string)

如果要支持更复杂的命令行参数，例如，添加可选参数，请使用命令行解析器，例如argparse或docopt模块提供的。

Answer 2

这会有用吗？

import re
import datetime

def is_valid_date_str(date_str):
    try:
        datetime.date(*map(int, re.match(r"^(\d{4})(\d{2})(\d{2})$", date_str).groups()))
    catch ValueError, AttributeError:
        print('Invalid date! ' + user_args[1])
        sys.exit(1)

尝试中的一行做了三件事：

首先使用正则表达式
然后将捕获的值映射到整数，
然后尝试使用这些值来创建datetime.date。

如果正则表达式失败，匹配返回None，None.groups()会引发AttributeError。如果这些数字不是有效的年，月，日元组，则datetime.date(y,m,d)会提升ValueError。

time.strptime（）python3使用它的最佳方法？

2 个答案: