用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_movie和get_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电影搜索工具可以帮助你在命令行中搜索电影和获取电影信息。你可以根据自己的需求对它进行扩展,例如添加更多的搜索选项或输出更详细的电影信息。
