Function main

python3_cron_scripts/get_umbrella_whois.py:124–168

Begin Main...

(logger=None)

def main(logger=None):
    """
    Begin Main...
    """
    if logger is None:
        logger = LoggingUtil.create_log(__name__)

    now = datetime.now()
    print("Starting: " + str(now))
    logger.info("Starting...")

    mongo_connector = MongoConnector.MongoConnector()
    umbrella = Umbrella.Umbrella()
    zi = ZoneIngestor.ZoneIngestor()

    # Obtain the list of known email addresses and name servers from the config collection
    config_collection = mongo_connector.get_config_connection()
    res = config_collection.find({})

    jobs_manager = JobsManager.JobsManager(mongo_connector, "get_umbrella_whois")
    jobs_manager.record_job_start()

    # Perform a search for each email address
    for i in range(0, len(res[0]["DNS_Admins"])):
        search_umbrella_by_email(
            logger, res[0]["DNS_Admins"][i], umbrella, zi, jobs_manager
        )

    # Perform a search based on each name server
    for i in range(0, len(res[0]["Whois_Name_Servers"])):
        search_umbrella_by_nameserver(
            logger,
            res[0]["Whois_Name_Servers"][i],
            res[0]["Whois_Orgs"],
            umbrella,
            zi,
            jobs_manager,
        )

    # Record status
    jobs_manager.record_job_complete()

    now = datetime.now()
    print("Complete: " + str(now))
    logger.info("Complete.")


if __name__ == "__main__":