Example #1
0
def setup_parent_logging(args, *, stream=sys.stderr):
    level = VERBOSITY.get(args.verbose, logging.DEBUG)
    handler = logging.StreamHandler()
    formatter = MissingFieldFormatter(LOGFORMAT)
    handler.setFormatter(formatter)
    if not args.skip_logging:
        logging.basicConfig(level=level, format=LOGFORMAT, handlers=[handler])

    return get_logger("dramatiq", "MainProcess")
Example #2
0
    def setup_logging(args, child_id, logging_pipe):
        # Redirect all output to the logging pipe so that all output goes
        # to stderr and output is serialized so there isn't any mangling.
        sys.stdout = logging_pipe
        sys.stderr = logging_pipe

        level = VERBOSITY.get(args.verbose, logging.DEBUG)
        logging.basicConfig(level=level, format=LOGFORMAT, stream=logging_pipe)
        logging.getLogger("pika").setLevel(logging.CRITICAL)
        return get_logger("dramatiq", "%s(%s)" % (prefix, child_id))
Example #3
0
def setup_worker_logging(args, worker_id, logging_pipe):
    # Redirect all output to the logging pipe so that all output goes
    # to stderr and output is serialized so there isn't any mangling.
    sys.stdout = logging_pipe
    sys.stderr = logging_pipe

    level = verbosity.get(args.verbose, logging.DEBUG)
    logging.basicConfig(level=level, format=logformat, stream=logging_pipe)
    logging.getLogger("pika").setLevel(logging.CRITICAL)
    return get_logger("dramatiq", "WorkerProcess(%s)" % worker_id)
Example #4
0
    def setup_logging(args, child_id, logging_pipe):
        # Redirect all output to the logging pipe so that all output goes
        # to stderr and output is serialized so there isn't any mangling.
        sys.stdout = logging_pipe
        sys.stderr = logging_pipe
        handler = logging.StreamHandler()
        formatter = MissingFieldFormatter(LOGFORMAT)
        handler.setFormatter(formatter)

        level = VERBOSITY.get(args.verbose, logging.DEBUG)
        if not args.skip_logging:
            logging.basicConfig(level=level,
                                format=LOGFORMAT,
                                handlers=[handler])

        logging.getLogger("pika").setLevel(logging.CRITICAL)
        return get_logger("dramatiq", "%s(%s)" % (prefix, child_id))
Example #5
0
def setup_parent_logging(args, *, stream=sys.stderr):
    level = VERBOSITY.get(args.verbose, logging.DEBUG)
    if not args.skip_logging:
        logging.basicConfig(level=level, format=LOGFORMAT, stream=stream)

    return get_logger("dramatiq", "MainProcess")
Example #6
0
def setup_parent_logging(args, *, stream=sys.stderr):
    level = verbosity.get(args.verbose, logging.DEBUG)
    logging.basicConfig(level=level, format=logformat, stream=stream)
    return get_logger("dramatiq", "MainProcess")
Example #7
0
def setup_worker_logging(args, worker_id, logging_pipe):
    level = verbosity.get(args.verbose, logging.DEBUG)
    logging.basicConfig(level=level, format=logformat, stream=logging_pipe)
    logging.getLogger("pika").setLevel(logging.ERROR)
    return get_logger("dramatiq", "WorkerProcess(%s)" % worker_id)
Example #8
0
 def __init__(self, max_tasks=100):
     self.counter_mu = Lock()
     self.counter = max_tasks
     self.signaled = False
     self.logger = get_logger("max_tasks_per_child.app", MaxTasksPerChild)
Example #9
0
from dramatiq import get_logger
from pymongo import MongoClient
from pymongo.database import Database
from pymongo.errors import ServerSelectionTimeoutError

from drama.config import settings

logger = get_logger(__name__)


class DataBase:
    client: MongoClient = None


db = DataBase()


def create_db_connection() -> Database:
    logger.debug("Connecting to database for the first time")
    client = MongoClient(settings.MONGO_DNS)

    try:
        client.server_info()
    except ServerSelectionTimeoutError:
        raise ConnectionError(f"Could not connect to database with connection string {settings.MONGO_DNS}")

    db.client = client

    return db.client.drama