用python如何写这个程序

就是把流程图转为代码吧

y = int(input("输入年份"))
if y % 4 != 0:
    print("B")
else:
    print("A")
    if y % 10 != 0:
        print("D")
    else:
        print("C")

def print_str_by_year(year):
  if year % 4 == 0:
    print("A")
    if year % 10 == 0:
      print("C")
    else:
      print("D")
  else:
    print("B")

if __name__ == '__main__':
  year = int(input("输入年份:"))
  print_str_by_year(year)