From 518680e13de53ac4f0c5571451b461fcba039837 Mon Sep 17 00:00:00 2001
From: Adrien Vannson <adrien.vannson@gardacp.com>
Date: Wed, 14 Aug 2024 15:28:42 +0200
Subject: [PATCH] Add import if needed

---
 src/betterproto/plugin/models.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/betterproto/plugin/models.py b/src/betterproto/plugin/models.py
index cf14cadb..03834a7e 100644
--- a/src/betterproto/plugin/models.py
+++ b/src/betterproto/plugin/models.py
@@ -275,8 +275,21 @@ def input_filenames(self) -> Iterable[str]:
     @property
     def python_module_imports(self) -> Set[str]:
         imports = set()
+
+        has_deprecated = False
+        if any(m.deprecated for m in self.messages):
+            has_deprecated = True
         if any(x for x in self.messages if any(x.deprecated_fields)):
+            has_deprecated = True
+        if any(
+            any(m.proto_obj.options.deprecated for m in s.methods)
+            for s in self.services
+        ):
+            has_deprecated = True
+
+        if has_deprecated:
             imports.add("warnings")
+
         if self.builtins_import:
             imports.add("builtins")
         return imports