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

cleanup

Browse Source
This commit is contained in:
Bouke Versteegh 2020-05-21 22:55:26 +02:00 committed by GitHub
parent 499489f1d3
commit 7e9ba0866c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
betterproto/plugin.py
View File

@ -75,10 +75,9 @@ def get_ref_type(package: str, imports: set, type_name: str, unwrap: bool = True
if type_name == "google.protobuf.Timestamp":
return "datetime"
else:
if wrapper:
imports.add(f"from {wrapper.__module__} import {wrapper.__name__}")
return f"{wrapper.__name__}"
elif wrapper:
imports.add(f"from {wrapper.__module__} import {wrapper.__name__}")
return f"{wrapper.__name__}"
if type_name.startswith(package):
parts = type_name.lstrip(package).lstrip(".").split(".")