vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

refactor: get_quarter_start returns current when dt is None

Browse source
This commit is contained in:
Hussain Nagaria 2023-12-17 14:54:59 +05:30
parent d0d78f65e9
commit 3899c14b67
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
frappe/utils/data.py
View file

@ -459,16 +459,24 @@ def get_first_day(
@typing.overload
def get_quarter_start(dt, as_str: Literal[False] = False) -> datetime.date:
def get_quarter_start(
dt: DateTimeLikeObject | None = None, as_str: Literal[False] = False
) -> datetime.date:
...
@typing.overload
def get_quarter_start(dt, as_str: Literal[True] = False) -> str:
def get_quarter_start(dt: DateTimeLikeObject | None = None, as_str: Literal[True] = False) -> str:
...
def get_quarter_start(dt, as_str: bool = False) -> str | datetime.date:
def get_quarter_start(
dt: DateTimeLikeObject | None = None, as_str: bool = False
) -> str | datetime.date:
"""Returns the start date of the quarter for the given datetime like object (`dt`).
If `dt` is None, the current quarter start date is returned.
"""
date = getdate(dt)
quarter = (date.month - 1) // 3 + 1
first_date_of_quarter = datetime.date(date.year, ((quarter - 1) * 3) + 1, 1)