v. 1.0b

code/script.py

@@ -1,6 +1,10 @@
 from datetime import datetime, time
 import icalendar
 
+from dateutil.relativedelta import *
+from dateutil.rrule import *
+from dateutil.parser import *
+
 
 def text_ical(user_id, tz):
     date = dt_now(tz)
@@ -14,15 +18,44 @@ def text_ical(user_id, tz):
     for i, component in enumerate(ecal.walk()):
         if component.name == "VEVENT":
 
-            dt_start = component.decoded("dtstart")
-            dt_start.replace(tzinfo=None)
-            dt_start = dt_start.astimezone(tz)
+            # НАЧАЛО
+            dt = icalendar.vDDDTypes.to_ical(component.get('dtstart')).decode('utf-8')
+            dt = dt.split("T")
 
+            dt_start = component.decoded("dtstart")
             dt_end = component.decoded("dtend")
-            dt_end.replace(tzinfo=None)
-            dt_end = dt_end.astimezone(tz)
-            
-            if date.date() == component.decoded("dtstart").date():
+
+            if len(dt) == 1:
+                dt_start = datetime.combine(dt_start, time(minute=1))
+                dt_end = datetime.combine(dt_end, time(minute=1))
+            else:
+                dt_start.replace(tzinfo=None)
+                dt_start = dt_start.astimezone(tz)
+
+                dt_end.replace(tzinfo=None)
+                dt_end = dt_end.astimezone(tz)
+
+            # НАСТРОЙКА
+            r_rule = component.get('rrule')
+            if r_rule:
+                list_rrule = icalendar.vRecur.to_ical(r_rule).decode('utf-8').split(';')
+
+                for i, elem in enumerate(list_rrule):
+                    if 'UNTIL' in elem:
+                        list_rrule.pop(i)
+
+                now_date = "".join(date.date().isoformat().split("-"))
+
+                until = "UNTIL=" + now_date + "T235900Z"
+                date_iso = "".join(dt_start.isoformat().split("-"))
+                time_iso = "".join(date_iso.split(":")).split("+")[0] + "Z"
+                list_rrule.append(until)
+
+                list_rrule = ";".join(list_rrule)
+                print(dt_start.date(), list(rrulestr(list_rrule, dtstart=parse(time_iso))))
+
+
+            if date.date() == dt_start.date():
                 org = component.get("organizer")
                 desc = component.get("description")
 
@@ -66,3 +99,50 @@ def delta_time(d_event, start, end):
 
 def dt_now(tz):
     return datetime.now(tz=tz)
+
+
+def recur_rule(create_date, now_date, rules):
+    freq = rules['FREQ'][0]
+    inter = rules['INTERVAL'][0]
+
+    if 'UNTIL' in rules.keys():
+        until = rules['UNTIL'][0]
+        until = until if type(until) == type(now_date) else until.date()
+
+        if until < now_date: return False
+
+    count_day = now_date - create_date
+
+    print(create_date, rules)
+    print(freq, inter)
+
+    if freq == 'DAILY':
+        if count_day.days % inter == 0: return True
+
+    elif freq == 'WEEKLY':
+        weekday = now_date.strftime('%a').upper()[:2]
+        days = rules['BYDAY']
+
+        if weekday in days:
+            week = count_day.days // 7  # получаем чётность недели
+            if week % inter == 0: return True
+
+    elif freq == 'MONTHLY':
+        if 'BYMONTHDAY' in rules.keys():
+            dates = list(rrule(MONTHLY, dtstart=create_date, interval=inter,
+                               bymonthday=rules['BYMONTHDAY'][0], until=now_date))
+            if now_date in dates: return True
+        else:
+            moment = rules['BYDAY'][0][:-2]
+            day = rules['BYDAY'][0][-2:]
+
+            dates = list(rrule(MONTHLY, dtstart=create_date, interval=inter,
+                               byweekday=day(moment), until=now_date))
+            if now_date in dates: return True
+
+    elif freq == 'YEARLY':
+        dates = list(rrule(YEARLY, dtstart=create_date, interval=inter, bymonth=rules['BYMONTH'][0],
+                           bymonthday=rules['BYMONTHDAY'][0], until=now_date))
+        if now_date in dates: return True
+
+    return False