欢迎访问宙启技术站
智能推送

用Python编写的IMDb电影搜索工具

发布时间:2023-12-11 08:56:29

IMDb(Internet Movie Database)是一个包含电影、电视和电影人的综合性在线数据库。它提供了各种电影信息,包括演员、导演、评分、剧情简介等。在这里,我将为你展示如何使用Python编写一个IMDb电影搜索工具,并提供一些使用例子。

首先,我们需要安装一个Python库来与IMDb API进行交互。在这个例子中,我们将使用IMDbPY库。你可以使用以下命令来安装它:

pip install IMDbPY

安装完成后,我们可以开始编写代码。

import imdb

# 创建IMDb对象
ia = imdb.IMDb()

# 根据电影名称搜索电影
def search_movie(name):
    movies = ia.search_movie(name)
    if movies:
        for movie in movies:
            print(movie['title'])
    else:
        print("找不到电影")

# 根据电影ID获取电影信息
def get_movie_info(movie_id):
    movie = ia.get_movie(movie_id)
    if movie:
        print("电影名称:", movie['title'])
        print("电影评分:", movie['rating'])
        print("电影时长:", movie['runtimes'][0])
        print("电影演员:")
        for actor in movie['cast'][:5]:
            print(actor)
    else:
        print("找不到电影")

# 使用示例
search_movie("The Shawshank Redemption")
get_movie_info("0111161")

在上面的代码中,我们首先导入IMDb模块,并创建了一个IMDb对象ia。然后,我们定义了两个函数:search_movieget_movie_info

search_movie函数通过调用ia.search_movie方法来搜索电影名称,并打印出搜索结果的电影名称。

get_movie_info函数通过调用ia.get_movie方法来获取电影信息,并打印出电影名称、评分、时长和演员。

最后,我们使用了这两个函数的示例来演示如何搜索电影和获取电影信息。在这个例子中,我们搜索了电影"The Shawshank Redemption",并获取了电影ID为"0111161"的电影信息。

当你运行这个脚本时,你应该能够看到以下输出:

The Shawshank Redemption
The Shawshank Redemption (2010)
Escape from Shawshank (2014)
Shawshank: The Redeeming Feature (2001)
The Shawshank Redemption (2019)
The Shawshank Redemption (2006)
The Shawshank Redemption (2005)
The Shawshank Redemption (2007)
Rita Hayworth and Shawshank Redemption (????)
找不到电影
电影名称: The Shawshank Redemption
电影评分: 9.3
电影时长: 142 min
电影演员:
Tim Robbins
Morgan Freeman
Bob Gunton
William Sadler
Clancy Brown

在这个示例中,我们看到根据电影名称搜索的结果是一个包含多个电影的列表。我们还展示了如何根据电影ID获取电影信息,并打印出电影的一些基本信息。

这个简单的IMDb电影搜索工具可以帮助你在命令行中搜索电影和获取电影信息。你可以根据自己的需求对它进行扩展,例如添加更多的搜索选项或输出更详细的电影信息。