From a2e06b453ad44136cb473714f85852a3fa5d69eb Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Tue, 4 Jan 2022 22:42:17 +0700
Subject: [PATCH] Update stubs

---
 odoo-stubs/tools/date_utils.pyi | 28 +++++++++++++++++-----------
 1 file changed, 17 insertions(+), 11 deletions(-)

diff --git a/odoo-stubs/tools/date_utils.pyi b/odoo-stubs/tools/date_utils.pyi
index b064cd9..ae69b60 100644
--- a/odoo-stubs/tools/date_utils.pyi
+++ b/odoo-stubs/tools/date_utils.pyi
@@ -1,12 +1,18 @@
-from typing import Any
+import datetime
+from dateutil.relativedelta import relativedelta
 
-def get_month(date: Any): ...
-def get_quarter_number(date: Any): ...
-def get_quarter(date: Any): ...
-def get_fiscal_year(date: Any, day: int = ..., month: int = ...): ...
-def start_of(value: Any, granularity: Any): ...
-def end_of(value: Any, granularity: Any): ...
-def add(value: Any, *args: Any, **kwargs: Any): ...
-def subtract(value: Any, *args: Any, **kwargs: Any): ...
-def json_default(obj: Any): ...
-def date_range(start: Any, end: Any, step: Any = ...): ...
+from typing import Any, Tuple, Iterator, TypeVar
+
+_T = TypeVar('_T', datetime.date, datetime.datetime)
+
+
+def get_month(date: _T) -> Tuple[_T, _T]: ...
+def get_quarter_number(date: Any) -> int: ...
+def get_quarter(date: _T) -> Tuple[_T, _T]: ...
+def get_fiscal_year(date: _T, day: int = ..., month: int = ...) -> Tuple[_T, _T]: ...
+def start_of(value: _T, granularity: Any) -> _T: ...
+def end_of(value: _T, granularity: Any) -> _T: ...
+def add(value: _T, *args: Any, **kwargs: Any) -> _T: ...
+def subtract(value: _T, *args: Any, **kwargs: Any) -> _T: ...
+def json_default(obj: Any) -> str: ...
+def date_range(start: datetime.datetime, end: datetime.datetime, step: relativedelta = ...) -> Iterator[datetime.datetime]: ...