Python IMDbPY – Getting seasons from the episodes info-set of the series

In this article we will see how we can get the seasons of the series from the episodes. Each series have seasons and each season has multiple episodes i.e episode is the subset of season and season is the subset of series. We get the episodes by adding episodes info set to the series

In order to get this we have to do the following – 
1. Get the series details with the help of get_movie method 
2. Add episodes info-set to it with the help of update method 
3. As this object will act as dictionary therefore we have to filter the object 
4. Get the main data of object with the help of data method which will return dictionary 
5. Get the episodes from the dictionary the keys of this dictionary will be seasons

Below is the implementation 

Python3

# importing the module
import imdb

# creating instance of IMDb
ia = imdb.IMDb()

# id
code = "6468322"

# getting information
series = ia.get_movie(code)

# adding new info set
ia.update(series, 'episodes')

# getting episodes of the series
episodes = series.data['episodes']

# printing the object i.e name
print(series)

# printing the keys of this dictionary
for i in episodes.keys():
    print(i)

Output :

Money Heist
4
1
2
3

Another example 

Python3

# importing the module
import imdb

# creating instance of IMDb
ia = imdb.IMDb()

# id
code = "0903747"

# getting information
series = ia.get_movie(code)

# adding new info set
ia.update(series, 'episodes')

# getting episodes of the series
episodes = series.data['episodes']

# printing the object i.e name
print(series)

# printing the keys of this dictionary
for i in episodes.keys():
    print(i)

Output :

Breaking Bad
5
1
2
3
4