un programa que pida un código de 3 letras al inicio y 5 números con if

def code():
  code_1 = input("Enter Code: ")
  if (code_1 == "jav"):
    code_2 = int(input("Enter Code: "))
    if (code_2 == 12345):
      print("Ok")
    else:
      print("Wrong Code")
  else:
    print("Wrong Code")