indymedia
/
epicyon
mirror of https://gitlab.com/bashrc2/epicyon
7
4
Fork 1
Code Issues 14 Releases Wiki Activity

Snake case

merge-requests/30/head
Bob Mottram 2021-12-31 10:49:09 +00:00
parent 3dfbb2bfc8
commit cafc946df7
1 changed files with 115 additions and 95 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

210
roles.py
View File

@ -21,9 +21,8 @@ def _clear_role_status(base_dir: str, role: str) -> None:
rarely when roles are appointed or removed rarely when roles are appointed or removed
""" """
directory = os.fsencode(base_dir + '/accounts/') directory = os.fsencode(base_dir + '/accounts/')
for f in os.scandir(directory): for fname in os.scandir(directory):
f = f.name filename = os.fsdecode(fname.name)
filename = os.fsdecode(f)
if '@' not in filename: if '@' not in filename:
continue continue
if not filename.endswith(".json"): if not filename.endswith(".json"):
@ -34,10 +33,10 @@ def _clear_role_status(base_dir: str, role: str) -> None:
actor_json = load_json(filename) actor_json = load_json(filename)
if not actor_json: if not actor_json:
continue continue
rolesList = get_actor_roles_list(actor_json) roles_list = get_actor_roles_list(actor_json)
if role in rolesList: if role in roles_list:
rolesList.remove(role) roles_list.remove(role)
set_rolesFromList(actor_json, rolesList) set_rolesFromList(actor_json, roles_list)
save_json(actor_json, filename) save_json(actor_json, filename)
@ -74,53 +73,74 @@ def clear_moderator_status(base_dir: str) -> None:
def _add_role(base_dir: str, nickname: str, domain: str, def _add_role(base_dir: str, nickname: str, domain: str,
roleFilename: str) -> None: role_filename: str) -> None:
"""Adds a role nickname to the file. """Adds a role nickname to the file.
This is a file containing the nicknames of accounts having this role This is a file containing the nicknames of accounts having this role
""" """
domain = remove_domain_port(domain) domain = remove_domain_port(domain)
roleFile = base_dir + '/accounts/' + roleFilename role_file = base_dir + '/accounts/' + role_filename
if os.path.isfile(roleFile): if os.path.isfile(role_file):
# is this nickname already in the file? # is this nickname already in the file?
with open(roleFile, 'r') as f:
lines = f.readlines() lines = []
for roleNickname in lines: try:
roleNickname = roleNickname.strip('\n').strip('\r') with open(role_file, 'r') as fp_role:
if roleNickname == nickname: lines = fp_role.readlines()
except OSError:
print('EX: _add_role, failed to read roles file ' + role_file)
for role_nickname in lines:
role_nickname = role_nickname.strip('\n').strip('\r')
if role_nickname == nickname:
return return
lines.append(nickname) lines.append(nickname)
with open(roleFile, 'w+') as f:
for roleNickname in lines: try:
roleNickname = roleNickname.strip('\n').strip('\r') with open(role_file, 'w+') as fp_role:
if len(roleNickname) < 2: for role_nickname in lines:
continue role_nickname = role_nickname.strip('\n').strip('\r')
if os.path.isdir(base_dir + '/accounts/' + if len(role_nickname) < 2:
roleNickname + '@' + domain): continue
f.write(roleNickname + '\n') if os.path.isdir(base_dir + '/accounts/' +
role_nickname + '@' + domain):
fp_role.write(role_nickname + '\n')
except OSError:
print('EX: _add_role, failed to write roles file1 ' + role_file)
else: else:
with open(roleFile, 'w+') as f: try:
accountDir = acct_dir(base_dir, nickname, domain) with open(role_file, 'w+') as fp_role:
if os.path.isdir(accountDir): account_dir = acct_dir(base_dir, nickname, domain)
f.write(nickname + '\n') if os.path.isdir(account_dir):
fp_role.write(nickname + '\n')
except OSError:
print('EX: _add_role, failed to write roles file2 ' + role_file)
def _remove_role(base_dir: str, nickname: str, roleFilename: str) -> None: def _remove_role(base_dir: str, nickname: str, role_filename: str) -> None:
"""Removes a role nickname from the file. """Removes a role nickname from the file.
This is a file containing the nicknames of accounts having this role This is a file containing the nicknames of accounts having this role
""" """
roleFile = base_dir + '/accounts/' + roleFilename role_file = base_dir + '/accounts/' + role_filename
if not os.path.isfile(roleFile): if not os.path.isfile(role_file):
return return
with open(roleFile, 'r') as f:
lines = f.readlines() try:
with open(roleFile, 'w+') as f: with open(role_file, 'r') as fp_role:
for roleNickname in lines: lines = fp_role.readlines()
roleNickname = roleNickname.strip('\n').strip('\r') except OSError:
if len(roleNickname) > 1 and roleNickname != nickname: print('EX: _remove_role, failed to read roles file ' + role_file)
f.write(roleNickname + '\n')
try:
with open(role_file, 'w+') as fp_role:
for role_nickname in lines:
role_nickname = role_nickname.strip('\n').strip('\r')
if len(role_nickname) > 1 and role_nickname != nickname:
fp_role.write(role_nickname + '\n')
except OSError:
print('EX: _remove_role, failed to regenerate roles file ' + role_file)
def _set_actor_role(actor_json: {}, roleName: str) -> bool: def _set_actor_role(actor_json: {}, role_name: str) -> bool:
"""Sets a role for an actor """Sets a role for an actor
""" """
if not actor_json.get('hasOccupation'): if not actor_json.get('hasOccupation'):
@ -130,35 +150,35 @@ def _set_actor_role(actor_json: {}, roleName: str) -> bool:
# occupation category from www.onetonline.org # occupation category from www.onetonline.org
category = None category = None
if 'admin' in roleName: if 'admin' in role_name:
category = '15-1299.01' category = '15-1299.01'
elif 'moderator' in roleName: elif 'moderator' in role_name:
category = '11-9199.02' category = '11-9199.02'
elif 'editor' in roleName: elif 'editor' in role_name:
category = '27-3041.00' category = '27-3041.00'
elif 'counselor' in roleName: elif 'counselor' in role_name:
category = '23-1022.00' category = '23-1022.00'
elif 'artist' in roleName: elif 'artist' in role_name:
category = '27-1024.00' category = '27-1024.00'
if not category: if not category:
return False return False
for index in range(len(actor_json['hasOccupation'])): for index in range(len(actor_json['hasOccupation'])):
occupationItem = actor_json['hasOccupation'][index] occupation_item = actor_json['hasOccupation'][index]
if not isinstance(occupationItem, dict): if not isinstance(occupation_item, dict):
continue continue
if not occupationItem.get('@type'): if not occupation_item.get('@type'):
continue continue
if occupationItem['@type'] != 'Role': if occupation_item['@type'] != 'Role':
continue continue
if occupationItem['hasOccupation']['name'] == roleName: if occupation_item['hasOccupation']['name'] == role_name:
return True return True
statusNumber, published = get_status_number() _, published = get_status_number()
newRole = { new_role = {
"@type": "Role", "@type": "Role",
"hasOccupation": { "hasOccupation": {
"@type": "Occupation", "@type": "Occupation",
"name": roleName, "name": role_name,
"description": "Fediverse instance role", "description": "Fediverse instance role",
"occupationLocation": { "occupationLocation": {
"@type": "City", "@type": "City",
@ -178,28 +198,28 @@ def _set_actor_role(actor_json: {}, roleName: str) -> bool:
}, },
"startDate": published "startDate": published
} }
actor_json['hasOccupation'].append(newRole) actor_json['hasOccupation'].append(new_role)
return True return True
def set_rolesFromList(actor_json: {}, rolesList: []) -> None: def set_rolesFromList(actor_json: {}, roles_list: []) -> None:
"""Sets roles from a list """Sets roles from a list
""" """
# clear Roles from the occupation list # clear Roles from the occupation list
emptyRolesList = [] empty_roles_list = []
for occupationItem in actor_json['hasOccupation']: for occupation_item in actor_json['hasOccupation']:
if not isinstance(occupationItem, dict): if not isinstance(occupation_item, dict):
continue continue
if not occupationItem.get('@type'): if not occupation_item.get('@type'):
continue continue
if occupationItem['@type'] == 'Role': if occupation_item['@type'] == 'Role':
continue continue
emptyRolesList.append(occupationItem) empty_roles_list.append(occupation_item)
actor_json['hasOccupation'] = emptyRolesList actor_json['hasOccupation'] = empty_roles_list
# create the new list # create the new list
for roleName in rolesList: for role_name in roles_list:
_set_actor_role(actor_json, roleName) _set_actor_role(actor_json, role_name)
def get_actor_roles_list(actor_json: {}) -> []: def get_actor_roles_list(actor_json: {}) -> []:
@ -209,18 +229,18 @@ def get_actor_roles_list(actor_json: {}) -> []:
return [] return []
if not isinstance(actor_json['hasOccupation'], list): if not isinstance(actor_json['hasOccupation'], list):
return [] return []
rolesList = [] roles_list = []
for occupationItem in actor_json['hasOccupation']: for occupation_item in actor_json['hasOccupation']:
if not isinstance(occupationItem, dict): if not isinstance(occupation_item, dict):
continue continue
if not occupationItem.get('@type'): if not occupation_item.get('@type'):
continue continue
if occupationItem['@type'] != 'Role': if occupation_item['@type'] != 'Role':
continue continue
roleName = occupationItem['hasOccupation']['name'] role_name = occupation_item['hasOccupation']['name']
if roleName not in rolesList: if role_name not in roles_list:
rolesList.append(roleName) roles_list.append(role_name)
return rolesList return roles_list
def set_role(base_dir: str, nickname: str, domain: str, def set_role(base_dir: str, nickname: str, domain: str,
@ -231,47 +251,47 @@ def set_role(base_dir: str, nickname: str, domain: str,
# avoid giant strings # avoid giant strings
if len(role) > 128: if len(role) > 128:
return False return False
actorFilename = acct_dir(base_dir, nickname, domain) + '.json' actor_filename = acct_dir(base_dir, nickname, domain) + '.json'
if not os.path.isfile(actorFilename): if not os.path.isfile(actor_filename):
return False return False
roleFiles = { role_files = {
"moderator": "moderators.txt", "moderator": "moderators.txt",
"editor": "editors.txt", "editor": "editors.txt",
"counselor": "counselors.txt", "counselor": "counselors.txt",
"artist": "artists.txt" "artist": "artists.txt"
} }
actor_json = load_json(actorFilename) actor_json = load_json(actor_filename)
if actor_json: if actor_json:
if not actor_json.get('hasOccupation'): if not actor_json.get('hasOccupation'):
return False return False
rolesList = get_actor_roles_list(actor_json) roles_list = get_actor_roles_list(actor_json)
actorChanged = False actor_changed = False
if role: if role:
# add the role # add the role
if roleFiles.get(role): if role_files.get(role):
_add_role(base_dir, nickname, domain, roleFiles[role]) _add_role(base_dir, nickname, domain, role_files[role])
if role not in rolesList: if role not in roles_list:
rolesList.append(role) roles_list.append(role)
rolesList.sort() roles_list.sort()
set_rolesFromList(actor_json, rolesList) set_rolesFromList(actor_json, roles_list)
actorChanged = True actor_changed = True
else: else:
# remove the role # remove the role
if roleFiles.get(role): if role_files.get(role):
_remove_role(base_dir, nickname, roleFiles[role]) _remove_role(base_dir, nickname, role_files[role])
if role in rolesList: if role in roles_list:
rolesList.remove(role) roles_list.remove(role)
set_rolesFromList(actor_json, rolesList) set_rolesFromList(actor_json, roles_list)
actorChanged = True actor_changed = True
if actorChanged: if actor_changed:
save_json(actor_json, actorFilename) save_json(actor_json, actor_filename)
return True return True
def actor_has_role(actor_json: {}, roleName: str) -> bool: def actor_has_role(actor_json: {}, role_name: str) -> bool:
"""Returns true if the given actor has the given role """Returns true if the given actor has the given role
""" """
rolesList = get_actor_roles_list(actor_json) roles_list = get_actor_roles_list(actor_json)
return roleName in rolesList return role_name in roles_list