Python-date-range-generator über Werktage

Ich versuche zu schaffen, eine generator-Funktion zum Durchlaufen Werktagen (Wochentage), wobei Wochenenden (und Feiertagen wäre schön, auch!). Bisher habe ich nur eine Funktion, die einfach iteriert über Tage:

def daterange(startDate, endDate):
    for i in xrange(int((endDate - startDate).days)):
        yield startDate + timedelta(i)

Ich bin kämpfen, um herauszufinden, eine saubere, effiziente und pythonic Weg, um den generator zu überspringen, an Wochenenden und Feiertagen. Vielen Dank im Voraus!

InformationsquelleAutor jh314 | 2012-07-18
  1. 25

    Ich würde stark empfehlen, die dateutil Bibliothek für solche Aufgaben. Eine grundlegende (nicht ignorieren Feiertagen) iterator über Werktagen, dann einfach ist:

    from dateutil.rrule import DAILY, rrule, MO, TU, WE, TH, FR

def daterange(start_date, end_date):
  return rrule(DAILY, dtstart=start_date, until=end_date, byweekday=(MO,TU,WE,TH,FR))
    • Schönes Beispiel! +1
    • Tut Sie dies tatsächlich tun, was Sie sagen? Ich versuchte es auf Python 2.7 und 3.3, die auf Linux und Mac OS, und in allen Fällen gibt es an allen Tagen, einschließlich Wochenenden. Wenn man sich dateutil.rrule.WDAYMASK Sie können sehen, dass es eine Liste von 0-6, d.h. alle Tage, nicht nur Montag-Freitag.
    • Recht, WDAYMASK ist in der Tat falsch (zumindest mit aktuellen Versionen von dateutil). Ich aktualisierte die Antwort, um diese zu reflektieren.
    • verwenden from dateutil.rrule import * sonst erhalten Sie Fehler wie MO nicht identifiziert
    • dateutil unterstützt keine Iteration rückwärts, nur für den Fall, Sie haben einen Anwendungsfall für diese. Anscheinend ist der Autor glaubt nicht, dass dies wichtig. bugs.launchpad.net/dateutil/+bug/903925
    InformationsquelleAutor earl
  2. 10

    Vorausgesetzt startDate und endDate einem datetime-oder date-Objekte, die Sie verwenden können, die Wochentag Methode um den Tag der Woche, dann überspringen Sie es, wenn es Samstag oder Sonntag. Nur:

    def daterange(startDate, endDate):
    for i in xrange(int((endDate - startDate).days)):
        nextDate = startDate + timedelta(i)
        if nextDate.weekday() not in (5, 6):
            yield startDate + timedelta(i)

    Für die Feiertage, die Sie haben zu prüfen Sie manuell, für jeden Urlaub, den Sie wollen. Einige Feiertage definiert sind, die in komplexer Weise, so könnte dies ein wenig schwierig sein.

    InformationsquelleAutor BrenBarn
  3. 7

    Es ist eine nützliche Bibliothek namens dateutil tun können diese Art der Sache für Sie. Es können sowohl Bereiche der Daten (oder Daten basierend auf benutzerdefinierten Regeln), außer an bestimmten Tagen, sollten Sie eine Woche ab, an einem Tag, etc... hat Auch eine etwas flexibler timedelta als die builtin-datetime-Bibliothek.

    Docs an http://labix.org/python-dateutil/ - und auf PyPi

    InformationsquelleAutor Jon Clements
  4. 1
    def get_date_range(start, end, workdays=None, holidays=None, skip_non_workdays=True):
"""
This function calculates the durations between 2 dates skipping non workdays and holidays as specified
:rtype : tuple
:param start: date
:param end: date
:param workdays: string [Comma Separated Values, 0 - Monday through to 6 - Sunday e.g "0,1,2,3,4"]
:param holidays: list
:param skip_non_workdays: boolean
:return:
"""
from datetime import timedelta

duration = 0

# define workdays
if workdays is None:
    workdays = [0, 1, 2, 3, 4]
else:
    workdays = workdays.split(",")

# check if we need to skip non workdays
if skip_non_workdays is False:
    workdays = [0, 1, 2, 3, 4, 5, 6]

# validate dates
if end < start:
    return False, "End date is before start date"

# now its time for us to iterate
i = start
while i <= end:

    # first let's give benefit of the doubt
    incr = True

    # lets see if day is in the workday array if not then fault it's existence here
    try:
        workdays.index(i.weekday())
    except ValueError:
        incr = False

    # lets check if day is an holiday, charge guilty if so.
    # We are checking the index in holiday array
    try:
        holidays.index(i)
        incr = False
    except (ValueError, AttributeError):
        pass

    if incr:
        duration += 1
        print "This day passed the criterion %s" % i

    i += timedelta(1)

return True, duration
    InformationsquelleAutor Peter Bob Ukonu
Schreibe einen Kommentar