In questo articolo vedremo un esempio pratico di operazioni sulle date in Python.
Realizzeremo una semplice applicazione da riga di comando che riceverà in input la data di nascita dell'utente e restituirà il nome del suo segno zodiacale.
I segni zodiacali sono dodici e ciascuno di loro copre un determinato arco temporale nel corso dell'anno. Per rappresentarli useremo un file JSON contenente un array di oggetti così strutturati:
[
{
"name": "Nome Segno",
"date": {
"from": [giorno, mese],
"to": [giorno, mese]
}
}
]
giorno e mese sono numeri interi. Una caratteristica del costruttore della classe date di Python è appunto quella di accettare solo numeri interi per dati come l'anno, il mese e il giorno. from rappresenta la data di inizio del periodo di un segno e to il suo termine. L'anno è ininfluente.
Dobbiamo accettare solo date in formato ISO, ossia YYYY-mm-dd e verificare che anno, mese e giorno siano validi. A tale scopo definiamo come prima cosa una funzione di validazione.
import json
import re
import sys
from datetime import date, datetime
def is_valid_date_str(date_str):
reg = re.compile(r'^\d{4}-\d{2}-\d{2}$')
match = reg.search(date_str)
if not match:
return False
try:
dt = date.fromisoformat(date_str)
year = dt.year
now = datetime.now()
if year >= now.year:
return False
month = dt.month
if month > 12 or month < 1:
return False
day = dt.day
if day > 31 or day < 1:
return False
except ValueError:
return False
return True
Il primo step è verificare il formato della data usando una semplice espressione regolare. Usiamo search() per effettuare un confronto sull'intera stringa di input. A questo punto se il formato è valido, corroboriamo la nostra routine ottenendo un oggetto di tipo date a partire dal formato ISO fornito usando il metodo fromisoformat(). Inoltre procediamo a verificare che anno, mese e giorno ricadano in un intervallo valido.
Ora possiamo definire una funzione per ottenere la lista di dizionari dal file JSON.
def get_signs_list(filename=None):
signs = []
if filename is None:
return signs
with open(filename, 'r') as f:
signs = json.load(f)
return signs
Sappiamo che i dati rilevanti sono il giorno e il mese della data di nascita dell'utente e che tali dati vanno confrontati con i valori contenuti nelle liste from e to di ciascun dizionario. Dobbiamo quindi operare un confronto tra tre oggetti date partendo dalla trasformazione della stringa nel formato YYYY-mm-dd della data di nascita in un oggetto data di Python. Quest'operazione può essere effettuata tramite il metodo datetime.strptime() che accetta come primo argomento la stringa data e come secondo argomento il formato atteso della data di input.
def find_sign_by_birth_date(birthdate=None):
if birthdate is None:
return None
if not is_valid_date_str(birthdate):
return None
signs_list = get_signs_list('./zodiac-signs.json')
format_date = '%Y-%m-%d'
dt = datetime.strptime(birthdate, format_date)
now = datetime.now()
day = dt.day
month = dt.month
reference_date = date(now.year, month, day)
sign_name = ''
for sign in signs_list:
from_day, from_month = sign['date']['from']
to_day, to_month = sign['date']['to']
from_date = date(now.year, from_month, from_day)
to_date = date(now.year, to_month, to_day)
if reference_date >= from_date and reference_date <= to_date:
sign_name = sign['name']
break
return sign_name
Se la data di nascita ricade nel range stabilito, ossia se è maggiore o uguale della data iniziale del periodo e se al contempo è minore o uguale della data finale del periodo, restituiamo il nome del segno zodiacale dell'utente. Ricordiamo ancora che l'anno è ininfluente ma è necessario specificarlo per creare un oggetto date di Python.
Possiamo usare il nostro codice nel seguente modo:
def main():
birth_date = input('Insert your birth date (YYYY-mm-dd)')
sign_found = find_sign_by_birth_date(birth_date)
if sign_found is None:
print('Invalid birth date')
sys.exit(1)
print(f'Your sign is: {sign_found}')
sys.exit(0)
if __name__ == '__main__':
main()