1 2 3 4 5 6 7 8 9 10 11 12 13 14
|
import datetime
def date_range(date_from, date_to, excluded_days=[]):
DATETIME_FORMAT = "%Y-%m-%d"
date_from = datetime.datetime.strptime(date_from, DATETIME_FORMAT)
date_to = datetime.datetime.strptime(date_to, DATETIME_FORMAT)
r = (date_to+datetime.timedelta(days=1)-date_from).days
return [date_from+datetime.timedelta(days=i) for i in range(r) if not (date_from+datetime.timedelta(days=i)).weekday() in excluded_days]
datelist = date_range("2011-01-01", "2011-01-31", [1])
print len(datelist), datelist
datelist = date_range("2011-01-01", "2011-01-31", [1, 2])
print len(datelist), datelist |
Partager