21 from logilab.common.date import strptime, todate, next_month

    22 

    26 # XXX should replace lgc.date.date_range implementation

    27 def date_range(begin, end, incday=None, incmonth=None):

    28     """yields each date between begin and end

    29     :param begin: the start date

    30     :param end: the end date

    31     :param incr: the step to use to iterate over dates. Default is

    32                  one day.

    33     :param include: None (means no exclusion) or a function taking a

    34                     date as parameter, and returning True if the date

    35                     should be included.

    36     """

    37     assert not (incday and incmonth)

    38     begin = todate(begin)

    39     end = todate(end)

    40     if incmonth:

    41         while begin < end:

    42             begin = next_month(begin, incmonth)

    43             yield begin

    44     else:

    45         if not incday:

    46             incr = ONEDAY

    47         else:

    48             incr = datetime.timedelta(incday)

    49         while begin <= end:

    50            yield begin

    51            begin += incr