def __init__(self): self.logger = get_logger(__name__) self.conn = DatabaseHandler.get_connection() self.module_store = ModulesStore() self.package_store = PackageStore() self.update_store = UpdateStore() self.content_set_to_db_id = self._prepare_content_set_map()
def __init__(self): self.logger = get_logger(__name__) self.downloader = FileDownloader() self.cvemap_store = CvemapStore() self.updated = False self.lastmodified = None self.tmp_directory = tempfile.mkdtemp(prefix="cvemap-")
def __init__(self): self.logger = get_logger(__name__) self.downloader = FileDownloader() self.unpacker = FileUnpacker() self.cverepo_store = CveRepoStore() self.repos = set() self.db_lastmodified = {} self.year_since = int(os.getenv('YEAR_SINCE', DEFAULT_YEAR_SINCE))
def __init__(self): self.logger = get_logger(__name__) self.downloader = FileDownloader() self.unpacker = FileUnpacker() self.repo_store = RepositoryStore() self.repositories = set() self.certs_tmp_directory = None self.certs_files = {}
def __init__(self): self.logger = get_logger(__name__) self.conn = DatabaseHandler.get_connection() self.cpe_label_to_id = {} self.cpe_label_to_name = {} cur = self.conn.cursor() cur.execute("select id, label, name from cpe") for cpe_id, label, name in cur.fetchall(): self.cpe_label_to_id[label] = cpe_id self.cpe_label_to_name[label] = name cur.close()
def __init__(self, oval_id, updated, url, local_path): self.logger = get_logger(__name__) self.oval_id = oval_id self.updated = updated self.url = url self.local_path = local_path self.root = None self.definitions = [] self.tests = [] self.objects = [] self.states = []
def test_progress_logger(caplog): """Test ProgressLogger.""" logger = logging_utils.get_logger(__name__) progress_logger = logging_utils.ProgressLogger(logger, 3, log_interval=0) progress_logger.update() progress_logger.update() progress_logger.update() progress_logger.reset(4) progress_logger.update() assert caplog.records[0].message == ' 33.33 % completed [1/3]' assert caplog.records[1].message == ' 66.67 % completed [2/3]' assert caplog.records[2].message == '100.00 % completed [3/3]' assert caplog.records[3].message == ' 25.00 % completed [1/4]'
import glob import os import json import gzip from common.logging_utils import get_logger, init_logging from common.dateutil import format_datetime, now from common.fileutil import remove_file_if_exists from common.rpm import join_rpm_name from database.database_handler import DatabaseHandler, NamedCursor, init_db DEFAULT_KEEP_COPIES = "2" PKGTREE_FILE = '/data/pkg_tree.json.gz' DEFAULT_PKGTREE_INDENT = "0" LOGGER = get_logger(__name__) class JsonPkgTree: # pylint: disable=too-many-instance-attributes """Class for creating package tree json file from database.""" def __init__(self, db_instance, filename): self.db_instance = db_instance self.filename = filename self.outputdata = {} self.datadict = {} self.pkgnameid2pkgname = {} self.evrid2evr = {} self.archid2arch = {} self.packagedata = {} self.repodata = {} self.cvename = {}
def __init__(self): self.logger = get_logger(__name__) self.repo = [] self.conn = DatabaseHandler.get_connection() self.cve_store = CveStore()
def __init__(self): self.queue = [] self.logger = get_logger(__name__) self.chunk_size = int(os.getenv('CHUNK_SIZE', DEFAULT_CHUNK_SIZE)) self.progress_logger = ProgressLogger(self.logger, 0)
def __init__(self): self.logger = get_logger(__name__) self.conn = DatabaseHandler.get_connection()
def __init__(self): self.queue = Queue() self.logger = get_logger(__name__) self.num_threads = int(os.getenv('THREADS', DEFAULT_THREADS))
def __init__(self): self.logger = get_logger(__name__) self.downloader = FileDownloader() self.cpe_store = CpeStore() self.tmp_directory = tempfile.mkdtemp(prefix="cpe-")
Wait for VMaaS services. """ import os import signal import sys import time from requests import request from common.config import Config from common.logging_utils import get_logger from common.logging_utils import init_logging from reposcan.database.database_handler import DatabaseHandler from reposcan.database.database_handler import init_db LOGGER = get_logger(__file__) def bye(): """Handle signal""" sys.exit("Stopped.") def wait(func, *args, delay=1, service=""): """Waits for success of `func`.""" LOGGER.info("Checking if %s is up", service) while True: try: result = func(*args) if result: return
#!/usr/bin/env python3 """ Main entrypoint of websocket server. """ import signal from tornado.ioloop import IOLoop, PeriodicCallback from tornado.web import Application, RequestHandler from tornado.websocket import WebSocketHandler from common.logging_utils import get_logger, init_logging WEBSOCKET_PING_INTERVAL = 5 WEBSOCKET_TIMEOUT = 60 LOGGER = get_logger("websocket") class NotificationHandler(WebSocketHandler): """Websocket handler to send messages to subscribed clients.""" connections = {} # What is the freshest data each webapp has webapp_export_timestamps = {} # What time did we advertise last last_refresh_sent = None def __init__(self, application, request, **kwargs): super(NotificationHandler, self).__init__(application, request, **kwargs) self.last_pong = None self.timeout_callback = None
def __init__(self): self.logger = get_logger(__name__) self.conn = DatabaseHandler.get_connection() self.cpe_label_to_id = {} self.cpe_label_to_name = {}