Add initial Trace model

1 files changed, 17 insertions, 2 deletions

diff --git a/app.py b/app.py
index 5dbc05f..3c79d2d 100644
--- a/app.py
+++ b/app.py
@@ -5,14 +5,29 @@ from flask.ext.sqlalchemy import SQLAlchemy
 from os import environ
 
 app = Flask(__name__)
-app.config['SQLALCHEMY_DATABASE_URI'] = environ.get('DATABASE_URL')
+app.config['SQLALCHEMY_DATABASE_URI'] = environ.get('DATABASE_URL', 'postgres://localhost:5432/traces')
 
 db = SQLAlchemy(app)
 api = restful.Api(app)
 
+
+class Trace(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    page = db.Column(db.String(100))
+    trace = db.Column(db.Text())
+
+    def __init__(self, page, trace):
+        self.page = page
+        self.trace = trace
+
+    def __repr__(self):
+        return '{} {}'.format(self.id, self.page)
+
+
 class HelloWorld(restful.Resource):
     def get(self):
-        return {'hello': 'world'}
+        traces = Trace.query.all()
+        return traces[0].trace
 
 api.add_resource(HelloWorld, '/')