Add /traces POST

1 files changed, 13 insertions, 0 deletions

diff --git a/app.py b/app.py
index ffad9fe..4eb4766 100644
--- a/app.py
+++ b/app.py
@@ -1,5 +1,6 @@
 from flask import Flask
 from flask.ext import restful
+from flask.ext.restful import reqparse
 from flask.ext.sqlalchemy import SQLAlchemy
 
 from os import environ
@@ -11,6 +12,11 @@ db = SQLAlchemy(app)
 api = restful.Api(app)
 
 
+trace_parser = reqparse.RequestParser()
+trace_parser.add_argument('page', type=str)
+trace_parser.add_argument('trace', type=str)
+
+
 class Trace(db.Model):
     id = db.Column(db.Integer, primary_key=True)
     page = db.Column(db.String(100))
@@ -32,6 +38,13 @@ class Traces(restful.Resource):
             'trace': trace.trace
         } for trace in traces]
 
+    def post(self):
+        args = trace_parser.parse_args()
+        trace = Trace(args['page'], args['trace'])
+        db.session.add(trace)
+        db.session.commit()
+        return 'ok', 201
+
 api.add_resource(Traces, '/traces')
 
 if __name__ == '__main__':