openapi-python-client/openapi_python_client/utils.py at editable · ramnes/openapi-python-client · GitHub

Repository navigation

History

import builtins

import re

from keyword import iskeyword

import stringcase

def sanitize(value: str) -> str:

""" Removes every character that isn't 0-9, A-Z, a-z, ' ', -, or _ """

return re.sub(r"[^\w _\-]+", "", value)

def fix_keywords(value: str) -> str:

if iskeyword(value):

return f"{value}_"

return value

RESERVED_WORDS = set(dir(builtins)).union({"self"})

def fix_reserved_words(value: str) -> str:

if value in RESERVED_WORDS:

return f"{value}_"

return value

def group_title(value: str) -> str:

value = re.sub(r"([A-Z]{2,})([A-Z][a-z]|[ \-_]|$)", lambda m: m.group(1).title() + m.group(2), value.strip())

value = re.sub(r"(^|[ _-])([A-Z])", lambda m: m.group(1) + m.group(2).lower(), value)

return value

def snake_case(value: str) -> str:

return fix_keywords(stringcase.snakecase(group_title(sanitize(value))))

def pascal_case(value: str) -> str:

return fix_keywords(stringcase.pascalcase(sanitize(value.replace(" ", ""))))

def kebab_case(value: str) -> str:

return fix_keywords(stringcase.spinalcase(group_title(sanitize(value))))

def remove_string_escapes(value: str) -> str:

return value.replace('"', r"\"")

# This can be changed by config.Config.load_config

FIELD_PREFIX = "field_"

def to_valid_python_identifier(value: str) -> str:

"""

Given a string, attempt to coerce it into a valid Python identifier by stripping out invalid characters and, if

necessary, prepending a prefix.

See:

https://docs.python.org/3/reference/lexical_analysis.html#identifiers

"""

new_value = fix_reserved_words(fix_keywords(sanitize(value)))

if new_value.isidentifier():

return new_value

return f"{FIELD_PREFIX}{new_value}"

openapi-python-client/openapi_python_client/utils.py at editable · ramnes/openapi-python-client · GitHub

Repository navigation

Expand file tree

Breadcrumbs

utils.py

Latest commit

History

Breadcrumbs

utils.py

File metadata and controls