pub/python-betterproto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix all broken imports

Browse Source
This commit is contained in:
boukeversteegh
2020-06-05 12:33:51 +02:00
parent 57523a9e7f
commit d7ba27de2b
2 changed files with 66 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
betterproto/compile/importing.py
View File

@@ -44,7 +44,6 @@ def get_ref_type(
if type_name == "google.protobuf.Timestamp":
return "datetime"
# Use precompiled classes for google.protobuf.* objects
if type_name.startswith("google.protobuf.") and type_name.count(".") == 2:
type_name = type_name.rsplit(".", maxsplit=1)[1]
@@ -53,12 +52,12 @@ def get_ref_type(
imports.add(f"import {import_package} as {import_alias}")
return f"{import_alias}.{type_name}"
importing_package: List[str] = type_name.split('.')
importing_package: List[str] = type_name.split(".")
importing_type: str = stringcase.pascalcase(importing_package.pop())
current_package: List[str] = package.split('.') if package else []
current_package: List[str] = package.split(".") if package else []
# importing sibling
'''
"""
package =
name = Foo
@@ -67,26 +66,26 @@ def get_ref_type(
package = foo.bar
name = foo.bar.Baz
'''
"""
if importing_package == current_package:
imports.add(f"from . import {importing_type}")
return importing_type
# importing child & descendent:
'''
"""
package =
name = foo.Bar
package =
name = foo.bar.Baz
'''
if importing_package[0:len(current_package)] == current_package:
importing_descendent = importing_package[len(current_package):]
string_from = '.'.join(importing_descendent[0:-1])
"""
if importing_package[0 : len(current_package)] == current_package:
importing_descendent = importing_package[len(current_package) :]
string_from = ".".join(importing_descendent[0:-1])
string_import = importing_descendent[-1]
if string_from:
string_alias = '_'.join(importing_descendent)
string_alias = "_".join(importing_descendent)
imports.add(f"from .{string_from} import {string_import} as {string_alias}")
return f"{string_alias}.{importing_type}"
else:
@@ -94,7 +93,7 @@ def get_ref_type(
return f"{string_import}.{importing_type}"
# importing parent & ancestor
'''
"""
package = foo.bar
name = foo.Foo
@@ -103,27 +102,34 @@ def get_ref_type(
package = foo.bar.baz
name = Bar
'''
if current_package[0:len(importing_package)] == importing_package:
distance = len(current_package) - len(importing_package)
imports.add(f"from .{'.' * distance} import {importing_type}")
"""
if current_package[0 : len(importing_package)] == importing_package:
distance_up = len(current_package) - len(importing_package)
imports.add(f"from .{'.' * distance_up} import {importing_type}")
return importing_type
# importing unrelated or cousin
'''
"""
package = foo.bar
name = baz.Foo
package = foo.bar.baz
name = foo.example.Bar
'''
root_distance = len(current_package)
shared_ancestory_length = reduce(lambda l, pair: l + (pair[0] == pair[1]), zip(current_package, importing_package), 0)
"""
string_from = f"{'.' * (shared_ancestory_length+1)}.{'.'.join(importing_package[0:-1])}"
shared_ancestory = [
pair[0]
for pair in zip(current_package, importing_package)
if pair[0] == pair[1]
]
distance_up = len(current_package) - len(shared_ancestory)
string_from = f".{'.' * distance_up}" + ".".join(
importing_package[len(shared_ancestory) : -1]
)
string_import = importing_package[-1]
string_alias = '_' * root_distance + safe_snake_case('.'.join(importing_package))
string_alias = f"{'_' * distance_up}" + safe_snake_case(
".".join(importing_package[len(shared_ancestory) :])
)
imports.add(f"from {string_from} import {string_import} as {string_alias}")
return f"{string_alias}.{importing_type}"
# return type_name