You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.1 KiB
44 lines
1.1 KiB
"""Alias generators for converting between different capitalization conventions.""" |
|
import re |
|
|
|
__all__ = ('to_pascal', 'to_camel', 'to_snake') |
|
|
|
|
|
def to_pascal(snake: str) -> str: |
|
"""Convert a snake_case string to PascalCase. |
|
|
|
Args: |
|
snake: The string to convert. |
|
|
|
Returns: |
|
The PascalCase string. |
|
""" |
|
camel = snake.title() |
|
return re.sub('([0-9A-Za-z])_(?=[0-9A-Z])', lambda m: m.group(1), camel) |
|
|
|
|
|
def to_camel(snake: str) -> str: |
|
"""Convert a snake_case string to camelCase. |
|
|
|
Args: |
|
snake: The string to convert. |
|
|
|
Returns: |
|
The converted camelCase string. |
|
""" |
|
camel = to_pascal(snake) |
|
return re.sub('(^_*[A-Z])', lambda m: m.group(1).lower(), camel) |
|
|
|
|
|
def to_snake(camel: str) -> str: |
|
"""Convert a PascalCase or camelCase string to snake_case. |
|
|
|
Args: |
|
camel: The string to convert. |
|
|
|
Returns: |
|
The converted string in snake_case. |
|
""" |
|
snake = re.sub(r'([a-zA-Z])([0-9])', lambda m: f'{m.group(1)}_{m.group(2)}', camel) |
|
snake = re.sub(r'([a-z0-9])([A-Z])', lambda m: f'{m.group(1)}_{m.group(2)}', snake) |
|
return snake.lower()
|
|
|