forked from indymedia/epicyon
flake8 format
parent
283a4d54c1
commit
34abcb54c2
322
happening.py
322
happening.py
|
|
@ -1,13 +1,11 @@
|
|||
__filename__="happening.py"
|
||||
__author__="Bob Mottram"
|
||||
__license__="AGPL3+"
|
||||
__version__="1.1.0"
|
||||
__maintainer__="Bob Mottram"
|
||||
__email__="bob@freedombone.net"
|
||||
__status__="Production"
|
||||
__filename__ = "happening.py"
|
||||
__author__ = "Bob Mottram"
|
||||
__license__ = "AGPL3+"
|
||||
__version__ = "1.1.0"
|
||||
__maintainer__ = "Bob Mottram"
|
||||
__email__ = "bob@freedombone.net"
|
||||
__status__ = "Production"
|
||||
|
||||
import json
|
||||
import time
|
||||
import os
|
||||
from datetime import datetime
|
||||
|
||||
|
|
@ -16,15 +14,17 @@ from utils import locatePost
|
|||
from utils import daysInMonth
|
||||
from utils import mergeDicts
|
||||
|
||||
|
||||
def isHappeningEvent(tag: {}) -> bool:
|
||||
"""Is this tag an Event or Place ActivityStreams type?
|
||||
"""
|
||||
if not tag.get('type'):
|
||||
return False
|
||||
if tag['type']!='Event' and tag['type']!='Place':
|
||||
if tag['type'] != 'Event' and tag['type'] != 'Place':
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
def isHappeningPost(postJsonObject: {}) -> bool:
|
||||
"""Is this a post with tags?
|
||||
"""
|
||||
|
|
@ -38,68 +38,69 @@ def isHappeningPost(postJsonObject: {}) -> bool:
|
|||
return False
|
||||
return True
|
||||
|
||||
def getTodaysEvents(baseDir: str,nickname: str,domain: str, \
|
||||
currYear=None,currMonthNumber=None, \
|
||||
|
||||
def getTodaysEvents(baseDir: str, nickname: str, domain: str,
|
||||
currYear=None, currMonthNumber=None,
|
||||
currDayOfMonth=None) -> {}:
|
||||
"""Retrieves calendar events for today
|
||||
Returns a dictionary of lists containing Event and Place activities
|
||||
"""
|
||||
now=datetime.now()
|
||||
now = datetime.now()
|
||||
if not currYear:
|
||||
year=now.year
|
||||
year = now.year
|
||||
else:
|
||||
year=currYear
|
||||
year = currYear
|
||||
if not currMonthNumber:
|
||||
monthNumber=now.month
|
||||
monthNumber = now.month
|
||||
else:
|
||||
monthNumber=currMonthNumber
|
||||
monthNumber = currMonthNumber
|
||||
if not currDayOfMonth:
|
||||
dayNumber=now.day
|
||||
dayNumber = now.day
|
||||
else:
|
||||
dayNumber=currDayOfMonth
|
||||
dayNumber = currDayOfMonth
|
||||
|
||||
calendarFilename= \
|
||||
baseDir+'/accounts/'+nickname+'@'+domain+ \
|
||||
'/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
|
||||
events={}
|
||||
calendarFilename = \
|
||||
baseDir + '/accounts/' + nickname + '@' + domain + \
|
||||
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
|
||||
events = {}
|
||||
if not os.path.isfile(calendarFilename):
|
||||
return events
|
||||
|
||||
calendarPostIds=[]
|
||||
recreateEventsFile=False
|
||||
with open(calendarFilename,'r') as eventsFile:
|
||||
calendarPostIds = []
|
||||
recreateEventsFile = False
|
||||
with open(calendarFilename, 'r') as eventsFile:
|
||||
for postId in eventsFile:
|
||||
postId=postId.replace('\n','')
|
||||
postFilename=locatePost(baseDir,nickname,domain,postId)
|
||||
postId = postId.replace('\n', '')
|
||||
postFilename = locatePost(baseDir, nickname, domain, postId)
|
||||
if not postFilename:
|
||||
recreateEventsFile=True
|
||||
recreateEventsFile = True
|
||||
continue
|
||||
|
||||
postJsonObject=loadJson(postFilename)
|
||||
postJsonObject = loadJson(postFilename)
|
||||
if not isHappeningPost(postJsonObject):
|
||||
continue
|
||||
|
||||
postEvent=[]
|
||||
dayOfMonth=None
|
||||
postEvent = []
|
||||
dayOfMonth = None
|
||||
for tag in postJsonObject['object']['tag']:
|
||||
if not isHappeningEvent(tag):
|
||||
continue
|
||||
# this tag is an event or a place
|
||||
if tag['type']=='Event':
|
||||
if tag['type'] == 'Event':
|
||||
# tag is an event
|
||||
if not tag.get('startTime'):
|
||||
continue
|
||||
eventTime= \
|
||||
datetime.strptime(tag['startTime'], \
|
||||
eventTime = \
|
||||
datetime.strptime(tag['startTime'],
|
||||
"%Y-%m-%dT%H:%M:%S%z")
|
||||
if int(eventTime.strftime("%Y"))==year and \
|
||||
int(eventTime.strftime("%m"))==monthNumber and \
|
||||
int(eventTime.strftime("%d"))==dayNumber:
|
||||
dayOfMonth=str(int(eventTime.strftime("%d")))
|
||||
if int(eventTime.strftime("%Y")) == year and \
|
||||
int(eventTime.strftime("%m")) == monthNumber and \
|
||||
int(eventTime.strftime("%d")) == dayNumber:
|
||||
dayOfMonth = str(int(eventTime.strftime("%d")))
|
||||
if '#statuses#' in postId:
|
||||
# link to the id so that the event can be
|
||||
# easily deleted
|
||||
tag['postId']=postId.split('#statuses#')[1]
|
||||
tag['postId'] = postId.split('#statuses#')[1]
|
||||
postEvent.append(tag)
|
||||
else:
|
||||
# tag is a place
|
||||
|
|
@ -107,41 +108,42 @@ def getTodaysEvents(baseDir: str,nickname: str,domain: str, \
|
|||
if postEvent and dayOfMonth:
|
||||
calendarPostIds.append(postId)
|
||||
if not events.get(dayOfMonth):
|
||||
events[dayOfMonth]=[]
|
||||
events[dayOfMonth] = []
|
||||
events[dayOfMonth].append(postEvent)
|
||||
|
||||
# if some posts have been deleted then regenerate the calendar file
|
||||
if recreateEventsFile:
|
||||
calendarFile=open(calendarFilename, "w")
|
||||
calendarFile = open(calendarFilename, "w")
|
||||
for postId in calendarPostIds:
|
||||
calendarFile.write(postId+'\n')
|
||||
calendarFile.close()
|
||||
|
||||
return events
|
||||
|
||||
def todaysEventsCheck(baseDir: str,nickname: str,domain: str) -> bool:
|
||||
|
||||
def todaysEventsCheck(baseDir: str, nickname: str, domain: str) -> bool:
|
||||
"""Are there calendar events today?
|
||||
"""
|
||||
now=datetime.now()
|
||||
year=now.year
|
||||
monthNumber=now.month
|
||||
dayNumber=now.day
|
||||
now = datetime.now()
|
||||
year = now.year
|
||||
monthNumber = now.month
|
||||
dayNumber = now.day
|
||||
|
||||
calendarFilename= \
|
||||
baseDir+'/accounts/'+nickname+'@'+domain+ \
|
||||
'/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
|
||||
calendarFilename = \
|
||||
baseDir + '/accounts/' + nickname + '@' + domain + \
|
||||
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
|
||||
if not os.path.isfile(calendarFilename):
|
||||
return False
|
||||
|
||||
eventsExist=False
|
||||
with open(calendarFilename,'r') as eventsFile:
|
||||
eventsExist = False
|
||||
with open(calendarFilename, 'r') as eventsFile:
|
||||
for postId in eventsFile:
|
||||
postId=postId.replace('\n','')
|
||||
postFilename=locatePost(baseDir,nickname,domain,postId)
|
||||
postId = postId.replace('\n', '')
|
||||
postFilename = locatePost(baseDir, nickname, domain, postId)
|
||||
if not postFilename:
|
||||
continue
|
||||
|
||||
postJsonObject=loadJson(postFilename)
|
||||
postJsonObject = loadJson(postFilename)
|
||||
if not isHappeningPost(postJsonObject):
|
||||
continue
|
||||
|
||||
|
|
@ -149,45 +151,46 @@ def todaysEventsCheck(baseDir: str,nickname: str,domain: str) -> bool:
|
|||
if not isHappeningEvent(tag):
|
||||
continue
|
||||
# this tag is an event or a place
|
||||
if tag['type']!='Event':
|
||||
if tag['type'] != 'Event':
|
||||
continue
|
||||
# tag is an event
|
||||
if not tag.get('startTime'):
|
||||
continue
|
||||
eventTime= \
|
||||
datetime.strptime(tag['startTime'], \
|
||||
eventTime = \
|
||||
datetime.strptime(tag['startTime'],
|
||||
"%Y-%m-%dT%H:%M:%S%z")
|
||||
if int(eventTime.strftime("%Y"))==year and \
|
||||
int(eventTime.strftime("%m"))==monthNumber and \
|
||||
int(eventTime.strftime("%d"))==dayNumber:
|
||||
eventsExist=True
|
||||
if int(eventTime.strftime("%Y")) == year and \
|
||||
int(eventTime.strftime("%m")) == monthNumber and \
|
||||
int(eventTime.strftime("%d")) == dayNumber:
|
||||
eventsExist = True
|
||||
break
|
||||
|
||||
return eventsExist
|
||||
|
||||
def thisWeeksEventsCheck(baseDir: str,nickname: str,domain: str) -> bool:
|
||||
|
||||
def thisWeeksEventsCheck(baseDir: str, nickname: str, domain: str) -> bool:
|
||||
"""Are there calendar events this week?
|
||||
"""
|
||||
now=datetime.now()
|
||||
year=now.year
|
||||
monthNumber=now.month
|
||||
dayNumber=now.day
|
||||
now = datetime.now()
|
||||
year = now.year
|
||||
monthNumber = now.month
|
||||
dayNumber = now.day
|
||||
|
||||
calendarFilename= \
|
||||
baseDir+'/accounts/'+nickname+'@'+domain+ \
|
||||
'/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
|
||||
calendarFilename = \
|
||||
baseDir + '/accounts/' + nickname + '@' + domain + \
|
||||
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
|
||||
if not os.path.isfile(calendarFilename):
|
||||
return False
|
||||
|
||||
eventsExist=False
|
||||
with open(calendarFilename,'r') as eventsFile:
|
||||
eventsExist = False
|
||||
with open(calendarFilename, 'r') as eventsFile:
|
||||
for postId in eventsFile:
|
||||
postId=postId.replace('\n','')
|
||||
postFilename=locatePost(baseDir,nickname,domain,postId)
|
||||
postId = postId.replace('\n', '')
|
||||
postFilename = locatePost(baseDir, nickname, domain, postId)
|
||||
if not postFilename:
|
||||
continue
|
||||
|
||||
postJsonObject=loadJson(postFilename)
|
||||
postJsonObject = loadJson(postFilename)
|
||||
if not isHappeningPost(postJsonObject):
|
||||
continue
|
||||
|
||||
|
|
@ -195,76 +198,77 @@ def thisWeeksEventsCheck(baseDir: str,nickname: str,domain: str) -> bool:
|
|||
if not isHappeningEvent(tag):
|
||||
continue
|
||||
# this tag is an event or a place
|
||||
if tag['type']!='Event':
|
||||
if tag['type'] != 'Event':
|
||||
continue
|
||||
# tag is an event
|
||||
if not tag.get('startTime'):
|
||||
continue
|
||||
eventTime= \
|
||||
datetime.strptime(tag['startTime'], \
|
||||
eventTime = \
|
||||
datetime.strptime(tag['startTime'],
|
||||
"%Y-%m-%dT%H:%M:%S%z")
|
||||
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
|
||||
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
|
||||
break
|
||||
|
||||
return eventsExist
|
||||
|
||||
def getThisWeeksEvents(baseDir: str,nickname: str,domain: str) -> {}:
|
||||
|
||||
def getThisWeeksEvents(baseDir: str, nickname: str, domain: str) -> {}:
|
||||
"""Retrieves calendar events for this week
|
||||
Returns a dictionary indexed by day number of lists containing
|
||||
Event and Place activities
|
||||
Note: currently not used but could be with a weekly calendar screen
|
||||
"""
|
||||
now=datetime.now()
|
||||
year=now.year
|
||||
monthNumber=now.month
|
||||
dayNumber=now.day
|
||||
now = datetime.now()
|
||||
year = now.year
|
||||
monthNumber = now.month
|
||||
dayNumber = now.day
|
||||
|
||||
calendarFilename= \
|
||||
baseDir+'/accounts/'+nickname+'@'+domain+ \
|
||||
'/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
|
||||
calendarFilename = \
|
||||
baseDir + '/accounts/' + nickname + '@' + domain + \
|
||||
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
|
||||
|
||||
events={}
|
||||
events = {}
|
||||
if not os.path.isfile(calendarFilename):
|
||||
return events
|
||||
|
||||
calendarPostIds=[]
|
||||
recreateEventsFile=False
|
||||
with open(calendarFilename,'r') as eventsFile:
|
||||
calendarPostIds = []
|
||||
recreateEventsFile = False
|
||||
with open(calendarFilename, 'r') as eventsFile:
|
||||
for postId in eventsFile:
|
||||
postId=postId.replace('\n','')
|
||||
postFilename=locatePost(baseDir,nickname,domain,postId)
|
||||
postId = postId.replace('\n', '')
|
||||
postFilename = locatePost(baseDir, nickname, domain, postId)
|
||||
if not postFilename:
|
||||
recreateEventsFile=True
|
||||
recreateEventsFile = True
|
||||
continue
|
||||
|
||||
postJsonObject=loadJson(postFilename)
|
||||
postJsonObject = loadJson(postFilename)
|
||||
if not isHappeningPost(postJsonObject):
|
||||
continue
|
||||
|
||||
postEvent=[]
|
||||
dayOfMonth=None
|
||||
weekDayIndex=None
|
||||
postEvent = []
|
||||
dayOfMonth = None
|
||||
weekDayIndex = None
|
||||
for tag in postJsonObject['object']['tag']:
|
||||
if not isHappeningEvent(tag):
|
||||
continue
|
||||
# this tag is an event or a place
|
||||
if tag['type']=='Event':
|
||||
if tag['type'] == 'Event':
|
||||
# tag is an event
|
||||
if not tag.get('startTime'):
|
||||
continue
|
||||
eventTime= \
|
||||
datetime.strptime(tag['startTime'], \
|
||||
eventTime = \
|
||||
datetime.strptime(tag['startTime'],
|
||||
"%Y-%m-%dT%H:%M:%S%z")
|
||||
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
|
||||
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)
|
||||
else:
|
||||
# tag is a place
|
||||
|
|
@ -272,79 +276,80 @@ def getThisWeeksEvents(baseDir: str,nickname: str,domain: str) -> {}:
|
|||
if postEvent and weekDayIndex:
|
||||
calendarPostIds.append(postId)
|
||||
if not events.get(dayOfMonth):
|
||||
events[weekDayIndex]=[]
|
||||
events[weekDayIndex] = []
|
||||
events[dayOfMonth].append(postEvent)
|
||||
|
||||
# if some posts have been deleted then regenerate the calendar file
|
||||
if recreateEventsFile:
|
||||
calendarFile=open(calendarFilename, "w")
|
||||
calendarFile = open(calendarFilename, "w")
|
||||
for postId in calendarPostIds:
|
||||
calendarFile.write(postId+'\n')
|
||||
calendarFile.write(postId + '\n')
|
||||
calendarFile.close()
|
||||
|
||||
lastDayOfMonth=daysInMonth(year,monthNumber)
|
||||
lastDayOfMonth = daysInMonth(year, monthNumber)
|
||||
if dayNumber+6 > lastDayOfMonth:
|
||||
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)
|
||||
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):
|
||||
newEvents={}
|
||||
newEvents[d+(7-(dayNumber+6-lastDayOfMonth))]= \
|
||||
newEvents = {}
|
||||
newEvents[d + (7 - (dayNumber + 6 - lastDayOfMonth))] = \
|
||||
dailyEvents[d]
|
||||
events=mergeDicts(events,newEvents)
|
||||
events = mergeDicts(events, newEvents)
|
||||
|
||||
return events
|
||||
|
||||
def getCalendarEvents(baseDir: str,nickname: str,domain: str, \
|
||||
year: int,monthNumber: int) -> {}:
|
||||
|
||||
def getCalendarEvents(baseDir: str, nickname: str, domain: str,
|
||||
year: int, monthNumber: int) -> {}:
|
||||
"""Retrieves calendar events
|
||||
Returns a dictionary indexed by day number of lists containing
|
||||
Event and Place activities
|
||||
"""
|
||||
calendarFilename= \
|
||||
baseDir+'/accounts/'+nickname+'@'+domain+ \
|
||||
'/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
|
||||
calendarFilename = \
|
||||
baseDir + '/accounts/' + nickname + '@' + domain + \
|
||||
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
|
||||
|
||||
events={}
|
||||
events = {}
|
||||
if not os.path.isfile(calendarFilename):
|
||||
return events
|
||||
|
||||
calendarPostIds=[]
|
||||
recreateEventsFile=False
|
||||
with open(calendarFilename,'r') as eventsFile:
|
||||
calendarPostIds = []
|
||||
recreateEventsFile = False
|
||||
with open(calendarFilename, 'r') as eventsFile:
|
||||
for postId in eventsFile:
|
||||
postId=postId.replace('\n','')
|
||||
postFilename=locatePost(baseDir,nickname,domain,postId)
|
||||
postId = postId.replace('\n', '')
|
||||
postFilename = locatePost(baseDir, nickname, domain, postId)
|
||||
if not postFilename:
|
||||
recreateEventsFile=True
|
||||
recreateEventsFile = True
|
||||
continue
|
||||
|
||||
postJsonObject=loadJson(postFilename)
|
||||
postJsonObject = loadJson(postFilename)
|
||||
if not isHappeningPost(postJsonObject):
|
||||
continue
|
||||
|
||||
postEvent=[]
|
||||
dayOfMonth=None
|
||||
postEvent = []
|
||||
dayOfMonth = None
|
||||
for tag in postJsonObject['object']['tag']:
|
||||
if not isHappeningEvent(tag):
|
||||
continue
|
||||
# this tag is an event or a place
|
||||
if tag['type']=='Event':
|
||||
if tag['type'] == 'Event':
|
||||
# tag is an event
|
||||
if not tag.get('startTime'):
|
||||
continue
|
||||
eventTime= \
|
||||
datetime.strptime(tag['startTime'], \
|
||||
eventTime = \
|
||||
datetime.strptime(tag['startTime'],
|
||||
"%Y-%m-%dT%H:%M:%S%z")
|
||||
if int(eventTime.strftime("%Y"))==year and \
|
||||
int(eventTime.strftime("%m"))==monthNumber:
|
||||
dayOfMonth=str(int(eventTime.strftime("%d")))
|
||||
if int(eventTime.strftime("%Y")) == year and \
|
||||
int(eventTime.strftime("%m")) == monthNumber:
|
||||
dayOfMonth = str(int(eventTime.strftime("%d")))
|
||||
postEvent.append(tag)
|
||||
else:
|
||||
# tag is a place
|
||||
|
|
@ -353,34 +358,35 @@ def getCalendarEvents(baseDir: str,nickname: str,domain: str, \
|
|||
if postEvent and dayOfMonth:
|
||||
calendarPostIds.append(postId)
|
||||
if not events.get(dayOfMonth):
|
||||
events[dayOfMonth]=[]
|
||||
events[dayOfMonth] = []
|
||||
events[dayOfMonth].append(postEvent)
|
||||
|
||||
# if some posts have been deleted then regenerate the calendar file
|
||||
if recreateEventsFile:
|
||||
calendarFile=open(calendarFilename, "w")
|
||||
calendarFile = open(calendarFilename, "w")
|
||||
for postId in calendarPostIds:
|
||||
calendarFile.write(postId+'\n')
|
||||
calendarFile.write(postId + '\n')
|
||||
calendarFile.close()
|
||||
|
||||
return events
|
||||
|
||||
def removeCalendarEvent(baseDir: str,nickname: str,domain: str, \
|
||||
year: int,monthNumber: int,messageId: str) -> None:
|
||||
|
||||
def removeCalendarEvent(baseDir: str, nickname: str, domain: str,
|
||||
year: int, monthNumber: int, messageId: str) -> None:
|
||||
"""Removes a calendar event
|
||||
"""
|
||||
calendarFilename= \
|
||||
baseDir+'/accounts/'+nickname+'@'+domain+ \
|
||||
'/calendar/'+str(year)+'/'+str(monthNumber)+'.txt'
|
||||
calendarFilename = \
|
||||
baseDir + '/accounts/' + nickname + '@' + domain + \
|
||||
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
|
||||
if not os.path.isfile(calendarFilename):
|
||||
return
|
||||
if '/' in messageId:
|
||||
messageId=messageId.replace('/','#')
|
||||
messageId = messageId.replace('/', '#')
|
||||
if messageId not in open(calendarFilename).read():
|
||||
return
|
||||
lines=None
|
||||
lines = None
|
||||
with open(calendarFilename, "r") as f:
|
||||
lines=f.readlines()
|
||||
lines = f.readlines()
|
||||
if not lines:
|
||||
return
|
||||
with open(calendarFilename, "w+") as f:
|
||||
|
|
|
|||
Loading…
Reference in New Issue