Skip to content

store_message

main.management.commands.store_message

Django management command to populate Kafka messages into application database.

Classes

Command

Bases: BaseCommand

Store Kafka messages in the database.

Functions
add_arguments(parser)

Add commandline options.

Source code in main/management/commands/store_message.py 
17
18
19
20
21
def add_arguments(self, parser: ArgumentParser) -> None:
    """Add commandline options."""
    parser.add_argument("-t", "--topic", default="NO_TOPIC")
    parser.add_argument("-m", "--message", default="NO_MESSAGE")
    parser.add_argument("-s", "--severity", default="INFO")
handle(*args, **kwargs)

Command business logic.

Source code in main/management/commands/store_message.py 
23
24
25
26
27
28
29
30
31
def handle(self, *args: Any, **kwargs: Any) -> None:  # type: ignore[explicit-any]
    """Command business logic."""
    topic = kwargs["topic"]
    message = kwargs["message"]
    timestamp = datetime.now(tz=timezone.utc)
    severity = kwargs["severity"]
    DruncMessage.objects.create(
        topic=topic, timestamp=timestamp, message=message, severity=severity
    )