From 6791faf8d2f5ce670f43ed612c8109d15756c31f Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.com>
Date: Tue, 24 Sep 2013 22:58:21 -0400
Subject: [PATCH] one more time

---
 fablib.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

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, '/<string:profile>/<path: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, '/<string:profile>/<path:document>/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, '/<string:profile>/<path:document>/html')