/api/routes/Drafts.py

import json
from api.model.models import Story, User, Category
from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound
from api.authentication.AuthenticatedHandler import AuthenticatedHandler
from tornado.gen import coroutine
from api.Utils import authenticated
import datetime


class DraftsHandler(AuthenticatedHandler):
    def data_received(self, chunk):
        pass

    # GET /drafts
    @coroutine
    def get(self):

        session = self.settings['db']
        story_id = self.get_argument("id", default=None)

        if story_id is not None:

            story = session.query(Story).filter(Story.id == story_id and Story.is_draft == True).one()

            if story.category is None:
                category = {'id': -1, 'name': "uncatalogued"}
            else:
                category = {'id': story.category.id, 'name': story.category.name}

            comments = []
            for comment in story.comments:
                json_comment = {
                    'id': comment.id,
                    'author': comment.author,
                    'content': comment.content,
                    'avatar': comment.avatar,
                    'url': comment.url,
                    'date': str(comment.date)
                }

                comments.append(json_comment)

            json_author = {
                'id': story.author.id,
                'name': story.author.fullname,
                'avatar': story.author.avatar,
                'username': story.author.username,
                'signature': story.author.signature
            }

            response = {
                'id': story.id,
                'title': story.title,
                'category': category,
                'content': story.content,
                'comments': comments,
                'tags': story.tags.split(','),
                'author': json_author,
                'date': str(story.date),
                'is_draft': True
            }

            self.set_header("Content-Type", "application/jsonp;charset=UTF-8")
            self.set_header("Access-Control-Allow-Origin", "*")
            self.write(response)

            return

        else:
            all_stories = session.query(Story).filter(Story.is_draft == True).order_by(Story.id.desc()).all()
            data = []

            for story in all_stories:

                if story.category is None:
                    category = {'id': -1, 'name': "Uncategorized"}
                else:
                    category = {'id': story.category.id, 'name': story.category.name}

                json_story = {
                    'id': story.id,
                    'name': story.title,
                    'content': story.content,
                    'tags': story.tags.split(','),
                    'author': {
                        'id': story.user.id,
                        'name': story.user.username,
                        'avatar': story.user.avatar
                    },
                    'comments': len(story.comments),
                    'date': str(story.date),
                    'category': category,
                    'is_draft': story.is_draft
                }

                data.append(json_story)

            response = {"page": 1, "items": data}

            self.set_header("Content-Type", "application/jsonp;charset=UTF-8")
            self.set_header("Access-Control-Allow-Origin", "*")
            self.write(response)

            return

        # POST /stories/new

    @authenticated
    def post(self):

        request = self.request.body.decode("utf-8")
        json_request = json.loads(json.loads(request))
        title = json_request["title"]
        tags = ",".join(json_request["tags"])
        content = json.dumps(json_request["content"])
        author_id = json_request["author"]
        category_id = json_request["category"]
        is_draft = json_request['is_draft']

        session = self.settings['db']

        try:
            category = session.query(Category).filter(Category.id == category_id).one()
        except NoResultFound:
            category = None

        story = self.save_story(session, content, title, author_id, category, tags, is_draft)

        if story.category is not None:
            category = {'id': story.category.id, 'name': story.category.name}
        else:
            category = None

        json_story = {
            'id': story.id,
            'name': story.title,
            'content': story.content,
            'tags': story.tags.split(','),
            'author': {
                'id': story.user.id,
                'name': story.user.username,
                'avatar': story.user.avatar
            },
            'comments': [],
            'category': category,
            'date': str(story.date),
            'is_draft': str(is_draft)
        }

        self.set_status(200, 'Ok')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(json_story))

    @authenticated
    def put(self, story_id):
        response = {"message": "This is not a valid method for this resource."}
        self.set_status(405, 'Error')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(response))

        return

    @authenticated
    def delete(self, story_id):
        response = {"message": "This is not a valid method for this resource."}
        self.set_status(405, 'Error')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(response))

        return

    @coroutine
    def trace(self):
        response = {"message": "This is not a valid method for this resource."}
        self.set_status(405, 'Error')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(response))

        return

    @coroutine
    def connect(self):
        response = {"message": "This is not a valid method for this resource."}
        self.set_status(405, 'Error')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(response))

        return

    @coroutine
    def options(self, *args):
        response = {}
        self.set_header("Content-Type", "test/plain;charset=UTF-8")
        self.set_header("Access-Control-Allow-Origin", "*")
        self.set_header("Access-Control-Allow-Headers", "Authorization")
        self.set_header("Access-Control-Allow-Methods ", "GET, POST, PUT, DELETE, OPTIONS")
        self.set_status(200, "Ok")
        self.write(response)

        return

    @coroutine
    def patch(self):
        response = {"message": "This is not a valid method for this resource."}
        self.set_status(405, 'Error')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(response))

        return

    @coroutine
    def head(self):
        response = {"message": "This is not a valid method for this resource."}
        self.set_status(405, 'Error')
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(json.dumps(response))

        return

    @staticmethod
    def save_story(session, content, title, author_id, category, tags, is_draft):

        user = session.query(User).filter(User.id == author_id).one()
        current_date = datetime.datetime.now()

        story = Story()
        story.title = title
        story.content = str(content)
        story.category = category
        story.user_id = author_id
        story.tags = tags
        story.date = current_date
        story.is_draft = is_draft
        user.stories.append(story)
        session.commit()

        return story


class DraftsByUserHandler(AuthenticatedHandler):

    # GET /user/stories
    def get(self, user_id):

        session = self.settings['db']

        data = []

        # TODO: esto no siempre es verdadero.
        # Asumimos que el usuario siempre existe.
        user = session.query(User).filter(User.id == user_id).one()
        all_stories = user.stories.filter(Story.is_draft == False)

        for story in all_stories:

            if story.category is None:
                category = {'id': -1, 'name': "Uncategorized"}
            else:
                category = {'id': story.category.id, 'name': story.category.name}

            json_story = {
                'id': story.id,
                'name': story.title,
                'author': {
                    'id': user.id,
                    'name': user.username,
                    'avatar': user.avatar,
                    'signature': user.signature
                },
                'comments': len(story.comments),
                'category': category,
                'is_draft': story.is_draft
            }

            data.append(json_story)

        response = {"page": 1, "stories": data}

        self.set_header("Content-Type", "application/jsonp;charset=UTF-8")
        self.set_header("Access-Control-Allow-Origin", "*")
        self.write(response)