Source code for caesar_external.data

"""
Storage module to track which classification was gotten last
"""

import os
import json


class Config:
    _config = None

    def __init__(self, name, project, workflow, last_id, **kwargs):
        self.name = name
        self.project = project
        self.workflow = workflow
        self.last_id = last_id

        self.caesar_name = kwargs.get('caesar_name', 'ext')

        self.__class__._config = self

    @staticmethod
    def _keys():
        return ['name', 'project', 'workflow', 'last_id', 'caesar_name']

    @classmethod
    def instance(cls):
        return cls._config

    @staticmethod
    def caesar_endpoint():
        return 'https://caesar.zooniverse.org:443'

    def workflow_path(self):
        return 'workflows/%d/reducers/%s/reductions' % \
            (self.workflow, self.caesar_name)

    def save(self):
        fname = '%s.json' % self.name
        with open(path(fname), 'w') as fp:

            json.dump(self.dump(), fp)

    @classmethod
    def load(cls, name):
        fname = '%s.json' % name
        with open(path(fname), 'r') as fp:
            data = json.load(fp)

        print(data)

        return cls(**data)

    def dump(self):
        data = self.__dict__
        data = {k: data[k] for k in self._keys()}
        return data

    def __str__(self):
        return str(self.dump())

    def __repr__(self):
        return str(self)


def dir():
    return os.path.abspath(os.path.dirname(__file__))


def path(fname):
    return os.path.join(dir(), fname)