epicyon/followingCalendar.py

183 lines
6.8 KiB
Python

__filename__ = "followingCalendar.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.3.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@libreserver.org"
__status__ = "Production"
__module_group__ = "Calendar"
import os
def _text_in_file2(text: str, filename: str,
case_sensitive: bool = True) -> bool:
"""is the given text in the given file?
"""
if not case_sensitive:
text = text.lower()
try:
with open(filename, 'r', encoding='utf-8') as file:
content = file.read()
if content:
if not case_sensitive:
content = content.lower()
if text in content:
return True
except OSError:
print('EX: unable to find text in missing file ' + filename)
return False
def _dir_acct(base_dir: str, nickname: str, domain: str) -> str:
"""Returns the directory of an account
"""
return base_dir + '/accounts/' + nickname + '@' + domain
def _port_domain_remove(domain: str) -> str:
"""If the domain has a port appended then remove it
eg. mydomain.com:80 becomes mydomain.com
same as remove_domain_port in utils.py
"""
if ':' in domain:
if domain.startswith('did:'):
return domain
domain = domain.split(':')[0]
return domain
def receiving_calendar_events(base_dir: str, nickname: str, domain: str,
following_nickname: str,
following_domain: str) -> bool:
"""Returns true if receiving calendar events from the given
account from following.txt
"""
if following_nickname == nickname and following_domain == domain:
# reminder post
return True
calendar_filename = \
_dir_acct(base_dir, nickname, domain) + '/followingCalendar.txt'
handle = following_nickname + '@' + following_domain
if not os.path.isfile(calendar_filename):
following_filename = \
_dir_acct(base_dir, nickname, domain) + '/following.txt'
if not os.path.isfile(following_filename):
return False
# create a new calendar file from the following file
following_handles = None
try:
with open(following_filename, 'r',
encoding='utf-8') as following_file:
following_handles = following_file.read()
except OSError:
print('EX: receiving_calendar_events ' + following_filename)
if following_handles:
try:
with open(calendar_filename, 'w+',
encoding='utf-8') as fp_cal:
fp_cal.write(following_handles)
except OSError:
print('EX: receiving_calendar_events 2 ' + calendar_filename)
return _text_in_file2(handle + '\n', calendar_filename)
def _receive_calendar_events(base_dir: str, nickname: str, domain: str,
following_nickname: str,
following_domain: str,
add: bool) -> None:
"""Adds or removes a handle from the following.txt list into a list
indicating whether to receive calendar events from that account
"""
# check that a following file exists
domain = _port_domain_remove(domain)
following_filename = \
_dir_acct(base_dir, nickname, domain) + '/following.txt'
if not os.path.isfile(following_filename):
print("WARN: following.txt doesn't exist for " +
nickname + '@' + domain)
return
handle = following_nickname + '@' + following_domain
# check that you are following this handle
if not _text_in_file2(handle + '\n', following_filename):
print('WARN: ' + handle + ' is not in ' + following_filename)
return
calendar_filename = \
_dir_acct(base_dir, nickname, domain) + '/followingCalendar.txt'
# get the contents of the calendar file, which is
# a set of handles
following_handles = ''
if os.path.isfile(calendar_filename):
print('Calendar file exists')
try:
with open(calendar_filename, 'r',
encoding='utf-8') as calendar_file:
following_handles = calendar_file.read()
except OSError:
print('EX: _receive_calendar_events ' + calendar_filename)
else:
# create a new calendar file from the following file
print('Creating calendar file ' + calendar_filename)
following_handles = ''
try:
with open(following_filename, 'r',
encoding='utf-8') as following_file:
following_handles = following_file.read()
except OSError:
print('EX: _receive_calendar_events 2 ' + calendar_filename)
if add:
try:
with open(calendar_filename, 'w+',
encoding='utf-8') as fp_cal:
fp_cal.write(following_handles + handle + '\n')
except OSError:
print('EX: unable to write ' + calendar_filename)
# already in the calendar file?
if handle + '\n' in following_handles:
print(handle + ' exists in followingCalendar.txt')
if add:
# already added
return
# remove from calendar file
following_handles = following_handles.replace(handle + '\n', '')
try:
with open(calendar_filename, 'w+',
encoding='utf-8') as fp_cal:
fp_cal.write(following_handles)
except OSError:
print('EX: _receive_calendar_events 3 ' + calendar_filename)
else:
print(handle + ' not in followingCalendar.txt')
# not already in the calendar file
if add:
# append to the list of handles
following_handles += handle + '\n'
try:
with open(calendar_filename, 'w+',
encoding='utf-8') as fp_cal:
fp_cal.write(following_handles)
except OSError:
print('EX: _receive_calendar_events 4 ' + calendar_filename)
def add_person_to_calendar(base_dir: str, nickname: str, domain: str,
following_nickname: str,
following_domain: str) -> None:
"""Add a person to the calendar
"""
_receive_calendar_events(base_dir, nickname, domain,
following_nickname, following_domain, True)
def remove_person_from_calendar(base_dir: str, nickname: str, domain: str,
following_nickname: str,
following_domain: str) -> None:
"""Remove a person from the calendar
"""
_receive_calendar_events(base_dir, nickname, domain,
following_nickname, following_domain, False)