python - 遍历除周末以外的日期

Question

所以我有一个脚本，它有不同函数的日期参数，我希望它从 01-01-2012 到 06-09-2012 循环，不包括周末.我试图找出一种可以使用时间增量的方法，因为我的脚本输出的文件带有文件名中使用的日期，例如:

items = (functions.getItems(item,date)
    print items
    test = sum(abs(l[-1]) for l in items)
    total = open('total' +str(datetime.today- datetime.timedelta(1)),'a')

我希望 timedelta(1) 循环遍历每个日期，以便输出文件在第一天的格式为 total2012-01-01 并循环直到它创建文件 total2012-06-09。此外，项目的日期参数格式为 MM-DD-YYYY

我认为我可以做到这一点:

sd = 01-01-2012
ed = 06-09-2012
delta = datetime.timedelta(days=1)
diff = 0
while sd != ed
    # do functions 
    # (have output files (datetime.today - datetime.delta(diff))
    diff +=1
    sd+=delta

所以基本上我只是想弄清楚如何让函数以 01-01-2012 开始并以 06-10-2012 结束循环不包括周末。我无法弄清楚如何排除周末以及如何让它以正确的顺序循环

谢谢

Answer 1

使用 datetime.weekday()方法。它返回与工作日相关的 0 到 6 之间的值。周六值为5，周日值为6；因此，如果您在这些值出现时跳过操作，您将跳过周末:

start = datetime(2012, 1, 1)
end = datetime(2012, 10, 6)
delta = timedelta(days=1)
d = start
diff = 0
weekend = set([5, 6])
while d <= end:
    if d.weekday() not in weekend:
        diff += 1
    d += delta