-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathkeywordPlaylist.py
133 lines (114 loc) · 3.96 KB
/
keywordPlaylist.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#!/usr/bin/python3
import requests
import json
import time
from plexapi.server import PlexServer
from lxml import html
PLEX_URL = ""
PLEX_TOKEN = ""
MOVIE_LIBRARY_NAME = "Films"
KIDSMOVIE_LIBRARY_NAME = "Kids Films"
TMDB_API_KEY = ""
PLAYLIST_NAME = 'Christmas Films'
TMDB_REQUEST_COUNT = 0
playlist_items = []
KEYWORDS = ['xmas','christmas','santa']
def get_tmdb_id_from_imdb(imdb_id):
global TMDB_REQUEST_COUNT
#print("Looking up TMDB ID for :"+imdb_id)
if not TMDB_API_KEY:
return None
# Wait 10 seconds for the TMDb rate limit
if TMDB_REQUEST_COUNT >= 40:
print("Rate limiting, waiting 10 seconds")
time.sleep(10)
TMDB_REQUEST_COUNT = 0
params = {"api_key": TMDB_API_KEY, "external_source":"imdb_id"}
url = "https://api.themoviedb.org/3/find/"+imdb_id
r = requests.get(url, params=params)
TMDB_REQUEST_COUNT += 1
if r.status_code == 200:
try:
movie = json.loads(r.text)
results = movie['movie_results'][0]
return results['id']
except:
return None
else:
return None
def checkXmas(tmdb_id):
try:
tmdb_id = int(tmdb_id)
except:
tmdb_id = 0
if tmdb_id > 0:
#print("Checking : "+str(tmdb_id))
res = requests.get("https://api.themoviedb.org/3/movie/"+str(tmdb_id)+"/keywords?api_key="+TMDB_API_KEY+"&language=en-US")
movie = res.json()
try:
for keyword in movie['keywords']:
#print(keyword['name'])
if any(matchingkeyword in keyword['name'] for matchingkeyword in KEYWORDS):
return True
except:
print("No Keywords found")
return False
return False
try:
plex = PlexServer(PLEX_URL, PLEX_TOKEN)
except:
print("No Plex server found at: {base_url}".format(base_url=PLEX_URL))
print("Exiting script.")
print("Retrieving a list of movies from the '{library}' library in Plex...".format(library=MOVIE_LIBRARY_NAME))
try:
movie_library = plex.library.section(MOVIE_LIBRARY_NAME)
movie_library_key = movie_library.key
library_language = movie_library.language
movies = movie_library.all()
except:
print("The '{library}' library does not exist in Plex.".format(library=MOVIE_LIBRARY_NAME))
#REMOVE THIS IF YOU ONLY HAVE 1 FILM LIBRARY
print("Retrieving a list of movies from the '{library}' library in Plex...".format(library=KIDSMOVIE_LIBRARY_NAME))
try:
movie_library = plex.library.section(KIDSMOVIE_LIBRARY_NAME)
movie_library_key = movie_library.key
library_language = movie_library.language
kids_movies = movie_library.all()
except:
print("The '{library}' library does not exist in Plex.".format(library=KIDSMOVIE_LIBRARY_NAME))
for m in movies:
imdb_id = "0"
tmdb_id = "0"
if 'imdb://' in m.guid:
imdb_id = m.guid.split('imdb://')[1].split('?')[0]
tmdb_id = get_tmdb_id_from_imdb(imdb_id)
elif 'themoviedb://' in m.guid:
tmdb_id = m.guid.split('themoviedb://')[1].split('?')[0]
try:
print(" Checking : " + m.title)
except:
print(" Non Ascii in title")
if checkXmas(tmdb_id):
print("CHRISTMAS FILM FOUND!!")
playlist_items += [m]
for m in kids_movies:
imdb_id = "0"
tmdb_id = "0"
if 'imdb://' in m.guid:
imdb_id = m.guid.split('imdb://')[1].split('?')[0]
tmdb_id = get_tmdb_id_from_imdb(imdb_id)
elif 'themoviedb://' in m.guid:
tmdb_id = m.guid.split('themoviedb://')[1].split('?')[0]
try:
print(" Checking : " + m.title)
except:
print(" Non Ascii in title")
if checkXmas(tmdb_id):
print("MATCHING FILM FOUND!!")
playlist_items += [m]
for playlist in plex.playlists():
if playlist.title == PLAYLIST_NAME:
print('Removing old Playlist')
playlist.delete()
print('Creating New Playlist')
plex.createPlaylist(PLAYLIST_NAME, playlist_items)