Skip to content

Commit 9a85e4d

Browse files
committed
Add back datetime imports
1 parent 7058811 commit 9a85e4d

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

src/betterproto/plugin/models.py

+8-9
Original file line numberDiff line numberDiff line change
@@ -440,15 +440,14 @@ def betterproto_field_args(self) -> List[str]:
440440

441441
@property
442442
def datetime_imports(self) -> Set[str]:
443-
# imports = set()
444-
# annotation = self.annotation
445-
# # FIXME: false positives - e.g. `MyDatetimedelta`
446-
# if "timedelta" in annotation:
447-
# imports.add("timedelta")
448-
# if "datetime" in annotation:
449-
# imports.add("datetime")
450-
# return imports
451-
return {"timedelta", "datetime"}
443+
imports = set()
444+
annotation = self.annotation
445+
# FIXME: false positives - e.g. `MyDatetimedelta`
446+
if "timedelta" in annotation:
447+
imports.add("timedelta")
448+
if "datetime" in annotation:
449+
imports.add("datetime")
450+
return imports
452451

453452
@property
454453
def pydantic_imports(self) -> Set[str]:

0 commit comments

Comments
 (0)