如何在Python中从JSON文件中返回基于语言名的文件扩展名？

import json

{
  "languages": [
    {
      "name": "Python",
      "extension": ".py"
    },
    {
      "name": "JavaScript",
      "extension": ".js"
    },
    {
      "name": "Java",
      "extension": ".java"
    }
  ]
}

def get_extension(language_name):
    with open('languages.json') as file:
        data = json.load(file)
        languages = data['languages']
        for language in languages:
            if language['name'] == language_name:
                return language['extension']
        return "Language not found"

language_name = input("Enter language name: ")
extension = get_extension(language_name)
print("File extension:", extension)