diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 152 |
1 files changed, 0 insertions, 152 deletions
diff --git a/main.py b/main.py deleted file mode 100644 index e39d82b..0000000 --- a/main.py +++ /dev/null @@ -1,152 +0,0 @@ -#!/usr/bin/env python3 -# -*- coding: utf-8 -*- - -import argparse -import os -import subprocess -import sys - -import requests -from ruamel.yaml import YAML, scalarstring - - -class BlacklistImporter: - def __init__(self, args): - self.outfile = args.outfile - self.dryrun = args.dryrun - self.path = os.path.dirname(__file__) - self.url = "https://raw.githubusercontent.com/JabberSPAM/blacklist/master/blacklist.txt" - self.blacklist = "" - self.change = False - - def request(self): - """ - determine if the download is required - """ - etag_path = "/".join([self.path, ".etag"]) - blacklist_path = "/".join([self.path, "blacklist.txt"]) - - # check if etag header is present if not set local_etag to "" - if os.path.isfile(etag_path): - # catch special case were etag file is present and blacklist.txt is not - if not os.path.isfile(blacklist_path): - local_etag = "" - else: - # if both files are present continue normally - with open(etag_path, "r") as local_file: - local_etag = local_file.read() - else: - local_etag = "" - - with requests.Session() as s: - # head request to check etag - head = s.head(self.url) - etag = head.headers['etag'] - - # if etags match up or if the connection is not possible fall back to local cache - if local_etag == etag or head.status_code != 200: - # if local cache is present overwrite blacklist var - if os.path.isfile(blacklist_path): - with open(blacklist_path, "r", encoding="utf-8") as local_file: - self.blacklist = local_file.read() - - # in any other case request a new file - else: - r = s.get(self.url) - r.encoding = 'utf-8' - local_etag = head.headers['etag'] - self.blacklist = r.content.decode() - - with open(blacklist_path, "w") as local_file: - local_file.write(self.blacklist) - - with open(etag_path, 'w') as local_file: - local_file.write(local_etag) - - def main(self): - # first check if blacklist is updated - self.request() - - # only output the selected outfile - if self.dryrun: - print("outfile selected: %s" % self.outfile) - - # blacklist processing - self.process() - - # reload config if changes have been applied - if self.change: - # catch ejabberdctl missing - if os.path.isfile('/usr/sbin/ejabberdctl'): - subprocess.call(['/usr/sbin/ejabberdctl', 'reload_config'], shell=False) - - # report missing ejabberdctl reload_config - else: - print('/usr/sbin/ejabberdctl was not found', file=sys.stderr) - print('blacklist changes have been applied\nejabberd config was not reloaded', file=sys.stderr) - sys.exit(1) - - def process(self): - """ - function to build and compare the local yaml file to the remote file - if the remote file is different, the local file gets overwritten - """ - # init new YAML variable - local_file = YAML(typ="safe") - - # None catch - if self.outfile is not None: - # prevent FileNotFoundError on first run or file missing - if os.path.isfile(self.outfile): - local_file = local_file.load(open(self.outfile, "r", encoding="utf-8")) - - # blacklist frame - remote_file = { - "acl": { - "spamblacklist": { - "server": [] - } - } - } - - # build the blacklist with the given frame to compare to local blacklist - for entry in self.blacklist.split(): - entry = scalarstring.DoubleQuotedScalarString(entry) - remote_file["acl"]["spamblacklist"]["server"].append(entry) - - yml = YAML() - yml.indent(offset=2) - yml.default_flow_style = False - - # if dry-run true print expected content - if self.dryrun: - yml.dump(remote_file, sys.stdout) - - # only if the local_file and remote_file are unequal write new file - elif local_file != remote_file: - - # prevent FileNotFoundError if self.outfile is not assigned - if self.outfile is None: - print("no outfile assigned", file=sys.stderr) - print(parser.format_help(), file=sys.stderr) - sys.exit(2) - - # proceed to update the defined outfile - elif self.outfile is not None: - self.change = True - yml.dump(remote_file, open(self.outfile, "w")) - - # if that's impossible break and display help message - else: - print(parser.format_help(), file=sys.stderr) - sys.exit(1) - - -if __name__ == "__main__": - parser = argparse.ArgumentParser() - parser.add_argument('-o', '--outfile', help='set path to output file', dest='outfile', default=None) - parser.add_argument('-dr', '--dry-run', help='perform a dry run', action='store_true', dest='dryrun', default=False) - args = parser.parse_args() - - # run - BlacklistImporter(args).main() |