from zeit.brightcove.convert import DeletedVideo
from zeit.cms.workflow.interfaces import IPublish, IPublishInfo
import gocept.runner
import logging
import z3c.celery
import zeit.brightcove.convert
import zeit.brightcove.session
import zeit.cms.interfaces
import zeit.content.video.playlist
import zeit.content.video.video


log = logging.getLogger(__name__)


class import_video(object):
    """Updates the CMS state to the given BC state, by exactly one of:
    deleting the CMS object, retracting deactivated objects, adding a new CMS
    object or updating the existing CMS object.

    This is a class only to better organize the code.
    """

    cms_class = zeit.content.video.video.Video

    def __init__(self, bcobj):
        log.debug('Import for video %s', bcobj.uniqueId)
        self.bcobj = bcobj
        self.cmsobj = zeit.cms.interfaces.ICMSContent(
            self.bcobj.uniqueId, None)
        log.debug('CMS object resolved: %r', self.cmsobj)
        success = (self.delete() or self.deactivate() or
                   self.add() or self.update())
        if not success:
            log.warning('Not processed: %s', self.bcobj.uniqueId)

    def delete(self):
        if not isinstance(self.bcobj, DeletedVideo):
            return False
        elif self.cmsobj is None:
            # Deleted in BC and no CMS object: we're done.
            return True
        log.info('Deleting %s', self.bcobj)
        if IPublishInfo(self.cmsobj).published:
            IPublish(self.cmsobj).retract(async=False)
        del self.bcobj.__parent__[self.bcobj.id]
        return True

    def deactivate(self):
        if self.bcobj.state != 'INACTIVE' or self.cmsobj is None:
            return False
        log.info('Deactivating %s', self.bcobj)
        if IPublishInfo(self.cmsobj).published:
            IPublish(self.cmsobj).retract(async=False)
        return True

    def add(self):
        if self.cmsobj is not None or self.bcobj.skip_import:
            return False
        self._add()
        if self.bcobj.state == 'ACTIVE':
            IPublish(self.cmsobj).publish(async=False)
        return True

    def _add(self):
        log.info('Adding %s', self.bcobj)
        folder = self.bcobj.__parent__
        cmsobj = self.cms_class()
        self.bcobj.apply_to_cms(cmsobj)
        folder[self.bcobj.id] = cmsobj
        self.cmsobj = folder[self.bcobj.id]

    def update(self):
        if self.bcobj.skip_import:
            return True
        self._update()
        if self.bcobj.state == 'ACTIVE':
            IPublish(self.cmsobj).publish(async=False)

    def _update(self):
        log.info('Updating %s', self.bcobj)
        # We overwrite last_semantic_change with the BC value.
        with zeit.cms.checkout.helper.checked_out(
                self.cmsobj, semantic_change=None, events=False) as co:
            # Don't send events here, a full checkout/checkin cycle is done
            # during publish anyway, directly below (and so we don't publish
            # twice due to the publish_on_checkin handler).
            if co is None:
                log.warning('Could not checkout %s', self.cmsobj)
            else:
                self.bcobj.apply_to_cms(co)


# Triggered by BC notification webhook, which we receive in
# zeit.brightcove.json.update.Notification
@z3c.celery.task(queuename='brightcove')
def import_video_async(video_id):
    import_video(zeit.brightcove.convert.Video.find_by_id(video_id))


def export_video(context, event):
    if not event.publishing:
        session = zeit.brightcove.session.get()
        session.update_video(zeit.brightcove.convert.Video.from_cms(context))


def publish_on_checkin(context, event):
    # prevent infinite loop, since there is a checkout/checkin cycle during
    # publishing (to update XML references etc.)
    if not event.publishing:
        zeit.cms.workflow.interfaces.IPublish(context).publish()


class import_playlist(import_video):
    # Inheriting from import_video is only mechanical, so we can reuse
    # _add() and _update(), we actually don't have anything else in common.

    cms_class = zeit.content.video.playlist.Playlist

    def __init__(self, bcobj):
        log.debug('Import for playlist %s', bcobj.uniqueId)
        self.bcobj = bcobj
        self.cmsobj = zeit.cms.interfaces.ICMSContent(
            self.bcobj.uniqueId, None)
        log.debug('CMS object resolved: %r', self.cmsobj)
        success = self.add() or self.update()
        if not success:
            log.warning('Not processed: %s', self.bcobj.uniqueId)

    def add(self):
        if self.cmsobj is not None:
            return False
        self._add()
        IPublish(self.cmsobj).publish(async=False)
        return True

    def update(self):
        lsc = zeit.cms.content.interfaces.ISemanticChange(
            self.cmsobj).last_semantic_change
        if self.bcobj.updated_at <= lsc:
            return False
        self._update()
        IPublish(self.cmsobj).publish(async=False)

    @classmethod
    def generate_actions(cls):
        """Fulfills the gocept.runner.transaction_per_item API by returning an
        iterable of callables.
        """
        playlists = zeit.brightcove.convert.Playlist.find_all()
        for item in playlists:
            yield lambda: cls(item)
        yield lambda: cls.delete_except(playlists)

    @classmethod
    def delete_except(cls, known):
        """There is no way in the BC API to get data about deleted playlists,
        so we fall back on deleting any playlist that BC does not know about.
        """
        if not known:  # safetybelt
            return
        folder = zeit.brightcove.convert.playlist_location(None)
        cms_names = set(folder.keys())
        bc_names = set(x.id for x in known)
        for name in cms_names - bc_names:
            log.info('Deleting <Playlist id=%s>', name)
            cmsobj = folder[name]
            if IPublishInfo(cmsobj).published:
                IPublish(cmsobj).retract(async=False)
            del folder[name]


@gocept.runner.appmain(ticks=120, principal=gocept.runner.from_config(
    'zeit.brightcove', 'index-principal'))
def import_playlists():
    log.info('Update run started')
    _import_playlists()
    log.info('Update run finished')


@gocept.runner.transaction_per_item
def _import_playlists():
    return import_playlist.generate_actions()

@@ -150,18 +150,23 @@
 import z
-3c
+eit.cms
 .celery%0A
@@ -3317,10 +3317,15 @@
 n%0A@z
-3c
+eit.cms
 .cel

Use zeit.cms API for celery