diff --git a/fablib.py b/fablib.py index 5fecc9e..98c3fc8 100644 --- a/fablib.py +++ b/fablib.py @@ -6,7 +6,7 @@ import hashlib import boto from markdown import markdown from flask import Flask, request, abort, redirect -from flask.ext.restful import Resource, Api, reqparse +from flask.ext.restful import Resource, Api, reqparse, abort as rest_abort from flask.ext.sqlalchemy import SQLAlchemy from werkzeug.security import generate_password_hash, check_password_hash @@ -129,11 +129,14 @@ api.add_resource(UserProfile, '/') class Document(Resource): def get(self, profile, document): - u = UserModel.from_username(profile) - # d = UserModel.from_username(profile) - d = DocumentModel.from_keys(profile, document) - content = trunk.get(d.content) + try: + u = UserModel.from_username(profile) + d = DocumentModel.from_keys(profile, document) + except AttributeError: + rest_abort(404) + + content = trunk.get(d.content) user = {'username': u.username, 'email': u.email} doc = {'text': content}