epicyon/happening.py

396 lines
14 KiB
Python
Raw Normal View History

2020-04-03 11:53:31 +00:00
__filename__ = "happening.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.1.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@freedombone.net"
__status__ = "Production"
import os
from datetime import datetime
2020-02-22 19:13:55 +00:00
from utils import loadJson
from utils import locatePost
from utils import daysInMonth
from utils import mergeDicts
2020-04-03 11:53:31 +00:00
2020-02-24 11:10:48 +00:00
def isHappeningEvent(tag: {}) -> bool:
"""Is this tag an Event or Place ActivityStreams type?
"""
if not tag.get('type'):
return False
2020-04-03 11:53:31 +00:00
if tag['type'] != 'Event' and tag['type'] != 'Place':
2020-02-24 11:10:48 +00:00
return False
return True
2020-04-03 11:53:31 +00:00
2020-02-24 11:10:48 +00:00
def isHappeningPost(postJsonObject: {}) -> bool:
"""Is this a post with tags?
"""
if not postJsonObject:
return False
if not postJsonObject.get('object'):
return False
if not isinstance(postJsonObject['object'], dict):
return False
if not postJsonObject['object'].get('tag'):
return False
return True
2020-04-03 11:53:31 +00:00
def getTodaysEvents(baseDir: str, nickname: str, domain: str,
currYear=None, currMonthNumber=None,
2020-02-24 10:55:49 +00:00
currDayOfMonth=None) -> {}:
"""Retrieves calendar events for today
Returns a dictionary of lists containing Event and Place activities
"""
2020-04-03 11:53:31 +00:00
now = datetime.now()
if not currYear:
2020-04-03 11:53:31 +00:00
year = now.year
else:
2020-04-03 11:53:31 +00:00
year = currYear
if not currMonthNumber:
2020-04-03 11:53:31 +00:00
monthNumber = now.month
else:
2020-04-03 11:53:31 +00:00
monthNumber = currMonthNumber
if not currDayOfMonth:
2020-04-03 11:53:31 +00:00
dayNumber = now.day
else:
2020-04-03 11:53:31 +00:00
dayNumber = currDayOfMonth
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
calendarFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
events = {}
if not os.path.isfile(calendarFilename):
return events
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
calendarPostIds = []
recreateEventsFile = False
with open(calendarFilename, 'r') as eventsFile:
for postId in eventsFile:
2020-04-03 11:53:31 +00:00
postId = postId.replace('\n', '')
postFilename = locatePost(baseDir, nickname, domain, postId)
if not postFilename:
2020-04-03 11:53:31 +00:00
recreateEventsFile = True
2020-02-23 09:49:10 +00:00
continue
2020-04-03 11:53:31 +00:00
postJsonObject = loadJson(postFilename)
2020-02-24 11:10:48 +00:00
if not isHappeningPost(postJsonObject):
2020-02-23 09:49:10 +00:00
continue
2020-04-03 11:53:31 +00:00
postEvent = []
dayOfMonth = None
2020-02-23 09:49:10 +00:00
for tag in postJsonObject['object']['tag']:
2020-02-24 11:10:48 +00:00
if not isHappeningEvent(tag):
continue
2020-02-24 11:10:48 +00:00
# this tag is an event or a place
2020-04-03 11:53:31 +00:00
if tag['type'] == 'Event':
2020-02-23 09:49:10 +00:00
# tag is an event
if not tag.get('startTime'):
continue
2020-04-03 11:53:31 +00:00
eventTime = \
datetime.strptime(tag['startTime'],
2020-02-23 09:49:10 +00:00
"%Y-%m-%dT%H:%M:%S%z")
2020-04-03 11:53:31 +00:00
if int(eventTime.strftime("%Y")) == year and \
int(eventTime.strftime("%m")) == monthNumber and \
int(eventTime.strftime("%d")) == dayNumber:
dayOfMonth = str(int(eventTime.strftime("%d")))
2020-02-23 11:25:16 +00:00
if '#statuses#' in postId:
2020-02-24 10:55:49 +00:00
# link to the id so that the event can be
# easily deleted
2020-04-03 11:53:31 +00:00
tag['postId'] = postId.split('#statuses#')[1]
postEvent.append(tag)
2020-02-23 09:49:10 +00:00
else:
# tag is a place
postEvent.append(tag)
if postEvent and dayOfMonth:
calendarPostIds.append(postId)
if not events.get(dayOfMonth):
2020-04-03 11:53:31 +00:00
events[dayOfMonth] = []
2020-02-23 09:49:10 +00:00
events[dayOfMonth].append(postEvent)
# if some posts have been deleted then regenerate the calendar file
if recreateEventsFile:
2020-04-03 11:53:31 +00:00
calendarFile = open(calendarFilename, "w")
for postId in calendarPostIds:
calendarFile.write(postId+'\n')
calendarFile.close()
return events
2020-04-03 11:53:31 +00:00
def todaysEventsCheck(baseDir: str, nickname: str, domain: str) -> bool:
"""Are there calendar events today?
"""
2020-04-03 11:53:31 +00:00
now = datetime.now()
year = now.year
monthNumber = now.month
dayNumber = now.day
calendarFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
if not os.path.isfile(calendarFilename):
return False
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
eventsExist = False
with open(calendarFilename, 'r') as eventsFile:
for postId in eventsFile:
2020-04-03 11:53:31 +00:00
postId = postId.replace('\n', '')
postFilename = locatePost(baseDir, nickname, domain, postId)
2020-02-23 09:49:10 +00:00
if not postFilename:
continue
2020-04-03 11:53:31 +00:00
postJsonObject = loadJson(postFilename)
2020-02-24 11:10:48 +00:00
if not isHappeningPost(postJsonObject):
2020-02-23 09:49:10 +00:00
continue
for tag in postJsonObject['object']['tag']:
2020-02-24 11:10:48 +00:00
if not isHappeningEvent(tag):
continue
2020-02-24 11:10:48 +00:00
# this tag is an event or a place
2020-04-03 11:53:31 +00:00
if tag['type'] != 'Event':
continue
2020-02-24 11:10:48 +00:00
# tag is an event
if not tag.get('startTime'):
continue
2020-04-03 11:53:31 +00:00
eventTime = \
datetime.strptime(tag['startTime'],
2020-02-24 11:10:48 +00:00
"%Y-%m-%dT%H:%M:%S%z")
2020-04-03 11:53:31 +00:00
if int(eventTime.strftime("%Y")) == year and \
int(eventTime.strftime("%m")) == monthNumber and \
int(eventTime.strftime("%d")) == dayNumber:
eventsExist = True
2020-02-24 11:10:48 +00:00
break
return eventsExist
2020-04-03 11:53:31 +00:00
def thisWeeksEventsCheck(baseDir: str, nickname: str, domain: str) -> bool:
"""Are there calendar events this week?
"""
2020-04-03 11:53:31 +00:00
now = datetime.now()
year = now.year
monthNumber = now.month
dayNumber = now.day
calendarFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
if not os.path.isfile(calendarFilename):
return False
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
eventsExist = False
with open(calendarFilename, 'r') as eventsFile:
for postId in eventsFile:
2020-04-03 11:53:31 +00:00
postId = postId.replace('\n', '')
postFilename = locatePost(baseDir, nickname, domain, postId)
2020-02-23 09:49:10 +00:00
if not postFilename:
continue
2020-04-03 11:53:31 +00:00
postJsonObject = loadJson(postFilename)
2020-02-24 11:10:48 +00:00
if not isHappeningPost(postJsonObject):
2020-02-23 09:49:10 +00:00
continue
for tag in postJsonObject['object']['tag']:
2020-02-24 11:10:48 +00:00
if not isHappeningEvent(tag):
continue
2020-02-24 11:10:48 +00:00
# this tag is an event or a place
2020-04-03 11:53:31 +00:00
if tag['type'] != 'Event':
continue
2020-02-24 11:10:48 +00:00
# tag is an event
if not tag.get('startTime'):
continue
2020-04-03 11:53:31 +00:00
eventTime = \
datetime.strptime(tag['startTime'],
2020-02-24 11:10:48 +00:00
"%Y-%m-%dT%H:%M:%S%z")
2020-04-03 11:53:31 +00:00
if (int(eventTime.strftime("%Y")) == year and
int(eventTime.strftime("%m")) == monthNumber and
(int(eventTime.strftime("%d")) > dayNumber and
int(eventTime.strftime("%d")) <= dayNumber + 6)):
eventsExist = True
2020-02-24 11:10:48 +00:00
break
return eventsExist
2020-04-03 11:53:31 +00:00
def getThisWeeksEvents(baseDir: str, nickname: str, domain: str) -> {}:
"""Retrieves calendar events for this week
2020-02-24 10:55:49 +00:00
Returns a dictionary indexed by day number of lists containing
Event and Place activities
2020-02-23 09:42:09 +00:00
Note: currently not used but could be with a weekly calendar screen
"""
2020-04-03 11:53:31 +00:00
now = datetime.now()
year = now.year
monthNumber = now.month
dayNumber = now.day
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
calendarFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
events = {}
if not os.path.isfile(calendarFilename):
return events
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
calendarPostIds = []
recreateEventsFile = False
with open(calendarFilename, 'r') as eventsFile:
for postId in eventsFile:
2020-04-03 11:53:31 +00:00
postId = postId.replace('\n', '')
postFilename = locatePost(baseDir, nickname, domain, postId)
if not postFilename:
2020-04-03 11:53:31 +00:00
recreateEventsFile = True
2020-02-23 09:49:10 +00:00
continue
2020-04-03 11:53:31 +00:00
postJsonObject = loadJson(postFilename)
2020-02-24 11:10:48 +00:00
if not isHappeningPost(postJsonObject):
2020-02-23 09:49:10 +00:00
continue
2020-04-03 11:53:31 +00:00
postEvent = []
dayOfMonth = None
weekDayIndex = None
2020-02-23 09:49:10 +00:00
for tag in postJsonObject['object']['tag']:
2020-02-24 11:10:48 +00:00
if not isHappeningEvent(tag):
continue
2020-02-24 11:10:48 +00:00
# this tag is an event or a place
2020-04-03 11:53:31 +00:00
if tag['type'] == 'Event':
2020-02-23 09:49:10 +00:00
# tag is an event
if not tag.get('startTime'):
continue
2020-04-03 11:53:31 +00:00
eventTime = \
datetime.strptime(tag['startTime'],
2020-02-23 09:49:10 +00:00
"%Y-%m-%dT%H:%M:%S%z")
2020-04-03 11:53:31 +00:00
if (int(eventTime.strftime("%Y")) == year and
int(eventTime.strftime("%m")) == monthNumber and
(int(eventTime.strftime("%d")) >= dayNumber and
int(eventTime.strftime("%d")) <= dayNumber + 6)):
dayOfMonth = str(int(eventTime.strftime("%d")))
weekDayIndex = dayOfMonth - dayNumber
postEvent.append(tag)
2020-02-23 09:49:10 +00:00
else:
# tag is a place
postEvent.append(tag)
if postEvent and weekDayIndex:
calendarPostIds.append(postId)
if not events.get(dayOfMonth):
2020-04-03 11:53:31 +00:00
events[weekDayIndex] = []
2020-02-23 09:49:10 +00:00
events[dayOfMonth].append(postEvent)
# if some posts have been deleted then regenerate the calendar file
if recreateEventsFile:
2020-04-03 11:53:31 +00:00
calendarFile = open(calendarFilename, "w")
for postId in calendarPostIds:
2020-04-03 11:53:31 +00:00
calendarFile.write(postId + '\n')
calendarFile.close()
2020-04-03 11:53:31 +00:00
lastDayOfMonth = daysInMonth(year, monthNumber)
if dayNumber+6 > lastDayOfMonth:
2020-04-03 11:53:31 +00:00
monthNumber += 1
if monthNumber > 12:
monthNumber = 1
year += 1
for d in range(1, dayNumber + 6 - lastDayOfMonth):
dailyEvents = \
getTodaysEvents(baseDir, nickname, domain,
year, monthNumber, d)
if dailyEvents:
if dailyEvents.get(d):
2020-04-03 11:53:31 +00:00
newEvents = {}
newEvents[d + (7 - (dayNumber + 6 - lastDayOfMonth))] = \
2020-02-24 10:55:49 +00:00
dailyEvents[d]
2020-04-03 11:53:31 +00:00
events = mergeDicts(events, newEvents)
return events
2020-04-03 11:53:31 +00:00
def getCalendarEvents(baseDir: str, nickname: str, domain: str,
year: int, monthNumber: int) -> {}:
"""Retrieves calendar events
2020-02-24 10:55:49 +00:00
Returns a dictionary indexed by day number of lists containing
Event and Place activities
"""
2020-04-03 11:53:31 +00:00
calendarFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
events = {}
if not os.path.isfile(calendarFilename):
return events
2020-02-23 09:45:04 +00:00
2020-04-03 11:53:31 +00:00
calendarPostIds = []
recreateEventsFile = False
with open(calendarFilename, 'r') as eventsFile:
for postId in eventsFile:
2020-04-03 11:53:31 +00:00
postId = postId.replace('\n', '')
postFilename = locatePost(baseDir, nickname, domain, postId)
if not postFilename:
2020-04-03 11:53:31 +00:00
recreateEventsFile = True
2020-02-23 09:42:09 +00:00
continue
2020-02-24 11:10:48 +00:00
2020-04-03 11:53:31 +00:00
postJsonObject = loadJson(postFilename)
2020-02-24 11:10:48 +00:00
if not isHappeningPost(postJsonObject):
2020-02-23 09:42:09 +00:00
continue
2020-04-03 11:53:31 +00:00
postEvent = []
dayOfMonth = None
2020-02-23 09:42:09 +00:00
for tag in postJsonObject['object']['tag']:
2020-02-24 11:10:48 +00:00
if not isHappeningEvent(tag):
2020-02-23 09:42:09 +00:00
continue
2020-02-24 11:10:48 +00:00
# this tag is an event or a place
2020-04-03 11:53:31 +00:00
if tag['type'] == 'Event':
2020-02-23 09:42:09 +00:00
# tag is an event
if not tag.get('startTime'):
continue
2020-04-03 11:53:31 +00:00
eventTime = \
datetime.strptime(tag['startTime'],
2020-02-23 09:42:09 +00:00
"%Y-%m-%dT%H:%M:%S%z")
2020-04-03 11:53:31 +00:00
if int(eventTime.strftime("%Y")) == year and \
int(eventTime.strftime("%m")) == monthNumber:
dayOfMonth = str(int(eventTime.strftime("%d")))
2020-02-23 09:42:09 +00:00
postEvent.append(tag)
else:
# tag is a place
postEvent.append(tag)
if postEvent and dayOfMonth:
calendarPostIds.append(postId)
if not events.get(dayOfMonth):
2020-04-03 11:53:31 +00:00
events[dayOfMonth] = []
2020-02-23 09:42:09 +00:00
events[dayOfMonth].append(postEvent)
# if some posts have been deleted then regenerate the calendar file
if recreateEventsFile:
2020-04-03 11:53:31 +00:00
calendarFile = open(calendarFilename, "w")
for postId in calendarPostIds:
2020-04-03 11:53:31 +00:00
calendarFile.write(postId + '\n')
calendarFile.close()
2020-03-22 21:16:02 +00:00
return events
2020-02-23 13:28:27 +00:00
2020-04-03 11:53:31 +00:00
def removeCalendarEvent(baseDir: str, nickname: str, domain: str,
year: int, monthNumber: int, messageId: str) -> None:
2020-02-23 13:28:27 +00:00
"""Removes a calendar event
"""
2020-04-03 11:53:31 +00:00
calendarFilename = \
baseDir + '/accounts/' + nickname + '@' + domain + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
2020-02-23 13:28:27 +00:00
if not os.path.isfile(calendarFilename):
return
if '/' in messageId:
2020-04-03 11:53:31 +00:00
messageId = messageId.replace('/', '#')
2020-02-23 13:28:27 +00:00
if messageId not in open(calendarFilename).read():
return
2020-04-03 11:53:31 +00:00
lines = None
2020-02-23 13:28:27 +00:00
with open(calendarFilename, "r") as f:
2020-04-03 11:53:31 +00:00
lines = f.readlines()
2020-02-23 13:28:27 +00:00
if not lines:
return
with open(calendarFilename, "w+") as f:
for line in lines:
if messageId not in line:
f.write(line)