diff --git a/fablib.py b/fablib.py index b3f57a3..2275a34 100644 --- a/fablib.py +++ b/fablib.py @@ -202,14 +202,28 @@ api.add_resource(Document, '//') class DocumentText(Resource): def get(self, profile, document): - return trunk.get(key) + try: + u = UserModel.from_username(profile) + d = DocumentModel.from_keys(profile, document) + + except AttributeError: + rest_abort(404) + + return trunk.get(d.content) api.add_resource(DocumentText, '///text') class DocumentHTML(Resource): def get(self, profile, document): - return trunk.get(key) + try: + u = UserModel.from_username(profile) + d = DocumentModel.from_keys(profile, document) + + except AttributeError: + rest_abort(404) + + return trunk.get(d.content, render=True) api.add_resource(DocumentHTML, '///html')