fuzeman/trakt.py

View on GitHub
trakt/interfaces/sync/collection.py

Summary

Maintainability
A
35 mins
Test Coverage


from trakt.interfaces.sync.core.mixins import Get, Add, Remove


class SyncCollectionInterface(Get, Add, Remove):
    path = 'sync/collection'
    flags = {'is_collected': True}

    def get(self, media=None, store=None, params=None, extended=None, **kwargs):
        if media is None:
            raise ValueError('Invalid value provided for the "media" parameter')

        # Build query
        query = {}

        if extended:
            query['extended'] = extended

        # Request collection
        return super(SyncCollectionInterface, self).get(
            media, store,
            params=params,
            query=query,
            **kwargs
        )