Black formatter added and applied.

Author: helgihg

requirements-dev.txt

@@ -0,0 +1 @@
+black

requirements.txt

@@ -8,67 +8,79 @@ from sys import stderr
 from sys import stdout
 from urllib.request import urlopen
 
+
 def run(command):
     output = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     lines, errors = output.communicate()
 
     if output.returncode != 0:
-        print('Error: %s' % errors.strip())
+        print("Error: %s" % errors.strip())
         quit(1)
 
-    return str(lines, 'utf-8'), errors
+    return str(lines, "utf-8"), errors
+
 
 def get_chain_lines(chain_name):
-    lines, errors = run(['iptables', '-n', '-L', chain_name])
-    lines = lines.split('\n')
+    lines, errors = run(["iptables", "-n", "-L", chain_name])
+    lines = lines.split("\n")
     return lines
 
+
 def get_source_ips(ip_url):
     response = urlopen(ip_url)
-    ip_list = str(response.read(), 'utf-8').split()
+    ip_list = str(response.read(), "utf-8").split()
     return ip_list
 
+
 def get_local_ips(lines):
     result = []
 
     for line in lines:
-        parts = re.split('\s+', line)
+        parts = re.split("\s+", line)
         # Check if this is the correct line
-        if parts[0] == 'ACCEPT' and parts[1] in ['all', '0'] and parts[4] == '0.0.0.0/0':
+        if (
+            parts[0] == "ACCEPT"
+            and parts[1] in ["all", "0"]
+            and parts[4] == "0.0.0.0/0"
+        ):
             ip = parts[3]
             result.append(ip)
 
     return result
 
+
 def add_ip(ip, chain_name):
-    stdout.write('Adding IP %s...' % ip)
+    stdout.write("Adding IP %s..." % ip)
     stdout.flush()
 
-    run(['iptables', '-A', chain_name, '-s', ip, '-j', 'ACCEPT'])
+    run(["iptables", "-A", chain_name, "-s", ip, "-j", "ACCEPT"])
+
+    stdout.write(" done\n")
 
-    stdout.write(' done\n')
 
 def remove_ip(ip, chain_name):
-    stdout.write('Removing IP %s...' % ip)
+    stdout.write("Removing IP %s..." % ip)
     stdout.flush()
 
-    run(['iptables', '-D', chain_name, '-s', ip, '-j', 'ACCEPT'])
+    run(["iptables", "-D", chain_name, "-s", ip, "-j", "ACCEPT"])
+
+    stdout.write(" done\n")
 
-    stdout.write(' done\n')
 
 def error(msg):
-    stderr.write('Error: %s\n' % msg)
+    stderr.write("Error: %s\n" % msg)
     quit(1)
 
+
 def help():
-    stdout.write('Usage: rifc <network-list-url> <iptables-chain-name>\n')
-    stdout.write('\n')
+    stdout.write("Usage: rifc <network-list-url> <iptables-chain-name>\n")
+    stdout.write("\n")
     quit(2)
 
-def main(argv):
 
+def main(argv):
     if os.geteuid() != 0:
-        error('You need root access to run this program.')
+        error("You need root access to run this program.")
 
     try:
         ip_url = argv[1]
@@ -89,6 +101,6 @@ def main(argv):
         if ip not in source_ips:
             remove_ip(ip, chain_name)
 
-if __name__ == '__main__':
-    main(sys.argv)
 
+if __name__ == "__main__":
+    main(sys.argv)