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