mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-11-24 12:31:08 +00:00
114 lines
3.0 KiB
Python
114 lines
3.0 KiB
Python
|
"""
|
|||
|
|
Parse strings obtained from the html to get the film metadata.
|
|||
|
|
Fix metadata and create an film object to be use it later.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from ClassFilm import Film
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Eliminate parenthesis from the text.
|
|||
|
|
'(2019)' -> '2019'
|
|||
|
|
"""
|
|||
|
|
def parse_film_year(year_text):
|
|||
|
|
found_numbers = re.findall("[0-9]", year_text)
|
|||
|
|
return ''.join(found_numbers[0:4])
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Obtain decimal value of the score from its text.
|
|||
|
|
'7' -> 70
|
|||
|
|
'7,9'-> 79
|
|||
|
|
"""
|
|||
|
|
def parse_imdb_score(score_text):
|
|||
|
|
units_digit = 0
|
|||
|
|
if ',' in score_text:
|
|||
|
|
tens_digit, units_digit = score_text.split(',')
|
|||
|
|
else:
|
|||
|
|
tens_digit = score_text.split(',')[0]
|
|||
|
|
return int(tens_digit) * 10 + int(units_digit)
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Parse runtime in minutes from runtime text.
|
|||
|
|
"134 min" -> 134
|
|||
|
|
"""
|
|||
|
|
def parse_runtime(runtime_text):
|
|||
|
|
return runtime_text.split(' ')[0]
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
From the string of genres, obtain the genres list.
|
|||
|
|
Remove extra spaces and new line characters.
|
|||
|
|
Return genres in a list.
|
|||
|
|
"""
|
|||
|
|
def obtain_all_genres(genres_text):
|
|||
|
|
obtained_genres = []
|
|||
|
|
for genre in genres_text.split(','):
|
|||
|
|
obtained_genres.append(genre.replace('\n', '').replace(' ', ''))
|
|||
|
|
return obtained_genres
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Storyline obtained as text from the html yet some characters must be deleted
|
|||
|
|
from it.
|
|||
|
|
"""
|
|||
|
|
def obtain_story_line(story_text):
|
|||
|
|
return story_text.replace('\n', '')
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Determine the film type from the year text.
|
|||
|
|
A TV-series will include '-' but a film will not include.
|
|||
|
|
"""
|
|||
|
|
def determine_film_type(year_text):
|
|||
|
|
if '–' in year_text:
|
|||
|
|
return 'tv-series'
|
|||
|
|
return 'film'
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Sometimes images cannot be loaded and its src will be a placeholder.
|
|||
|
|
For such cases, loadlate tag will be the real source.
|
|||
|
|
"""
|
|||
|
|
def obtain_image_source(img_html):
|
|||
|
|
if 'loadlate' in img_html.attrs:
|
|||
|
|
return img_html['loadlate']
|
|||
|
|
else:
|
|||
|
|
return img_html['src']
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
Take a html block representing the film item
|
|||
|
|
Apply parsing and return film object
|
|||
|
|
"""
|
|||
|
|
def obtain_film_object(content, image_raw):
|
|||
|
|
# Runtime and score of a film might not given in the list item.
|
|||
|
|
runtime = "unknown"
|
|||
|
|
point = "unknown"
|
|||
|
|
|
|||
|
|
raw_name_with_link = content.find('a')
|
|||
|
|
raw_name = raw_name_with_link.text
|
|||
|
|
film_imdb_link = raw_name_with_link['href']
|
|||
|
|
raw_year = content.find("span", class_="lister-item-year text-muted unbold").text
|
|||
|
|
raw_runtime = content.find("span", class_="runtime")
|
|||
|
|
|
|||
|
|
if raw_runtime is not None:
|
|||
|
|
raw_runtime = raw_runtime.text
|
|||
|
|
runtime = int(parse_runtime(raw_runtime))
|
|||
|
|
|
|||
|
|
raw_genre = content.find("span", class_="genre").text
|
|||
|
|
raw_point = content.find("span", class_="ipl-rating-star__rating")
|
|||
|
|
|
|||
|
|
if raw_point is not None:
|
|||
|
|
raw_point = raw_point.text
|
|||
|
|
point = int(parse_imdb_score(raw_point))
|
|||
|
|
|
|||
|
|
raw_storyline = content.find("p", class_="").text
|
|||
|
|
|
|||
|
|
year = parse_film_year(raw_year)
|
|||
|
|
genre_list = obtain_all_genres(raw_genre)
|
|||
|
|
storyline = obtain_story_line(raw_storyline)
|
|||
|
|
f_type = determine_film_type(year)
|
|||
|
|
image_source = obtain_image_source(image_raw)
|
|||
|
|
|
|||
|
|
return Film(raw_name, year, point, genre_list, runtime, storyline, f_type, image_source, film_imdb_link)
|