def birthday_between(self, start=None, end=None): context = setup_proxy_roles(("Manager")) try: f = StringIO.StringIO((self.context.bursdag)) file = f.read() csv_reader = csv.reader(file.splitlines(), encoding="latin-1", delimiter=";") bursdager = [] start = "01.01" end = "12.12" if "start" in self.request: start = self.request.get("start") if "end" in self.request: end = self.request.get("end") st = datetime.strptime(start, "%d.%m") en = datetime.strptime(end, "%d.%m") for i in csv_reader: bdate = i[4][0:5] mydate = datetime.strptime(bdate, "%d.%m") if st <= mydate <= en: bursdager.append(i) return bursdager finally: cleanup_proxy_roles(context)
def has_birthday(self, dato=None): daymonth = self.date() context = setup_proxy_roles(("Manager")) try: # Read the CSV file f = StringIO.StringIO((self.context.bursdag)) file = f.read() csv_reader = csv.reader(file.splitlines(), encoding="latin-1", delimiter=";") bursdager = [] saturday = [] sunday = [] if "dato" in self.request: daymonth = self.request.get("dato") for i in csv_reader: if i[4][0:5] == daymonth: bursdager.append(i) return bursdager, saturday, sunday for i in csv_reader: if i[4][0:5] == daymonth: bursdager.append(i) if self.friday(): if i[4][0:5] == self.tomorrow(): saturday.append(i) if i[4][0:5] == self.aftertomorrow(): sunday.append(i) return bursdager, saturday, sunday finally: cleanup_proxy_roles(context)