sync_soho_doorway.py

#!/usr/bin/env python3

import concurrent.futures
import glob
import time
import os
import subprocess
from typing import Generator


class SohoDoorway:
    def __init__(self):
        self.log_file = os.getcwd() + "/sync_doorway.log"
        self.non_telstra_organisations =  os.getcwd() + "/sync_non_telstra_organisations.txt"
        self.summary = os.getcwd() + "/sync_summary.txt"

    def read_ip_file(self) -> Generator:
        try:
            pwd = os.getcwd()
            file = glob.glob(pwd + "/*IPranges.txt")
            if file:
                for f in file:
                    filer = f
                    with open(filer, "r") as file:
                        lines = (f for f in file.readlines())
                        return lines
            else:
                raise Exception
        except Exception as e:
            print("Error! in read_ip_file: " + str(e))

    def make_cmds(self, ips):
        cmds = ["/bin/whois " + i for i in ips]
        return cmds

    def who_are_you(self, cmd):
        try:
            size = len(cmd)
            i: int = 1
            t3 = time.perf_counter()
            for c in cmd:
                t1 = time.perf_counter()
                print("\nfetching whois record: " + str(i) + " of " + str(size))
                print("running " + str(c))
                whois = subprocess.run([c], shell=True, stdout=subprocess.PIPE)
                res = whois.stdout.decode().strip()
                with open(self.log_file, "a") as file:
                    file.writelines(res)
                t2 = time.perf_counter()
                t4 = time.perf_counter()
                print("completed in: " + str(round(t2-t1, 4)) + "sec")
                print("Total time: " + str(round(t4-t3, 4)) + "sec")
                i += 1
        except Exception as e:
            print("Error! in who_are_you: " + str(e))

    def clearfile(self):
        os.system("clear")
        print("Processing..\n")
        files = [self.log_file, self.non_telstra_organisations, self.summary]
        for f in files:
            with open(f, "w") as file:
                file.write("")

    def paperback(self, message, file, mode):
        with open(file, "a") as writable:
            if "list" in mode:
                [writable.write(m + "\n") for m in message]
            elif "str" in mode:
                writable.write(message)

    def parselog(self):
        with open(self.log_file, "r") as file:
            lines = [f for f in file.readlines()]
            for l in lines:
                if ("netname:" in l or "descr:" in l  or "org-name" in l
                    or "OrgName" in l or "NetName" in l):
                    print(l)
                    self.paperback(l, self.summary, "str")
                elif "NetRange:" in l or "inetnum:" in l:
                    message = "\n----------------------------------\n" + l
                    print(message)
                    self.paperback(message, self.summary, "str")
            nontelstra = [l.strip() for l in lines if "etname" in l or "OrgName" in l or "org-name" in l
            and not "telstra" in l and not "TELSTRAINTERNET3-AU" in l
            and not "Telstra" in l and not "TELSTRA" in l]
            nontelstra = set(nontelstra)
            nontelstra = [n.strip() for n in nontelstra if "TELSTRA" not in n and "remarks" not in n]
            print("\n\n--------------------------------\n"
                  "Non-Telstra entities in IP list:" +
                  "\n--------------------------------\n")
            [print(n) for n in nontelstra]
            self.paperback(nontelstra, self.non_telstra_organisations, "list")

    def controller(self):
        try:
            ips = self.read_ip_file()
            ips = [i.split(" ") for i in ips if i]
            ips = [i.strip("\t").strip("\n").strip(" ").rstrip("\n") for i in ips for i in i if i]
            ips = [i for i in ips if i]
            ips = [i.split("/") for i in ips]
            ips = [i[0] for i in ips]
            cmds = self.make_cmds(ips)
            self.who_are_you(cmds)
            self.parselog()
            print("\nCompleted Successfully. \n\nDetails are in " + self.log_file + "\n")
        except Exception as e:
            print("Error! controller: " + str(e))

def main():
    new_soho = SohoDoorway()
    new_soho.clearfile()
    new_soho.controller()

if __name__ == '__main__':
    main()