diff --git a/models.py b/models.py index 7ac907d..6b94dcc 100644 --- a/models.py +++ b/models.py @@ -1,3 +1,4 @@ +import json from peewee import * db = SqliteDatabase('violence.db') @@ -6,6 +7,15 @@ class BaseModel(Model): class Meta: database = db + def __str__(self): + r = {} + for k in self._data.keys(): + try: + r[k] = str(getattr(self, k)) + except: + r[k] = json.dumps(getattr(self, k)) + return str(r) + class Article(BaseModel): """ An article is a single incident as crawled from the reach-out webpage diff --git a/server.py b/server.py new file mode 100644 index 0000000..b2f5408 --- /dev/null +++ b/server.py @@ -0,0 +1,15 @@ +import bottle +from json import dumps +from models import * + +@bottle.get('/locations/') +def location(article_id): + return (Location + .select() + .where(Location.article == article_id) + .order_by(Location.confidence.desc(), Location.id.asc()) + .dicts() + .get()) + +if __name__ == '__main__': + bottle.run(host='localhost', port=12345, reloader=True, debug=True)