mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-18 09:10:16 +00:00
2702bf9400
* Enable ruff S113 rule * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
31 lines
937 B
Python
31 lines
937 B
Python
from __future__ import annotations
|
|
|
|
import csv
|
|
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
|
def get_imdb_top_250_movies(url: str = "") -> dict[str, float]:
|
|
url = url or "https://www.imdb.com/chart/top/?ref_=nv_mv_250"
|
|
soup = BeautifulSoup(requests.get(url, timeout=10).text, "html.parser")
|
|
titles = soup.find_all("td", attrs="titleColumn")
|
|
ratings = soup.find_all("td", class_="ratingColumn imdbRating")
|
|
return {
|
|
title.a.text: float(rating.strong.text)
|
|
for title, rating in zip(titles, ratings)
|
|
}
|
|
|
|
|
|
def write_movies(filename: str = "IMDb_Top_250_Movies.csv") -> None:
|
|
movies = get_imdb_top_250_movies()
|
|
with open(filename, "w", newline="") as out_file:
|
|
writer = csv.writer(out_file)
|
|
writer.writerow(["Movie title", "IMDb rating"])
|
|
for title, rating in movies.items():
|
|
writer.writerow([title, rating])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
write_movies()
|