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

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

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