Skip to content

Commit 671e114

Browse files
committed
fix: Recognize more print-safe type
1 parent 7c20ecd commit 671e114

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pybind11_stubgen/parser/mixins/parse.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from __future__ import annotations
22

33
import ast
4+
import datetime
45
import inspect
56
import re
67
import types
@@ -34,6 +35,7 @@
3435
TypeVar_,
3536
Value,
3637
)
38+
from pybind11_stubgen.typing_ext import DynamicSize, FixedSize
3739

3840
_generic_args = [
3941
Argument(name=Identifier("args"), variadic=True),
@@ -370,6 +372,10 @@ def handle_value(self, value: Any) -> Value:
370372
return Value(repr=str(self.handle_type(value)), is_print_safe=True)
371373
if inspect.ismodule(value):
372374
return Value(repr=value.__name__, is_print_safe=True)
375+
if isinstance(value, datetime.timedelta):
376+
return Value(repr=repr(value), is_print_safe=True)
377+
if isinstance(value, (FixedSize, DynamicSize)):
378+
return Value(repr=repr(value), is_print_safe=True)
373379
return Value(repr=repr(value), is_print_safe=False)
374380

375381
def handle_type(self, type_: type) -> QualifiedName:

0 commit comments

Comments
 (0)