Snake case

merge-requests/26/head
Bob Mottram 2022-01-02 15:16:47 +00:00
parent 7da2381aa0
commit 27c91490b0
1 changed files with 179 additions and 177 deletions

View File

@ -20,33 +20,34 @@ from utils import has_object_dict
from utils import acct_dir
def _valid_uuid(testUuid: str, version: int):
def _valid_uuid(test_uuid: str, version: int):
"""Check if uuid_to_test is a valid UUID
"""
try:
uuid_obj = UUID(testUuid, version=version)
uuid_obj = UUID(test_uuid, version=version)
except ValueError:
return False
return str(uuid_obj) == testUuid
return str(uuid_obj) == test_uuid
def _remove_event_from_timeline(eventId: str, tlEventsFilename: str) -> None:
def _remove_event_from_timeline(event_id: str,
tl_events_filename: str) -> None:
"""Removes the given event Id from the timeline
"""
if eventId + '\n' not in open(tlEventsFilename).read():
if event_id + '\n' not in open(tl_events_filename).read():
return
with open(tlEventsFilename, 'r') as fp:
eventsTimeline = fp.read().replace(eventId + '\n', '')
with open(tl_events_filename, 'r') as fp_tl:
events_timeline = fp_tl.read().replace(event_id + '\n', '')
try:
with open(tlEventsFilename, 'w+') as fp2:
fp2.write(eventsTimeline)
with open(tl_events_filename, 'w+') as fp2:
fp2.write(events_timeline)
except OSError:
print('EX: ERROR: unable to save events timeline')
def save_event_post(base_dir: str, handle: str, post_id: str,
eventJson: {}) -> bool:
event_json: {}) -> bool:
"""Saves an event to the calendar and/or the events timeline
If an event has extra fields, as per Mobilizon,
Then it is saved as a separate entity and added to the
@ -57,96 +58,96 @@ def save_event_post(base_dir: str, handle: str, post_id: str,
if not os.path.isdir(base_dir + '/accounts/' + handle):
print('WARN: Account does not exist at ' +
base_dir + '/accounts/' + handle)
calendarPath = base_dir + '/accounts/' + handle + '/calendar'
if not os.path.isdir(calendarPath):
os.mkdir(calendarPath)
calendar_path = base_dir + '/accounts/' + handle + '/calendar'
if not os.path.isdir(calendar_path):
os.mkdir(calendar_path)
# get the year, month and day from the event
eventTime = datetime.strptime(eventJson['startTime'],
"%Y-%m-%dT%H:%M:%S%z")
eventYear = int(eventTime.strftime("%Y"))
if eventYear < 2020 or eventYear >= 2100:
event_time = datetime.strptime(event_json['startTime'],
"%Y-%m-%dT%H:%M:%S%z")
event_year = int(event_time.strftime("%Y"))
if event_year < 2020 or event_year >= 2100:
return False
eventMonthNumber = int(eventTime.strftime("%m"))
if eventMonthNumber < 1 or eventMonthNumber > 12:
event_month_number = int(event_time.strftime("%m"))
if event_month_number < 1 or event_month_number > 12:
return False
eventDayOfMonth = int(eventTime.strftime("%d"))
if eventDayOfMonth < 1 or eventDayOfMonth > 31:
event_day_of_month = int(event_time.strftime("%d"))
if event_day_of_month < 1 or event_day_of_month > 31:
return False
if eventJson.get('name') and eventJson.get('actor') and \
eventJson.get('uuid') and eventJson.get('content'):
if not _valid_uuid(eventJson['uuid'], 4):
if event_json.get('name') and event_json.get('actor') and \
event_json.get('uuid') and event_json.get('content'):
if not _valid_uuid(event_json['uuid'], 4):
return False
print('Mobilizon type event')
# if this is a full description of an event then save it
# as a separate json file
eventsPath = base_dir + '/accounts/' + handle + '/events'
if not os.path.isdir(eventsPath):
os.mkdir(eventsPath)
eventsYearPath = \
base_dir + '/accounts/' + handle + '/events/' + str(eventYear)
if not os.path.isdir(eventsYearPath):
os.mkdir(eventsYearPath)
eventId = str(eventYear) + '-' + eventTime.strftime("%m") + '-' + \
eventTime.strftime("%d") + '_' + eventJson['uuid']
eventFilename = eventsYearPath + '/' + eventId + '.json'
events_path = base_dir + '/accounts/' + handle + '/events'
if not os.path.isdir(events_path):
os.mkdir(events_path)
events_year_path = \
base_dir + '/accounts/' + handle + '/events/' + str(event_year)
if not os.path.isdir(events_year_path):
os.mkdir(events_year_path)
event_id = str(event_year) + '-' + event_time.strftime("%m") + '-' + \
event_time.strftime("%d") + '_' + event_json['uuid']
event_filename = events_year_path + '/' + event_id + '.json'
save_json(eventJson, eventFilename)
save_json(event_json, event_filename)
# save to the events timeline
tlEventsFilename = base_dir + '/accounts/' + handle + '/events.txt'
tl_events_filename = base_dir + '/accounts/' + handle + '/events.txt'
if os.path.isfile(tlEventsFilename):
_remove_event_from_timeline(eventId, tlEventsFilename)
if os.path.isfile(tl_events_filename):
_remove_event_from_timeline(event_id, tl_events_filename)
try:
with open(tlEventsFilename, 'r+') as tlEventsFile:
content = tlEventsFile.read()
if eventId + '\n' not in content:
tlEventsFile.seek(0, 0)
tlEventsFile.write(eventId + '\n' + content)
with open(tl_events_filename, 'r+') as tl_events_file:
content = tl_events_file.read()
if event_id + '\n' not in content:
tl_events_file.seek(0, 0)
tl_events_file.write(event_id + '\n' + content)
except OSError as ex:
print('EX: Failed to write entry to events file ' +
tlEventsFilename + ' ' + str(ex))
tl_events_filename + ' ' + str(ex))
return False
else:
try:
with open(tlEventsFilename, 'w+') as tlEventsFile:
tlEventsFile.write(eventId + '\n')
with open(tl_events_filename, 'w+') as tl_events_file:
tl_events_file.write(event_id + '\n')
except OSError:
print('EX: unable to write ' + tlEventsFilename)
print('EX: unable to write ' + tl_events_filename)
# create a directory for the calendar year
if not os.path.isdir(calendarPath + '/' + str(eventYear)):
os.mkdir(calendarPath + '/' + str(eventYear))
if not os.path.isdir(calendar_path + '/' + str(event_year)):
os.mkdir(calendar_path + '/' + str(event_year))
# calendar month file containing event post Ids
calendarFilename = calendarPath + '/' + str(eventYear) + \
'/' + str(eventMonthNumber) + '.txt'
calendar_filename = calendar_path + '/' + str(event_year) + \
'/' + str(event_month_number) + '.txt'
# Does this event post already exist within the calendar month?
if os.path.isfile(calendarFilename):
if post_id in open(calendarFilename).read():
if os.path.isfile(calendar_filename):
if post_id in open(calendar_filename).read():
# Event post already exists
return False
# append the post Id to the file for the calendar month
try:
with open(calendarFilename, 'a+') as calendarFile:
calendarFile.write(post_id + '\n')
with open(calendar_filename, 'a+') as calendar_file:
calendar_file.write(post_id + '\n')
except OSError:
print('EX: unable to append ' + calendarFilename)
print('EX: unable to append ' + calendar_filename)
# create a file which will trigger a notification that
# a new event has been added
calNotifyFilename = base_dir + '/accounts/' + handle + '/.newCalendar'
notifyStr = \
'/calendar?year=' + str(eventYear) + '?month=' + \
str(eventMonthNumber) + '?day=' + str(eventDayOfMonth)
cal_notify_filename = base_dir + '/accounts/' + handle + '/.newCalendar'
notify_str = \
'/calendar?year=' + str(event_year) + '?month=' + \
str(event_month_number) + '?day=' + str(event_day_of_month)
try:
with open(calNotifyFilename, 'w+') as calendarNotificationFile:
calendarNotificationFile.write(notifyStr)
with open(cal_notify_filename, 'w+') as cal_file:
cal_file.write(notify_str)
except OSError:
print('EX: unable to write ' + calNotifyFilename)
print('EX: unable to write ' + cal_notify_filename)
return False
return True
@ -185,39 +186,39 @@ def get_todays_events(base_dir: str, nickname: str, domain: str,
else:
year = currYear
if not currMonthNumber:
monthNumber = now.month
month_number = now.month
else:
monthNumber = currMonthNumber
month_number = currMonthNumber
if not currDayOfMonth:
dayNumber = now.day
day_number = now.day
else:
dayNumber = currDayOfMonth
day_number = currDayOfMonth
calendarFilename = \
calendar_filename = \
acct_dir(base_dir, nickname, domain) + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
'/calendar/' + str(year) + '/' + str(month_number) + '.txt'
events = {}
if not os.path.isfile(calendarFilename):
if not os.path.isfile(calendar_filename):
return events
calendarPostIds = []
recreateEventsFile = False
with open(calendarFilename, 'r') as eventsFile:
for post_id in eventsFile:
calendar_post_ids = []
recreate_events_file = False
with open(calendar_filename, 'r') as events_file:
for post_id in events_file:
post_id = post_id.replace('\n', '').replace('\r', '')
post_filename = locate_post(base_dir, nickname, domain, post_id)
if not post_filename:
recreateEventsFile = True
recreate_events_file = True
continue
post_json_object = load_json(post_filename)
if not _is_happening_post(post_json_object):
continue
publicEvent = is_public_post(post_json_object)
public_event = is_public_post(post_json_object)
postEvent = []
dayOfMonth = None
post_event = []
day_of_month = None
for tag in post_json_object['object']['tag']:
if not _is_happening_event(tag):
continue
@ -226,38 +227,38 @@ def get_todays_events(base_dir: str, nickname: str, domain: str,
# tag is an event
if not tag.get('startTime'):
continue
eventTime = \
event_time = \
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(event_time.strftime("%Y")) == year and \
int(event_time.strftime("%m")) == month_number and \
int(event_time.strftime("%d")) == day_number:
day_of_month = str(int(event_time.strftime("%d")))
if '#statuses#' in post_id:
# link to the id so that the event can be
# easily deleted
tag['post_id'] = post_id.split('#statuses#')[1]
tag['sender'] = post_id.split('#statuses#')[0]
tag['sender'] = tag['sender'].replace('#', '/')
tag['public'] = publicEvent
postEvent.append(tag)
tag['public'] = public_event
post_event.append(tag)
else:
# tag is a place
postEvent.append(tag)
if postEvent and dayOfMonth:
calendarPostIds.append(post_id)
if not events.get(dayOfMonth):
events[dayOfMonth] = []
events[dayOfMonth].append(postEvent)
post_event.append(tag)
if post_event and day_of_month:
calendar_post_ids.append(post_id)
if not events.get(day_of_month):
events[day_of_month] = []
events[day_of_month].append(post_event)
# if some posts have been deleted then regenerate the calendar file
if recreateEventsFile:
if recreate_events_file:
try:
with open(calendarFilename, 'w+') as calendarFile:
for post_id in calendarPostIds:
calendarFile.write(post_id + '\n')
with open(calendar_filename, 'w+') as calendar_file:
for post_id in calendar_post_ids:
calendar_file.write(post_id + '\n')
except OSError:
print('EX: unable to write ' + calendarFilename)
print('EX: unable to write ' + calendar_filename)
return events
@ -267,18 +268,18 @@ def day_events_check(base_dir: str, nickname: str, domain: str,
"""Are there calendar events for the given date?
"""
year = currDate.year
monthNumber = currDate.month
dayNumber = currDate.day
month_number = currDate.month
day_number = currDate.day
calendarFilename = \
calendar_filename = \
acct_dir(base_dir, nickname, domain) + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
if not os.path.isfile(calendarFilename):
'/calendar/' + str(year) + '/' + str(month_number) + '.txt'
if not os.path.isfile(calendar_filename):
return False
eventsExist = False
with open(calendarFilename, 'r') as eventsFile:
for post_id in eventsFile:
events_exist = False
with open(calendar_filename, 'r') as events_file:
for post_id in events_file:
post_id = post_id.replace('\n', '').replace('\r', '')
post_filename = locate_post(base_dir, nickname, domain, post_id)
if not post_filename:
@ -297,19 +298,19 @@ def day_events_check(base_dir: str, nickname: str, domain: str,
# tag is an event
if not tag.get('startTime'):
continue
eventTime = \
event_time = \
datetime.strptime(tag['startTime'],
"%Y-%m-%dT%H:%M:%S%z")
if int(eventTime.strftime("%d")) != dayNumber:
if int(event_time.strftime("%d")) != day_number:
continue
if int(eventTime.strftime("%m")) != monthNumber:
if int(event_time.strftime("%m")) != month_number:
continue
if int(eventTime.strftime("%Y")) != year:
if int(event_time.strftime("%Y")) != year:
continue
eventsExist = True
events_exist = True
break
return eventsExist
return events_exist
def get_this_weeks_events(base_dir: str, nickname: str, domain: str) -> {}:
@ -319,34 +320,34 @@ def get_this_weeks_events(base_dir: str, nickname: str, domain: str) -> {}:
Note: currently not used but could be with a weekly calendar screen
"""
now = datetime.now()
endOfWeek = now + timedelta(7)
end_of_week = now + timedelta(7)
year = now.year
monthNumber = now.month
month_number = now.month
calendarFilename = \
calendar_filename = \
acct_dir(base_dir, nickname, domain) + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
'/calendar/' + str(year) + '/' + str(month_number) + '.txt'
events = {}
if not os.path.isfile(calendarFilename):
if not os.path.isfile(calendar_filename):
return events
calendarPostIds = []
recreateEventsFile = False
with open(calendarFilename, 'r') as eventsFile:
for post_id in eventsFile:
calendar_post_ids = []
recreate_events_file = False
with open(calendar_filename, 'r') as events_file:
for post_id in events_file:
post_id = post_id.replace('\n', '').replace('\r', '')
post_filename = locate_post(base_dir, nickname, domain, post_id)
if not post_filename:
recreateEventsFile = True
recreate_events_file = True
continue
post_json_object = load_json(post_filename)
if not _is_happening_post(post_json_object):
continue
postEvent = []
weekDayIndex = None
post_event = []
week_day_index = None
for tag in post_json_object['object']['tag']:
if not _is_happening_event(tag):
continue
@ -355,63 +356,63 @@ def get_this_weeks_events(base_dir: str, nickname: str, domain: str) -> {}:
# tag is an event
if not tag.get('startTime'):
continue
eventTime = \
event_time = \
datetime.strptime(tag['startTime'],
"%Y-%m-%dT%H:%M:%S%z")
if eventTime >= now and eventTime <= endOfWeek:
weekDayIndex = (eventTime - now).days()
postEvent.append(tag)
if event_time >= now and event_time <= end_of_week:
week_day_index = (event_time - now).days()
post_event.append(tag)
else:
# tag is a place
postEvent.append(tag)
if postEvent and weekDayIndex:
calendarPostIds.append(post_id)
if not events.get(weekDayIndex):
events[weekDayIndex] = []
events[weekDayIndex].append(postEvent)
post_event.append(tag)
if post_event and week_day_index:
calendar_post_ids.append(post_id)
if not events.get(week_day_index):
events[week_day_index] = []
events[week_day_index].append(post_event)
# if some posts have been deleted then regenerate the calendar file
if recreateEventsFile:
if recreate_events_file:
try:
with open(calendarFilename, 'w+') as calendarFile:
for post_id in calendarPostIds:
calendarFile.write(post_id + '\n')
with open(calendar_filename, 'w+') as calendar_file:
for post_id in calendar_post_ids:
calendar_file.write(post_id + '\n')
except OSError:
print('EX: unable to write ' + calendarFilename)
print('EX: unable to write ' + calendar_filename)
return events
def get_calendar_events(base_dir: str, nickname: str, domain: str,
year: int, monthNumber: int) -> {}:
year: int, month_number: int) -> {}:
"""Retrieves calendar events
Returns a dictionary indexed by day number of lists containing
Event and Place activities
"""
calendarFilename = \
calendar_filename = \
acct_dir(base_dir, nickname, domain) + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
'/calendar/' + str(year) + '/' + str(month_number) + '.txt'
events = {}
if not os.path.isfile(calendarFilename):
if not os.path.isfile(calendar_filename):
return events
calendarPostIds = []
recreateEventsFile = False
with open(calendarFilename, 'r') as eventsFile:
for post_id in eventsFile:
calendar_post_ids = []
recreate_events_file = False
with open(calendar_filename, 'r') as events_file:
for post_id in events_file:
post_id = post_id.replace('\n', '').replace('\r', '')
post_filename = locate_post(base_dir, nickname, domain, post_id)
if not post_filename:
recreateEventsFile = True
recreate_events_file = True
continue
post_json_object = load_json(post_filename)
if not _is_happening_post(post_json_object):
continue
postEvent = []
dayOfMonth = None
post_event = []
day_of_month = None
for tag in post_json_object['object']['tag']:
if not _is_happening_event(tag):
continue
@ -420,57 +421,58 @@ def get_calendar_events(base_dir: str, nickname: str, domain: str,
# tag is an event
if not tag.get('startTime'):
continue
eventTime = \
event_time = \
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")))
postEvent.append(tag)
if int(event_time.strftime("%Y")) == year and \
int(event_time.strftime("%m")) == month_number:
day_of_month = str(int(event_time.strftime("%d")))
post_event.append(tag)
else:
# tag is a place
postEvent.append(tag)
post_event.append(tag)
if postEvent and dayOfMonth:
calendarPostIds.append(post_id)
if not events.get(dayOfMonth):
events[dayOfMonth] = []
events[dayOfMonth].append(postEvent)
if post_event and day_of_month:
calendar_post_ids.append(post_id)
if not events.get(day_of_month):
events[day_of_month] = []
events[day_of_month].append(post_event)
# if some posts have been deleted then regenerate the calendar file
if recreateEventsFile:
if recreate_events_file:
try:
with open(calendarFilename, 'w+') as calendarFile:
for post_id in calendarPostIds:
calendarFile.write(post_id + '\n')
with open(calendar_filename, 'w+') as calendar_file:
for post_id in calendar_post_ids:
calendar_file.write(post_id + '\n')
except OSError:
print('EX: unable to write ' + calendarFilename)
print('EX: unable to write ' + calendar_filename)
return events
def remove_calendar_event(base_dir: str, nickname: str, domain: str,
year: int, monthNumber: int, messageId: str) -> None:
year: int, month_number: int,
message_id: str) -> None:
"""Removes a calendar event
"""
calendarFilename = \
calendar_filename = \
acct_dir(base_dir, nickname, domain) + \
'/calendar/' + str(year) + '/' + str(monthNumber) + '.txt'
if not os.path.isfile(calendarFilename):
'/calendar/' + str(year) + '/' + str(month_number) + '.txt'
if not os.path.isfile(calendar_filename):
return
if '/' in messageId:
messageId = messageId.replace('/', '#')
if messageId not in open(calendarFilename).read():
if '/' in message_id:
message_id = message_id.replace('/', '#')
if message_id not in open(calendar_filename).read():
return
lines = None
with open(calendarFilename, 'r') as f:
lines = f.readlines()
with open(calendar_filename, 'r') as fp_cal:
lines = fp_cal.readlines()
if not lines:
return
try:
with open(calendarFilename, 'w+') as f:
with open(calendar_filename, 'w+') as fp_cal:
for line in lines:
if messageId not in line:
f.write(line)
if message_id not in line:
fp_cal.write(line)
except OSError:
print('EX: unable to write ' + calendarFilename)
print('EX: unable to write ' + calendar_filename)