aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authornico <nico@magicbroccoli.de>2018-12-21 20:49:15 +0100
committernico <nico@magicbroccoli.de>2018-12-21 20:49:15 +0100
commit92efbd421ea11200aded351282e900986492cbbf (patch)
tree52775d142d98fb40855b5a019709d0dcacad1623
Initial Commit
+ request working + ejabberd working
-rw-r--r--main.py112
1 files changed, 112 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..6608a85
--- /dev/null
+++ b/main.py
@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# workflow
+# start options main.py --ejabberd/prosody --dry-run --outfile file
+
+import requests
+import sys
+import os
+import argparse
+import yaml
+
+
+class BlacklistImporter:
+ def __init__(self, args):
+ self.server = args.software
+ self.outfile = args.outfile
+ self.dryrun = args.dryrun
+ self.url = "https://raw.githubusercontent.com/JabberSPAM/blacklist/master/blacklist.txt"
+ self.blacklist = None
+
+ def request(self):
+ # check if etag header is present if not set local_etag to ""
+ if os.path.isfile(".etag"):
+ with open(".etag") as file:
+ local_etag = file.read()
+ else:
+ local_etag = ""
+
+ with requests.Session() as s:
+ # head request to check etag
+ head = s.head(self.url)
+ etag = head.headers['etag']
+
+ # compare etag with local_etag if they match up no request is made
+ if local_etag == etag:
+ with open("blacklist.txt", "r") as file:
+ self.blacklist = file.readline()
+
+ # if the connection is not possible use cached xml if present
+ elif os.path.isfile("blacklist.txt") and head.status_code != 200:
+ with open("blacklist.txt", "r") as file:
+ self.blacklist = file.readline()
+
+ # in any other case request a new file
+ else:
+ r = s.get(self.url)
+ r.encoding = 'utf-8'
+ local_etag = head.headers['etag']
+
+ with open("blacklist.txt", "w") as file:
+ file.write(r.content.decode())
+
+ with open('.etag', 'w') as string:
+ string.write(local_etag)
+
+ def main(self):
+ # first check if blacklist is updated
+ self.request()
+
+ if self.dryrun:
+ # only output the selected software and outfile
+ print("server software selected: %s" % self.server)
+ print("outfile selected: %s" % self.outfile)
+
+ if self.server == "ejabberd":
+ # select ejabberd processing
+ self.ejabberd()
+
+ elif self.server == "prosody":
+ # select prosody processing
+ self.prosody()
+ else:
+ # in any other case exit
+ sys.exit(3)
+
+ def ejabberd(self):
+ # check if file was altered
+ local_file = yaml.load(open(self.outfile, "r"))
+
+ remote_file = {
+ "acl": {
+ "spamblacklist": {
+ "server": []
+ }
+ }
+ }
+
+ for entry in self.blacklist.split():
+ remote_file["acl"]["spamblacklist"]["server"].append(entry)
+
+ if self.dryrun:
+ print(yaml.dump(remote_file))
+
+ elif local_file != remote_file:
+ yaml.dump(remote_file, open(self.outfile, "w"))
+
+ def prosody(self):
+ pass
+
+
+if __name__ == "__main__":
+ parser = argparse.ArgumentParser()
+ parser.add_argument('-e', '--ejabberd', help='set server software to ejabberd', action='store_const', dest='software',
+ const="ejabberd", default=None)
+ parser.add_argument('-p', '--prosody', help='set server software to prosody', action='store_const', dest='software',
+ const="prosody", default=None)
+ parser.add_argument('-o', '--outfile', help='set path to output file', dest='outfile', default=None)
+ parser.add_argument('--dry-run', help='perform only a dry run', action='store_true', dest='dryrun', default=False)
+ args = parser.parse_args()
+
+ BlacklistImporter(args).main()