62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from collections import ChainMap
|
|
import re
|
|
from re import Match
|
|
|
|
from files.helpers.const import SLURS
|
|
|
|
|
|
def create_replace_map():
|
|
dicts = [{
|
|
slur.strip(): replacer,
|
|
slur.strip().title(): replacer.title(),
|
|
slur.strip().capitalize(): replacer.capitalize(),
|
|
slur.strip().upper(): replacer.upper(),
|
|
} for (slur, replacer) in SLURS.items()]
|
|
|
|
# flattens the list of dict to a single dict
|
|
return dict(ChainMap(*dicts))
|
|
|
|
|
|
REPLACE_MAP = create_replace_map()
|
|
|
|
|
|
def create_variations_slur_regex(slur: str):
|
|
stripped = slur.strip()
|
|
variations = [stripped, stripped.upper(), stripped.capitalize()]
|
|
|
|
# capitalize multiple words if there are multiple words (just in case)
|
|
if " " in stripped:
|
|
variations.append(stripped.title())
|
|
|
|
if slur.startswith(" ") and slur.endswith(" "):
|
|
return [rf"(\s|>)({var})(\s|<)" for var in variations]
|
|
else:
|
|
return [rf"(\s|>)({var})|({var})(\s|<)" for var in variations]
|
|
|
|
|
|
def sub_matcher(match: Match):
|
|
# special case when it should match exact word
|
|
if len(match.groups()) is 3:
|
|
found = match.group(2)
|
|
replacer = REPLACE_MAP[found]
|
|
return match.group(1) + replacer + match.group(3)
|
|
|
|
else: # normal case with prefix or suffix
|
|
found = match.group(2) if (match.group(2) is not None) else match.group(3)
|
|
replacer = REPLACE_MAP[found]
|
|
return (match.group(1) or '') + replacer + (match.group(4) or '')
|
|
|
|
|
|
def censor_slurs(body: str, logged_user):
|
|
if logged_user and not logged_user.slurreplacer:
|
|
return body
|
|
|
|
for (slur, replace) in SLURS.items():
|
|
for variation in create_variations_slur_regex(slur):
|
|
try:
|
|
body = re.sub(variation, sub_matcher, body)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
return body
|