Remove dependency on stringcase, apply black

This commit is contained in:
boukeversteegh 2020-06-11 00:22:23 +02:00
parent 5d2f3a2cd9
commit 3ca75dadd7
2 changed files with 7 additions and 6 deletions

View File

@ -28,9 +28,8 @@ from typing import (
import grpclib.const
import stringcase
from .casing import safe_snake_case, snake_case
from .casing import camel_case, safe_snake_case, snake_case
if TYPE_CHECKING:
from grpclib._protocols import IProtoMessage
@ -131,7 +130,7 @@ DATETIME_ZERO = datetime_default_gen()
class Casing(enum.Enum):
"""Casing constants for serialization."""
CAMEL = stringcase.camelcase
CAMEL = camel_case
SNAKE = snake_case

View File

@ -154,8 +154,10 @@ def import_cousin(current_package, imports, py_package, py_type):
)
string_import = py_package[-1]
# Add trailing __ to avoid name mangling (python.org/dev/peps/pep-0008/#id34)
string_alias = f"{'_' * distance_up}" + safe_snake_case(
".".join(py_package[len(shared_ancestry) :])
) + "__"
string_alias = (
f"{'_' * distance_up}"
+ safe_snake_case(".".join(py_package[len(shared_ancestry) :]))
+ "__"
)
imports.add(f"from {string_from} import {string_import} as {string_alias}")
return f"{string_alias}.{py_type}"