Compare commits
	
		
			3 Commits
		
	
	
		
			dev
			...
			f939127a0c
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| f939127a0c | |||
| 056779f7d2 | |||
| fec09edb88 | 
							
								
								
									
										26
									
								
								__init__.py
									
									
									
									
									
								
							
							
						
						
									
										26
									
								
								__init__.py
									
									
									
									
									
								
							@@ -16,8 +16,9 @@ API_SERVICE_NAME = 'calendar'
 | 
			
		||||
API_VERSION = 'v3'
 | 
			
		||||
 | 
			
		||||
class Calendar:
 | 
			
		||||
    def __init__(self, name, calendarId, toggle='False', color="#000000"):
 | 
			
		||||
    def __init__(self, name, calendarId, calType, toggle='False', color="#000000"):
 | 
			
		||||
        self.name = name
 | 
			
		||||
        self.calType = calType
 | 
			
		||||
        self.color = color
 | 
			
		||||
        self.toggle=toggle
 | 
			
		||||
        self.calendarId = calendarId
 | 
			
		||||
@@ -80,7 +81,8 @@ def updateCalendars(user, calendars, colors):
 | 
			
		||||
    for calendar in calendars:
 | 
			
		||||
        if not user.hasCalendar(calendar.calendarId):
 | 
			
		||||
            color = fromColorIdGetColor(calendar.color, colormap, colors)
 | 
			
		||||
            c = dbCalendar(calendar_id=calendar.calendarId, 
 | 
			
		||||
            c = dbCalendar(calendar_id = calendar.calendarId, 
 | 
			
		||||
                           calendar_type = calendar.calType,
 | 
			
		||||
                           name = calendar.name,
 | 
			
		||||
                           toggle = "False",
 | 
			
		||||
                           color = color)
 | 
			
		||||
@@ -96,8 +98,9 @@ def calendarsFromDb(user):
 | 
			
		||||
        calendarId = calendar.calendar_id
 | 
			
		||||
        toggle = calendar.toggle
 | 
			
		||||
        color = calendar.color
 | 
			
		||||
        calType = calendar.calendar_type
 | 
			
		||||
 | 
			
		||||
        pyCalendars.append(Calendar(name, calendarId, toggle, color))
 | 
			
		||||
        pyCalendars.append(Calendar(name, calendarId, calType, toggle, color))
 | 
			
		||||
 | 
			
		||||
    return pyCalendars
 | 
			
		||||
 | 
			
		||||
@@ -110,6 +113,7 @@ def purgeCalendars(calendars, visibleCalendars):
 | 
			
		||||
    return purged
 | 
			
		||||
 | 
			
		||||
# remaps a event color id to a calendar color id
 | 
			
		||||
# for google calendars
 | 
			
		||||
def toCalendarColorId(colormap, colorId):
 | 
			
		||||
    for remap in colormap['eventRemap']:
 | 
			
		||||
        if remap['from'] == colorId:
 | 
			
		||||
@@ -119,6 +123,7 @@ def toCalendarColorId(colormap, colorId):
 | 
			
		||||
    return colorId
 | 
			
		||||
 | 
			
		||||
# remaps a calendar color ID to its natural color
 | 
			
		||||
# for google calendars
 | 
			
		||||
def toNaturalColor(colormap, orgColor):
 | 
			
		||||
    for remap in colormap['colors']:
 | 
			
		||||
        if remap['api'] == orgColor:
 | 
			
		||||
@@ -127,8 +132,10 @@ def toNaturalColor(colormap, orgColor):
 | 
			
		||||
    print(f"failed with {orgColor}")
 | 
			
		||||
    return orgColor
 | 
			
		||||
 | 
			
		||||
# use the event color id to convert it to a hex color
 | 
			
		||||
def colorizeEvents(events, colors):
 | 
			
		||||
# uses the event color id to convert it to a hex color
 | 
			
		||||
# does this for every event in events
 | 
			
		||||
# this is used for Google Calendars
 | 
			
		||||
def colorizeGoogleEvents(events, colors):
 | 
			
		||||
    COLORS_FILE = os.path.join(os.path.dirname(__file__), 'colors.json')
 | 
			
		||||
    with open(COLORS_FILE) as f:
 | 
			
		||||
        colormap = json.load(f)
 | 
			
		||||
@@ -138,10 +145,12 @@ def colorizeEvents(events, colors):
 | 
			
		||||
            event.colorHex = forEventGetColor(event, colormap, colors)
 | 
			
		||||
 | 
			
		||||
# returns a color for a specific calendar color id
 | 
			
		||||
# for google calendars
 | 
			
		||||
def fromColorIdGetColor(color, colormap, colors):
 | 
			
		||||
    return toNaturalColor(colormap, colors['calendar'][color]['background'])
 | 
			
		||||
 | 
			
		||||
# generates the natural color for a event color id
 | 
			
		||||
# for google calendars
 | 
			
		||||
def forEventGetColor(event, colormap, colors):
 | 
			
		||||
    calColorId = toCalendarColorId(colormap, event.eventColorId)
 | 
			
		||||
    bg = colors['calendar'][calColorId]['background']     
 | 
			
		||||
@@ -181,17 +190,18 @@ def getTimeStamps():
 | 
			
		||||
 | 
			
		||||
    # define today and tomorrow
 | 
			
		||||
    now = datetime.datetime.now(datetime.timezone.utc).astimezone()
 | 
			
		||||
    today = now.replace(hour=0, minute=0, second = 0).isoformat() 
 | 
			
		||||
    today = now.replace(hour=0, minute=0, second = 0) 
 | 
			
		||||
    tomorrow = now.replace(hour=23, minute=59, second=59).isoformat() # + '+01:00'
 | 
			
		||||
    twodaysfromnow = datetime.datetime.today() + datetime.timedelta(days=2)
 | 
			
		||||
    twodaysfromnow = twodaysfromnow.replace(hour=23, minute=59, second=59).astimezone().isoformat()
 | 
			
		||||
    twodaysfromnow = twodaysfromnow.replace(hour=23, minute=59, second=59).astimezone()
 | 
			
		||||
    
 | 
			
		||||
    return today, twodaysfromnow
 | 
			
		||||
 | 
			
		||||
def generateJsonFromCalendarEntries(events, colors):
 | 
			
		||||
    
 | 
			
		||||
    # fix all colors in events that have a different event color
 | 
			
		||||
    colorizeEvents(events, colors)
 | 
			
		||||
    if colors != None:
 | 
			
		||||
        colorizeGoogleEvents(events, colors)
 | 
			
		||||
    # if not events:
 | 
			
		||||
    #    print('No upcoming events found.')
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										68
									
								
								icalHandler.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										68
									
								
								icalHandler.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,68 @@
 | 
			
		||||
from server import db
 | 
			
		||||
from database.models import Calendar as dbCalendar
 | 
			
		||||
from backend import Event as bEvent
 | 
			
		||||
from ics import Calendar as iCalendar
 | 
			
		||||
import requests
 | 
			
		||||
 | 
			
		||||
def icalToCalendarDb(url, name, user):
 | 
			
		||||
    try:
 | 
			
		||||
        c = iCalendar(requests.get(url).text)
 | 
			
		||||
    except:
 | 
			
		||||
        return False
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    c = dbCalendar(calendar_id = url,
 | 
			
		||||
                      calendar_type = 'ical',
 | 
			
		||||
                      name = name,
 | 
			
		||||
                      toggle = "False",
 | 
			
		||||
                      color = "#000000")
 | 
			
		||||
    db.session.add(c)
 | 
			
		||||
    user.calendars.append(c)
 | 
			
		||||
    db.session.commit()
 | 
			
		||||
 | 
			
		||||
def fetchCalendarEvents(calendars, startDate, endDate):
 | 
			
		||||
 | 
			
		||||
    all_events = []
 | 
			
		||||
    for calendar in calendars:
 | 
			
		||||
        if calendar.toggle == "True" and calendar.calendar_type == "ical":
 | 
			
		||||
            ical = iCalendar(requests.get(calendar.calendar_id).text)
 | 
			
		||||
            for event in ical.events:
 | 
			
		||||
                event = fitEventToCalendarTimezone(event, ical)
 | 
			
		||||
                name = event.name
 | 
			
		||||
                start = event.begin.format()
 | 
			
		||||
                end = event.end.format()
 | 
			
		||||
 | 
			
		||||
                newEvent = bEvent(name, start, end)
 | 
			
		||||
               
 | 
			
		||||
 | 
			
		||||
                if eventWithinStartEnd(startDate, endDate, event):
 | 
			
		||||
                    if not eventAllDay(event):
 | 
			
		||||
                    
 | 
			
		||||
                        print(start, flush=True)
 | 
			
		||||
                        all_events.append(newEvent) 
 | 
			
		||||
 | 
			
		||||
    return all_events
 | 
			
		||||
 | 
			
		||||
def fitEventToCalendarTimezone(event, calendar):
 | 
			
		||||
    event.begin = event.begin.to(next(iter(calendar._timezones)))
 | 
			
		||||
    event.end = event.end.to(next(iter(calendar._timezones)))
 | 
			
		||||
    return event
 | 
			
		||||
 | 
			
		||||
# converts everything to epoch seconds and utc
 | 
			
		||||
# and then checks if at either event start or event end
 | 
			
		||||
# is is within startDate and endDate
 | 
			
		||||
def eventWithinStartEnd(startDate, endDate, event):
 | 
			
		||||
    if((startDate.timestamp() <= event.begin.timestamp and
 | 
			
		||||
       event.begin.timestamp <= endDate.timestamp()) or
 | 
			
		||||
       startDate.timestamp() <= event.end.timestamp and
 | 
			
		||||
       event.end.timestamp <= endDate.timestamp()):
 | 
			
		||||
        return True
 | 
			
		||||
    return False
 | 
			
		||||
 | 
			
		||||
def eventAllDay(event):
 | 
			
		||||
    beginDay = event.begin.replace(hour=0, minute=0, second=0)
 | 
			
		||||
    endDay = event.end.replace(hour=0, minute=0, second=0)
 | 
			
		||||
    delta = endDay - beginDay
 | 
			
		||||
    if delta.days >= 2:
 | 
			
		||||
        return True
 | 
			
		||||
    return False
 | 
			
		||||
		Reference in New Issue
	
	Block a user