""" utils.py - Utility functions for the project. """ import logging def postprocess(text: str): """ postprocess - remove common values in scraped dataset Args: text (str): the text to postprocess """ replacements = { "ENA": "", "Enron": "", "Enron Corporation": "", "Sony Pictures Entertainment": "", "Columbia Pictures": "", "Sony": "", "Columbia": "", "Hillary": "Jane", "Clinton": "Smith", "Amy": "Jane", "Sara": "Jane", "Harambe": "Jane", "Pascal": "", } # replace common values, also check lowercase for k, v in replacements.items(): text = text.replace(k, v) text = text.replace(k.lower(), v) return text def clear(text, verbose=False, **kwargs): """for use with buttons""" if verbose: logging.info(f"Clearing text: {text}") return "" def make_email_link( subject: str = "Email subject - This was generated by Postbot", link_text: str = "click to open in your email client", body: str = None, ): """ email_link - generate an email link Args: subject (str, optional): the subject of the email. Defaults to "Email subject - This was generated by Postbot". link_text (str, optional): the text of the link. Defaults to "click to open in your email client". body (_type_, optional): the body of the email. Defaults to None. Returns: str: the email link """ if body is None: body = "hmm - no body. replace me" return f'{link_text}'