# -*- coding: utf-8 -*-

# This file is part of Tautulli.
#
#  Tautulli is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
#  Tautulli is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with Tautulli.  If not, see <http://www.gnu.org/licenses/>.

import plexpy

class DummyObject(object):
    def __init__(self, *args, **kwargs):
        pass

    def __call__(self, *args, **kwargs):
        return self

    def __getattr__(self, attr):
        return self

    def __iter__(self):
        return self

    def __next__(self):
        raise StopIteration


PlexObject = DummyObject


def initialize_plexapi():
    from plexapi.server import PlexServer
    global PlexObject
    PlexObject = PlexServer


class Plex(object):
    def __init__(self, url=None, token=None):
        url = url or plexpy.CONFIG.PMS_URL
        token = token or plexpy.CONFIG.PMS_TOKEN
        self.PlexServer = PlexObject(url, token)

    def get_library(self, section_id):
        return self.PlexServer.library.sectionByID(int(section_id))

    def get_library_items(self, section_id):
        return self.get_library(section_id).all()

    def get_item(self, rating_key):
        return self.PlexServer.fetchItem(rating_key)