__filename__ = "filters.py" __author__ = "Bob Mottram" __license__ = "AGPL3+" __version__ = "1.1.0" __maintainer__ = "Bob Mottram" __email__ = "bob@freedombone.net" __status__ = "Production" import os def addFilter(baseDir: str, nickname: str, domain: str, words: str) -> bool: """Adds a filter for particular words within the content of a incoming posts """ filtersFilename = baseDir + '/accounts/' + \ nickname + '@' + domain + '/filters.txt' if os.path.isfile(filtersFilename): if words in open(filtersFilename).read(): return False filtersFile = open(filtersFilename, "a+") filtersFile.write(words + '\n') filtersFile.close() return True def removeFilter(baseDir: str, nickname: str, domain: str, words: str) -> bool: """Removes a word filter """ filtersFilename = baseDir + '/accounts/' + \ nickname + '@' + domain + '/filters.txt' if os.path.isfile(filtersFilename): if words in open(filtersFilename).read(): with open(filtersFilename, 'r') as fp: with open(filtersFilename + '.new', 'w') as fpnew: for line in fp: line = line.replace('\n', '') if line != words: fpnew.write(line + '\n') if os.path.isfile(filtersFilename + '.new'): os.rename(filtersFilename + '.new', filtersFilename) return True return False def isTwitterPost(content: str) -> bool: """Returns true if the given post content is a retweet or twitter crosspost """ if '/twitter.' in content or '@twitter.' in content: return True elif '>RT <' in content: return True return False def isFiltered(baseDir: str, nickname: str, domain: str, content: str) -> bool: """Should the given content be filtered out? This is a simple type of filter which just matches words, not a regex You can add individual words or use word1+word2 to indicate that two words must be present although not necessarily adjacent """ # optionally remove retweets removeTwitter = baseDir + '/accounts/' + \ nickname + '@' + domain + '/.removeTwitter' if os.path.isfile(removeTwitter): if isTwitterPost(content): return True filtersFilename = baseDir + '/accounts/' + \ nickname + '@' + domain + '/filters.txt' if os.path.isfile(filtersFilename): with open(filtersFilename, 'r') as fp: for line in fp: filterStr = line.replace('\n', '') if '+' not in filterStr: if filterStr in content: return True else: filterWords = filterStr.replace('"', '').split('+') for word in filterWords: if word not in content: return False return True return False