dates.py update

This commit is contained in:
tiagorg 2022-10-23 16:24:24 +01:00
parent 2b01fbde51
commit cea2d432ec
Signed by untrusted user who does not match committer: TiagoRG
GPG Key ID: DFCD48E3F420DB42
1 changed files with 6 additions and 33 deletions

View File

@ -1,50 +1,23 @@
def isLeapYear(year): def isLeapYear(year):
# Garante que apenas os anos de mudança de século múltiplos de 400 return year % 400 == 0 if year % 100 == 0 else year % 4 == 0
# devolvem True
if year % 100 == 0:
return year % 400 == 0
# Devolve True para o resto dos anos múltiplos de 4
return year%4 == 0
def monthDays(year, month): def monthDays(year, month):
assert month > 0 assert month > 0
MONTHDAYS = (0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31) MONTHDAYS = (0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31)
days = MONTHDAYS[month] days = MONTHDAYS[month]
# Só adicionado mais 1 dia se o mẽs for fevereiro (2) e o
# ano for bissexto
if isLeapYear(year) and month == 2:
days += 1
return days
return days + 1 if (isLeapYear(year) and month == 2) else days
def nextDay(year, month, day): def nextDay(year, month, day):
assert 13 > month > 0 # Verifica se é o último dia do ano
assert day > 0
MONTHS31DAYS = (1, 3, 5, 7, 8, 10)
MONTHS30DAYS = (4, 6, 9, 11)
# Último dia do ano
if (month, day) == (12, 31): if (month, day) == (12, 31):
year += 1 year += 1
month = 1 month = 1
day = 1 day = 1
# Último dia de um mês segundo os casos: # Verifica se é o último dia do mês
# > Mês de 31 dias e é o dia 31 do mês elif (monthDays(year, month) == day):
# > Mês de 30 dias e é o dia 30 do mês
# > Fevereiro e (
# o ano é bissexto e é o dia 29 do mês
# ou o ano não é bissexto e é o dia 28 do mês
# )
elif ( (month in MONTHS31DAYS and day == 31)
or (month in MONTHS30DAYS and day == 30)
or (month == 2 and (
(isLeapYear(year) and day == 29)
or (not isLeapYear(year) and day == 28)
))):
month += 1 month += 1
day = 1 day = 1