forked from openapi-generators/openapi-python-client
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathutils.py
More file actions
71 lines (47 loc) · 1.94 KB
/
utils.py
File metadata and controls
71 lines (47 loc) · 1.94 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import builtins
import re
from keyword import iskeyword
from typing import Any, List
delimiters = " _-"
class PythonIdentifier(str):
"""A string which has been validated / transformed into a valid identifier for Python"""
def __new__(cls, value: str, prefix: str) -> "PythonIdentifier":
new_value = fix_reserved_words(fix_keywords(snake_case(sanitize(value))))
if not new_value.isidentifier():
new_value = f"{prefix}{new_value}"
return str.__new__(cls, new_value)
def __deepcopy__(self, _: Any) -> "PythonIdentifier":
return self
def sanitize(value: str) -> str:
"""Removes every character that isn't 0-9, A-Z, a-z, or a known delimiter"""
return re.sub(rf"[^\w{delimiters}]+", "", value)
def split_words(value: str) -> List[str]:
"""Split a string on words and known delimiters"""
# We can't guess words if there is no capital letter
if any(c.isupper() for c in value):
value = " ".join(re.split("([A-Z]?[a-z]+)", value))
return re.findall(rf"[^{delimiters}]+", value)
def fix_keywords(value: str) -> str:
if iskeyword(value):
return f"{value}_"
return value
RESERVED_WORDS = (set(dir(builtins)) | {"self"}) - {"type", "id"}
def fix_reserved_words(value: str) -> str:
if value in RESERVED_WORDS:
return f"{value}_"
return value
def snake_case(value: str) -> str:
words = split_words(sanitize(value))
value = "_".join(words).lower()
return fix_keywords(value)
def pascal_case(value: str) -> str:
words = split_words(sanitize(value))
capitalized_words = (word.capitalize() if not word.isupper() else word for word in words)
value = "".join(capitalized_words)
return fix_keywords(value)
def kebab_case(value: str) -> str:
words = split_words(sanitize(value))
value = "-".join(words).lower()
return fix_keywords(value)
def remove_string_escapes(value: str) -> str:
return value.replace('"', r"\"")