python - 如何将日期传递给 Python 中的脚本？

Question

我有一个用于删除早于某个日期的图像的脚本。

我可以在调用运行脚本时将此日期作为参数传递吗？

示例:此脚本 delete_images.py 删除早于某个日期 (YYYY-MM-DD) 的图像

python delete_images.py 2010-12-31

脚本(使用固定日期(xDate 变量))

import os, glob, time

root = '/home/master/files/' # one specific folder
#root = 'D:\\Vacation\\*'          # or all the subfolders too
# expiration date in the format YYYY-MM-DD

### I have to pass the date from the script ###
xDate = '2010-12-31' 

print '-'*50
for folder in glob.glob(root):
    print folder
    # here .jpg image files, but could be .txt files or whatever
    for image in glob.glob(folder + '/*.jpg'):
        # retrieves the stats for the current jpeg image file
        # the tuple element at index 8 is the last-modified-date
        stats = os.stat(image)
        # put the two dates into matching format    
        lastmodDate = time.localtime(stats[8])
        expDate = time.strptime(xDate, '%Y-%m-%d')
        print image, time.strftime("%m/%d/%y", lastmodDate)
        # check if image-last-modified-date is outdated
        if  expDate > lastmodDate:
            try:
                print 'Removing', image, time.strftime("(older than %m/%d/%y)", expDate)
                os.remove(image)  # commented out for testing
            except OSError:
                print 'Could not remove', image 

Answer 1

快速但粗暴的方法是使用sys.argv。

import sys
xDate = sys.argv[1]

一种更健壮、可扩展的方法是使用 argparse模块:

import argparse

parser=argparse.ArgumentParser()
parser.add_argument('xDate')
args=parser.parse_args()

然后要访问用户提供的值，您将使用 args.xDate 而不是 xDate。

使用 argparse 模块，您可以在用户输入时自动免费获得帮助消息

delete_images.py -h

如果用户未能提供正确的输入，它还会提供有用的错误消息。

您还可以轻松地为 xDate 设置默认值，将 xDate 转换为 datetime.date 对象，并且，正如他们所说在电视上，“很多很多!”。

---

稍后我会在您使用的脚本中看到

expDate = time.strptime(xDate, '%Y-%m-%d')

将 xDate 字符串转换为时间元组。您可以使用 argparse 执行此操作，因此 args.xDate 自动成为时间元组。例如，

import argparse
import time

def mkdate(datestr):
    return time.strptime(datestr, '%Y-%m-%d')
parser=argparse.ArgumentParser()
parser.add_argument('xDate',type=mkdate)
args=parser.parse_args()
print(args.xDate)

当这样运行时:

% test.py 2000-1-1

产量

time.struct_time(tm_year=2000, tm_mon=1, tm_mday=1, tm_hour=0, tm_min=0, tm_sec=0, tm_wday=5, tm_yday=1, tm_isdst=-1)

---

附言。无论您选择使用什么方法(sys.argv 或 argparse)，拉

都是个好主意

expDate = time.strptime(xDate, '%Y-%m-%d')

在 for 循环 之外。由于 xDate 的值永远不会改变，您只需计算一次 expDate。