bardo-copyright-db/model/elastic.py

62 lines
1.2 KiB
Python

import os
import yaml
from elasticsearch_dsl import (
Index,
Document,
Keyword,
Text,
Date,
InnerDoc,
Nested
)
class BaseDoc(Document):
date_created = Date()
date_modified = Date()
def save(self, **kwargs):
return super(BaseDoc, self).save(**kwargs)
class BaseInner(InnerDoc):
date_created = Date()
date_modified = Date()
def save(self, **kwargs):
return super(BaseInner, self).save(**kwargs)
class Registration(BaseInner):
regnum = Keyword()
regdate = Date()
class Claimant(BaseInner):
name = Text(fields={'keyword': Keyword()})
claim_type = Keyword()
class Renewal(BaseDoc):
uuid = Keyword(store=True)
rennum = Keyword()
rendate = Date()
title = Text(fields={'keyword': Keyword()})
claimants = Nested(Claimant)
class Index:
name = os.environ['ES_CCR_INDEX']
class CCE(BaseDoc):
uuid = Keyword(store=True)
title = Text(fields={'keyword': Keyword()})
authors = Text(multi=True)
publishers = Text(multi=True)
lccns = Keyword(multi=True)
registrations = Nested(Registration)
class Index:
name = os.environ['ES_CCE_INDEX']