epicyon/notifyOnPost.py

110 lines
4.2 KiB
Python
Raw Normal View History

__filename__ = "notifyOnPost.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.2.0"
__maintainer__ = "Bob Mottram"
2021-09-10 16:14:50 +00:00
__email__ = "bob@libreserver.org"
__status__ = "Production"
__module_group__ = "Calendar"
import os
2021-12-26 18:17:37 +00:00
from utils import remove_domain_port
2021-12-26 12:02:29 +00:00
from utils import acct_dir
2021-12-29 21:55:09 +00:00
def _notify_on_post_arrival(base_dir: str, nickname: str, domain: str,
2022-01-03 12:41:06 +00:00
following_nickname: str,
following_domain: str,
2021-12-29 21:55:09 +00:00
add: bool) -> None:
"""Adds or removes a handle from the following.txt list into a list
indicating whether to notify when a new post arrives from that account
"""
# check that a following file exists
2021-12-26 18:17:37 +00:00
domain = remove_domain_port(domain)
2022-01-03 12:41:06 +00:00
following_filename = \
acct_dir(base_dir, nickname, domain) + '/following.txt'
if not os.path.isfile(following_filename):
print("WARN: following.txt doesn't exist for " +
nickname + '@' + domain)
return
2022-01-03 12:41:06 +00:00
handle = following_nickname + '@' + following_domain
# check that you are following this handle
2022-01-03 12:41:06 +00:00
if handle + '\n' not in open(following_filename).read():
print('WARN: ' + handle + ' is not in ' + following_filename)
return
2022-01-03 12:41:06 +00:00
notify_on_post_filename = \
2021-12-26 12:02:29 +00:00
acct_dir(base_dir, nickname, domain) + '/notifyOnPost.txt'
# get the contents of the notifyOnPost file, which is
# a set of handles
2022-01-03 12:41:06 +00:00
following_handles = ''
if os.path.isfile(notify_on_post_filename):
print('notify file exists')
2022-01-03 12:41:06 +00:00
with open(notify_on_post_filename, 'r') as calendar_file:
following_handles = calendar_file.read()
else:
# create a new notifyOnPost file from the following file
2022-01-03 12:41:06 +00:00
print('Creating notifyOnPost file ' + notify_on_post_filename)
following_handles = ''
with open(following_filename, 'r') as following_file:
following_handles = following_file.read()
if add:
2022-01-03 12:41:06 +00:00
with open(notify_on_post_filename, 'w+') as fp_notify:
fp_notify.write(following_handles + handle + '\n')
# already in the notifyOnPost file?
2022-01-03 12:41:06 +00:00
if handle + '\n' in following_handles:
print(handle + ' exists in notifyOnPost.txt')
if add:
# already added
return
# remove from calendar file
2022-01-03 12:41:06 +00:00
following_handles = following_handles.replace(handle + '\n', '')
with open(notify_on_post_filename, 'w+') as fp_notify:
fp_notify.write(following_handles)
else:
print(handle + ' not in notifyOnPost.txt')
# not already in the notifyOnPost file
if add:
# append to the list of handles
2022-01-03 12:41:06 +00:00
following_handles += handle + '\n'
with open(notify_on_post_filename, 'w+') as fp_notify:
fp_notify.write(following_handles)
2021-12-29 21:55:09 +00:00
def add_notify_on_post(base_dir: str, nickname: str, domain: str,
2022-01-03 12:41:06 +00:00
following_nickname: str,
following_domain: str) -> None:
"""Add a notification
"""
2021-12-29 21:55:09 +00:00
_notify_on_post_arrival(base_dir, nickname, domain,
2022-01-03 12:41:06 +00:00
following_nickname, following_domain, True)
2021-12-29 21:55:09 +00:00
def remove_notify_on_post(base_dir: str, nickname: str, domain: str,
2022-01-03 12:41:06 +00:00
following_nickname: str,
following_domain: str) -> None:
"""Remove a notification
"""
2021-12-29 21:55:09 +00:00
_notify_on_post_arrival(base_dir, nickname, domain,
2022-01-03 12:41:06 +00:00
following_nickname, following_domain, False)
2021-12-29 21:55:09 +00:00
def notify_when_person_posts(base_dir: str, nickname: str, domain: str,
2022-01-03 12:41:06 +00:00
following_nickname: str,
following_domain: str) -> bool:
"""Returns true if receiving notifications when the given publishes a post
"""
2022-01-03 12:41:06 +00:00
if following_nickname == nickname and following_domain == domain:
return False
2022-01-03 12:41:06 +00:00
notify_on_post_filename = \
2021-12-26 12:02:29 +00:00
acct_dir(base_dir, nickname, domain) + '/notifyOnPost.txt'
2022-01-03 12:41:06 +00:00
handle = following_nickname + '@' + following_domain
if not os.path.isfile(notify_on_post_filename):
# create a new notifyOnPost file
2022-01-03 12:41:06 +00:00
with open(notify_on_post_filename, 'w+') as fp_notify:
fp_notify.write('')
return handle + '\n' in open(notify_on_post_filename).read()