Промежуточное программное обеспечение в FastAPI для генерации UUID для каждого запроса и отправки в журналы - PullRequest
Новая
       25

Промежуточное программное обеспечение в FastAPI для генерации UUID для каждого запроса и отправки в журналы

0 голосов
/ 24 марта 2020

Я ищу промежуточное ПО в Fast API для генерации UUID для каждого запроса и отправки его в logs . 

# To see the logs, run this in python interpreter
# import with_logger
# with_logger.main()

import uuid
from logging import getLogger
import logging.config
from yaml import load, FullLoader
from fastapi import FastAPI
import uvicorn

app = FastAPI()


# @app.middleware("http")
@app.get("/")
def gen_uuid():
    # For every get request generate uuid
    global uuid_var
    uuid_var = uuid.uuid4()
    return uuid_var


log_dict = {
    'version': 1,
    'disable_existing_loggers': False,
    'formatters': {
        'standard': {
            'format': '%(uuid_var)s %(asctime)s %(levelname)-8s %(name)-15s %(message)s -Thread id: %(thread)d  Thread name: %(threadName)s'
        },
    },
    'handlers': {
        'default': {
            'level': 'INFO',
            'formatter': 'standard',
            'class': 'logging.StreamHandler',
        },
        'file_handler': {
            'level': 'INFO',
            'filename': 'mylogfile.log',
            'class': 'logging.FileHandler',
            'formatter': 'standard'
        }
    },
    'loggers': {
        '': {
            'handlers': ['file_handler'],
            'level': 'INFO',
            'propagate': True
        },
    }
}


class ContextFilter(logging.Filter):
    def filter(self, record):
        record.uuid_var = gen_uuid()
        return True


def main():
    logging.config.dictConfig(log_dict)

    handler = logging.StreamHandler()
    handler.formatter = logging.Formatter(
        "%(uuid_var)s %(asctime)s %(levelname)-8s %(name)-15s %(message)s -Thread id: %(thread)d  Thread name: %(threadName)s")
    handler.addFilter(ContextFilter())
    logger = getLogger(__name__)
    logger.addHandler(handler)

    logger.warning(" This is a warning message")

    uvicorn.run(app, host="127.0.0.1", port=8080)

Мне нужно, чтобы uuid был отправлен в заголовок X-Request-ID, и я буду использовать библиотеку журналов python, чтобы извлечь его оттуда, как, 

@app.middleware("http")
async def get_x_request_id_header(*, x_request_id: str = Header(None)):
    # return x-request-id header
    return {"X-Request-ID": x_request_id}

I нужно все это в промежуточном программном обеспечении. Есть ли промежуточное программное обеспечение, которое уже существует в Fast API, как это?

Мне нужно сгенерировать uuid внутри промежуточного программного обеспечения, например, 

@app.middleware("http")
def gen_uuid():
    # For every get request generate uuid
    global uuid_var
    uuid_var = uuid.uuid4()
    return uuid_var

Но я не могу этого сделать. Я получаю сообщение об ошибке: gen_uuid принимает 2 параметра, но передает 0 параметров

...