from config import EPOMSConfig
from peewee import *
import MySQLdb
import json
c = EPOMSConfig();
db_config = c.get('db')
database = MySQLDatabase( db_config.pop('db'), **db_config )
class BaseModel(Model):
class Meta:
database = database
class News(BaseModel):
id = PrimaryKeyField()
title = CharField()
url = CharField()
sitename = CharField()
content = TextField()
entities = TextField()
relevant = BooleanField()
published_time = DateTimeField()
indexed_time = DateTimeField()
filename = TextField()
def as_dict( self ):
entities = json.loads(self.entities)
return {
"id" : self.id,
def init(self):
config = EPOMSConfig()
es_nodes = config.get('elasticsearch')['nodes']
es = Elasticsearch( es_nodes )
return es
