Function main

python3_cron_scripts/create_cert_graphs.py:691–793

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()
    mongo_ct = mongo_connector.get_certificate_transparency_connection()
    cert_graphs_collection = mongo_connector.get_cert_graphs_connection()
    jobs_manager = JobsManager.JobsManager(mongo_connector, "create_cert_graphs")
    jobs_manager.record_job_start()

    zones = ZoneManager.get_distinct_zones(mongo_connector)

    parser = argparse.ArgumentParser(
        description="Creates and stores certificate graphs in the database based on one or more sources."
    )
    parser.add_argument(
        "--check_censys",
        action="store_true",
        default=False,
        required=False,
        help="Whether to check the Censys collection in the database",
    )
    parser.add_argument(
        "--check_443_scans",
        action="store_true",
        default=False,
        required=False,
        help="Whether to check the zgrab collection in the database",
    )
    parser.add_argument(
        "--check_ct_scans",
        action="store_true",
        default=False,
        required=False,
        help="Whether to check the CT collection in the database",
    )
    parser.add_argument(
        "--zgrab_version",
        default=2,
        type=int,
        choices=[1, 2],
        metavar="version",
        help="The version of ZGrab used to collect data",
    )
    args = parser.parse_args()

    if args.check_censys is True:
        censys_collection = mongo_connector.get_censys_connection()

    if args.check_443_scans is True:
        zgrab_collection = mongo_connector.get_zgrab_443_data_connection()