# scopes.storage.concept

"""Core classes for concept map structure."""

from zope.interface import implementer
from scopes.interfaces import IConcept
from scopes.storage.common import registerContainerClass, registry
from scopes.storage.tracking import Container, Track


@implementer(IConcept)
class Concept(Track):

    headFields = ['name']


class Concepts(Container):

    insertOnChange = False
    indexes = None
 

class Predicate(Concept):

    prefix = 'pred'


@registerContainerClass
class Predicates(Concepts):

    itemFactory = Predicate
    tableName = 'preds'


class Triple(Track):

    headFields = ['first', 'second', 'predicate']
    prefix = 'rel'


@registerContainerClass
class Rels(Container):

    itemFactory = Triple
    indexes = [('first', 'predicate', 'second'),
               ('first', 'second'), ('predicate', 'second')]
    tableName = 'rels'
    insertOnChange = False


# types stuff

class Type(Concept):

    headFields = ['name', 'tprefix']
    prefix = 'type'

    def get(self, key, default=None):
        cont = self.container.storage.getContainer(self.tprefix)
        return cont.queryLast(name=key) or default

    def values(self):
        cont = self.container.storage.getContainer(self.tprefix)
        return cont.query()


@registerContainerClass
class Types(Concepts):

    itemFactory = Type
    indexes = [('name',), ('tprefix',)]
    tableName = 'types'


def storeType(storage, cls, name):
    types = storage.create(Types)
    types.save(Type(name, cls.prefix))
    storage.commit()

def setupCoreTypes(storage):
    for c in registry.values():
        cls = c.itemFactory
        storeType(storage, cls, cls.__name__.lower())