Changed from the plex scripts doing the comparisson to letting the client or another seperate script to it.

This commit is contained in:
2017-02-18 11:52:59 +01:00
parent 659f306907
commit 98eb2fff1d
3 changed files with 51 additions and 67 deletions

8
app.py
View File

@@ -52,7 +52,7 @@ def get_pw(username):
# to not match.
@auth.error_handler
def unauthorized():
return make_response(jsonify({'error': 'Unauthorized access'}), 401)
return make_response(jsonify({'errors': 'Unauthorized access'}), 401)
# This would be replaced with a database, but single process and thread
# can use local data like this for simplicity.
@@ -61,10 +61,10 @@ def unauthorized():
# Want all return data to be JSON so create custom error response
@app.errorhandler(404)
def not_found(error):
return make_response(jsonify({'error': 'Not found'}), 404)
return make_response(jsonify({'errors': 'Not found'}), 404)
@app.errorhandler(400)
def bad_request(error):
return make_response(jsonify({'error': 'Bad request'}), 400)
return make_response(jsonify({'errors': 'Bad request'}), 400)
# --- Apollo Activity --- #
@@ -96,7 +96,7 @@ def get_movieRequest():
# TODO if list is empty
return jsonify(tmdbSearch(query))
else: return jsonify({ "Error": "Query not defined." })
else: return jsonify({ "errors": "Query not defined." })
@app.route('/api/v1/plex/movies', methods=['GET'])
def getPlexMovies():

View File

@@ -3,9 +3,10 @@
# @Author: KevinMidboe
# @Date: 2017-02-08 14:00:04
# @Last Modified by: KevinMidboe
# @Last Modified time: 2017-02-10 01:31:43
# @Last Modified time: 2017-02-18 11:37:08
from requests import get
import requests
from pprint import pprint
plexBaseURL = "http://10.0.0.41:32400/"
@@ -42,21 +43,36 @@ def getShowInfo(item):
return {"title":title, "year":year, "seasons":seasons, "episodes":episodes, "rating":rating,
"art":art, "thumb":thumb}
def plexSearch(query):
payload = {"query":query, "type":"1,2"}
header = {'Accept': 'application/json'}
r = requests.get(plexBaseURL+"search",params=payload, headers=header)
rObject = r.json()["MediaContainer"]
if (r.status_code != requests.codes.ok or rObject["size"] == 0):
return {"errors": "Nothing found for query: " + query}
return rObject["Metadata"]
## MAJOR TODO
# Seems to be a change in the return obj.
# This looks way more like json. Need to re-write all this.
# IDEA: Send the size and resolution for comaprison
# No this is for a admin page. OR maybe a github project for
# people wanting to update movies. MAJOR IDEA HERE NOW! :D
def plexSearch(query):
def plexXMLSearch(query):
print(query)
requestType = "search?"
requestQuery = "query=" + str(query)
header = {'Accept': 'application/json'}
url = plexBaseURL + requestType + requestQuery
response = get(url, headers=header)
print(response.json())
response = requests.get(url, headers=header)
pprint(response.json())
if response.status_code == 200:
resContent = response.json()
media = []
@@ -74,7 +90,4 @@ def plexSearch(query):
if __name__ == "__main__":
# print(plexSearch("star+wars"))
tiss = plexSearch("star+wars")
for al in tiss:
if (al['year']==2015):
print('thishsihis')
pprint(plexSearch("star"))

View File

@@ -3,77 +3,48 @@
# @Author: KevinMidboe
# @Date: 2017-02-08 14:00:04
# @Last Modified by: KevinMidboe
# @Last Modified time: 2017-02-08 23:19:53
# @Last Modified time: 2017-02-16 17:08:08
from requests import get
import requests
from pprint import pprint
try:
from plexSearch import plexSearch
except ImportError:
from plex.plexSearch import plexSearch
tmdbBaseURL = "https://api.themoviedb.org/3/"
def checkPlexExistance(tmdb, plex):
# THIS IS ONLY COMPARED ON YEAR
yearList = [movie["year"] for movie in plex]
print(yearList)
for movie in tmdb["movies"]:
print(movie["year"])
movie["exists"] = movie["year"] in str(yearList)
return tmdb
def createTMDBResultList(resContent):
movies = []
tvshows = []
for res in resContent["results"]:
if res["media_type"] == "movie":
id = res["id"]
title = res["original_title"]
year = res["release_date"][:4]
poster_path = res["poster_path"]
movies.append({"id": id, "title": title, "year": year, "poster_path": poster_path})
elif res["media_type"] == "tv":
id = res["id"]
name = res["original_name"]
year = res["first_air_date"][:4]
poster_path = res["poster_path"]
tvshows.append({"id": id, "title": name, "year": year, "poster_path": poster_path})
return { "movies": movies, "tvshows": tvshows }
apiKey = "9fa154f5355c37a1b9b57ac06e7d6712"
def tmdbSearch(query, page=1):
requestType = "search/multi?"
requestAPI = "api_key=" + "9fa154f5355c37a1b9b57ac06e7d6712"
requestQuery = "&query=" + str(query)
requestLanguage = "&language=en.US"
requestPage = "&page="+str(page)
payload = {"api_key":apiKey, "query":str(query), "language":"en.US", "page":str(page), }
header = {'Accept': 'application/json'}
url = tmdbBaseURL + requestType + requestAPI + requestQuery + requestLanguage + requestPage
print(url)
try:
r = requests.get("https://api.themoviedb.org/3/search/multi", params=payload, headers=header)
except requests.exceptions.ConnectionError:
return {"errors": "Could not connecting to: tmdb.com"}
except requests.exceptions.Timeout:
return {"errors": "Request timed out."}
except requests.exceptions.TooManyRedirects:
return {"errors": "Too many redirects, do you full network access?"}
response = get(url)
if response.status_code == 200:
resContent = response.json()
if r.status_code == 401:
return {"errors": "api key is not valid."}
elif r.status_code == 404:
return {"errors": "Please check url. (404)"}
elif r.status_code == requests.codes.ok and r.json()['total_results'] == 0:
return {"errors": "No results found."}
plexSearchRes = plexSearch(query)
tmdbSearchRes = createTMDBResultList(resContent)
return checkPlexExistance(tmdbSearchRes, plexSearchRes)
return r.json()
if __name__ == "__main__":
import sys
print(sys.argv)
if len(sys.argv) > 2:
print(tmdbSearch(sys.argv[1], int(sys.argv[2])))
pprint(tmdbSearch(sys.argv[1], int(sys.argv[2])))
elif len(sys.argv) > 1:
print(tmdbSearch(sys.argv[1]))
pprint(tmdbSearch(sys.argv[1]))
else:
print(tmdbSearch("star+wars",1))
pprint(tmdbSearch("star+wars",1))