pincer._config

Complexity

Total Complexity

3

Size/Duplication

Total Lines	43
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	GatewayConfig.compressed()	0	7	1
A	GatewayConfig.uri()	0	16	2

# Copyright Pincer 2021-Present

Missing module docstring

# Full MIT License can be found in `LICENSE` at the project root.

from dataclasses import dataclass
from typing import Optional


@dataclass
class GatewayConfig:
    """
    This file is to make maintaining the library its gateway
        configuration easier.
    """
    socket_base_url: str = "wss://gateway.discord.gg/"
    version: int = 9
    encoding: str = "json"
    compression: Optional[str] = "zlib-stream"

    @staticmethod
    def uri() -> str:
        """
        :return uri:
            The GatewayConfig's uri.
        """
        uri = (
            f"{GatewayConfig.socket_base_url}"
            f"?v={GatewayConfig.version}"
            f"&encoding={GatewayConfig.encoding}"
        )

        if GatewayConfig.compressed():
            uri += f"&compress={GatewayConfig.compression}"

        return uri

    @staticmethod
    def compressed() -> bool:
        """
        :return compressed:
            Whether the Gateway should compress payloads or not.
        """
        return GatewayConfig.compression == "zlib-stream"