Skip to content

Commit f4c4724

Browse files
committed
Add type annotations to timedelta
1 parent 9181f37 commit f4c4724

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

adafruit_datetime.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -537,7 +537,7 @@ def __mul__(self, other: float) -> "timedelta":
537537
__rmul__ = __mul__
538538

539539
# Supported comparisons
540-
def __eq__(self, other: "timedelta") -> bool:
540+
def __eq__(self, other: Any) -> bool:
541541
if not isinstance(other, timedelta):
542542
return False
543543
return self._cmp(other) == 0
@@ -799,7 +799,7 @@ class timezone(tzinfo):
799799
# Sentinel value to disallow None
800800
_Omitted = object()
801801

802-
def __new__(cls, offset, name=_Omitted):
802+
def __new__(cls, offset: timedelta, name: Union[str, object] = _Omitted) -> "timezone":
803803
if not isinstance(offset, timedelta):
804804
raise TypeError("offset must be a timedelta")
805805
if name is cls._Omitted:
@@ -824,36 +824,36 @@ def __new__(cls, offset, name=_Omitted):
824824

825825
# pylint: disable=protected-access, bad-super-call
826826
@classmethod
827-
def _create(cls, offset, name=None):
827+
def _create(cls, offset: timedelta, name: Optional[str] = None) -> "timezone":
828828
"""High-level creation for a timezone object."""
829829
self = super(tzinfo, cls).__new__(cls)
830830
self._offset = offset
831831
self._name = name
832832
return self
833833

834834
# Instance methods
835-
def utcoffset(self, dt):
835+
def utcoffset(self, dt: Optional["datetime"]) -> timedelta:
836836
if isinstance(dt, datetime) or dt is None:
837837
return self._offset
838838
raise TypeError("utcoffset() argument must be a datetime instance" " or None")
839839

840-
def tzname(self, dt):
840+
def tzname(self, dt: Optional["datetime"]) -> str:
841841
if isinstance(dt, datetime) or dt is None:
842842
if self._name is None:
843843
return self._name_from_offset(self._offset)
844844
return self._name
845845
raise TypeError("tzname() argument must be a datetime instance" " or None")
846846

847847
# Comparison to other timezone objects
848-
def __eq__(self, other):
848+
def __eq__(self, other: Any) -> bool:
849849
if not isinstance(other, timezone):
850850
return False
851851
return self._offset == other._offset
852852

853-
def __hash__(self):
853+
def __hash__(self) -> int:
854854
return hash(self._offset)
855855

856-
def __repr__(self):
856+
def __repr__(self) -> str:
857857
"""Convert to formal string, for repr()."""
858858
if self is self.utc:
859859
return "datetime.timezone.utc"
@@ -865,11 +865,11 @@ def __repr__(self):
865865
self._name,
866866
)
867867

868-
def __str__(self):
868+
def __str__(self) -> str:
869869
return self.tzname(None)
870870

871871
@staticmethod
872-
def _name_from_offset(delta):
872+
def _name_from_offset(delta: timedelta) -> str:
873873
if delta < timedelta(0):
874874
sign = "-"
875875
delta = -delta

0 commit comments

Comments
 (0)