snakelet.utilities.conversion

Complexity

Total Complexity

5

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	camel()	0	6	1
A	snake()	0	6	1

import re


def snake(string: str):
    """
    :param string:
    :return:
    """
    return re.sub("([A-Z])", "_\\1", string).lower().lstrip("_")


def camel(string: str):
    """
    :param string:
    :return:
    """
    return "".join(map(str.capitalize, string.split("_")))


class Conversion:
    cases = {
        'snake': snake,
        'camel': camel
    }

    def __init__(self, case: str = None):
        """
        :param case:
        """
        self.case = self.cases['snake' if not case or case not in self.cases else case]

    def encode(self, data: str):
        """
        :param data:
        :return:
        """
        return self.case(data)