From 7e9ba0866c8d636296f463345ec42609a3bae4ea Mon Sep 17 00:00:00 2001 From: Bouke Versteegh Date: Thu, 21 May 2020 22:55:26 +0200 Subject: [PATCH] cleanup --- betterproto/plugin.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/betterproto/plugin.py b/betterproto/plugin.py index d70786b..83b87d6 100755 --- a/betterproto/plugin.py +++ b/betterproto/plugin.py @@ -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(".")