예제 #1
0
def scrape_imdb():
    data = request.json
    if (not data):
        return {"error": "You need to include an IMDb id"}, 400

    imdb_id = data[IMDB_ID]

    if (not imdb_id):
        return {"error": "You need to include an IMDb id"}, 400

    username = session[USERNAME]
    password = session[PASSWORD]
    server = session[SERVER]

    if (not username or not password or not server):
        return {"error": "Could not find session data"}, 500

    plex_server = Plex(username, password)
    plex_server.add_resource_by_name(server)
    plex_server.connect()
    movies = plex_server.scrape_imdb("https://www.imdb.com/list/" + imdb_id)

    if (not movies):
        return {"error": "You need to include a valid IMDb id"}, 400

    session[IMDB_ID] = imdb_id

    return jsonify(movies)