Comment vérifier si un objet datetime est localisé avec pytz ?

Question

Je veux stocker un objet datetime avec un fuseau horaire UTC localisé. La méthode qui stocke l'objet datetime peut recevoir un objet datetime (naïf) non localisé ou un objet qui a déjà été localisé. Comment puis-je déterminer si la localisation est nécessaire ?

Code avec condition if manquante :

class MyClass:
  def set_date(self, d):
    # what do i check here?
    # if(d.tzinfo):
      self.date = d.astimezone(pytz.utc)
    # else:
      self.date = pytz.utc.localize(d)

Answer 1

Comment déterminer si une localisation est nécessaire ?

De datetime docs :

• un objet date-heure d est conscient si :

  d.tzinfo is not None and d.tzinfo.utcoffset(d) is not None

• d est naïf si :

  d.tzinfo is None or d.tzinfo.utcoffset(d) is None

Bien que si d est un objet datetime représentant le temps dans le fuseau horaire UTC, vous pouvez l'utiliser dans les deux cas :

self.date = d.replace(tzinfo=pytz.utc)

Cela fonctionne indépendamment d est sensible au fuseau horaire ou naïf.

Note : n'utilisez pas datetime.replace() méthode avec un fuseau horaire avec un décalage utc non fixe (il est possible de l'utiliser avec le fuseau horaire UTC, mais sinon vous devriez utiliser tz.localize() méthode).

Answer 2

Si vous voulez vérifier si un objet datetime 'd' est localisé, vérifiez le d.tzinfo, si c'est None, pas de localisation.

Answer 3

Voici une fonction qui résume la première réponse.

def tz_aware(dt):
    return dt.tzinfo is not None and dt.tzinfo.utcoffset(dt) is not None

Answer 4

Voici une fonction plus complète pour convertir ou contraindre un timestamp obj en utc. Si elle atteint l'exception, cela signifie que le timestamp n'est pas localisé. Puisque c'est une bonne pratique de toujours travailler en UTC dans le code, cette fonction est très utile au niveau d'entrée de la persistance.

def convert_or_coerce_timestamp_to_utc(timeobj):
        out = timeobj
        try:
            out = timeobj.astimezone(pytz.utc) # aware object can be in any timezone
        except (ValueError,TypeError) as exc: # naive
            out = timeobj.replace(tzinfo=pytz.utc)
        return out

Le petit ajout par rapport au "try catch" de la réponse de J.F. Sebastian est la condition supplémentaire "catch", sans laquelle tous les cas naïfs ne seront pas pris en compte par la fonction.