错误：ValueError：时间数据“ N / A”与格式“％Y-％m-％d”不匹配-Python

Question

I am a junior in DevOps and I need to create a code that returns some results from spreadsheet, therefore I have updated this code that I found from internet.

The result:

Result code without error

#!/usr/bin/env python
from sys import stdin
import datetime
import json
import base64
from datetime import date, timedelta

dt = date.today()

report = json.loads(stdin.read())
table = base64.b64decode(report["Content"]).splitlines()
head = table[0].split(",")
table = table[1:]

for row in iter(table):
    user = dict(zip(head, row.split(",")))
    print ("\033[1m  USER: \033[0m \t  %s \033[1m  MFA_ACTIVE: \033[0m  %s  \033[1m PASSWORD ENABLED: \033[0m  %s  \033[1m  KEY ROTATED: \033[0m %s \n") % (user["user"], user["mfa_active"], user["password_enabled"], ((dt-(datetime.datetime.strptime((user["access_key_1_last_rotated"]).split('T')[0],"%Y-%m-%d").date()))))

My issue is, when I have a data like N/A for access_key_1_last_rotated , it returns a error:

ValueError: time data 'N/A' does not match format '%Y-%m-%d'

How can I customize the code to ignore a data like N/A ?

Answer 1

In general, I'd move

((dt-(datetime.datetime.strptime((user["access_key_1_last_rotated"]).split('T')[0],"%Y-%m-%d").date()))))

out of the print statement and assign it to a variable. Then, quick and dirty, you can wrap in a try-except

var my_date = "N/A"
try:
    my_date = ((dt-(datetime.datetime.strptime((user["access_key_1_last_rotated"]).split('T')[0],"%Y-%m-%d").date()))))
except ValueError as v:
    # print v somewhere
    # print(v)
    # or just pass
    pass
print ("\033[1m  USER: \033[0m \t  %s \033[1m  MFA_ACTIVE: \033[0m  %s  \033[1m PASSWORD ENABLED: \033[0m  %s  \033[1m  KEY ROTATED: \033[0m %s \n") % (user["user"], user["mfa_active"], user["password_enabled"], my_date)

There's quite a lot going on in that date calculation so you may want to break it down further if dates start appearing in strange formats