diff options
| -rw-r--r-- | pyaggr3g470r/rest.py | 57 |
1 files changed, 38 insertions, 19 deletions
diff --git a/pyaggr3g470r/rest.py b/pyaggr3g470r/rest.py index 4c95e8bf..c2e1a4d8 100644 --- a/pyaggr3g470r/rest.py +++ b/pyaggr3g470r/rest.py @@ -26,12 +26,13 @@ __revision__ = "$Date: 2014/06/18 $" __copyright__ = "Copyright (c) Cedric Bonhomme" __license__ = "AGPLv3" +import dateutil from functools import wraps from flask import g, Response, request, session, jsonify from flask.ext.restful import Resource, reqparse from pyaggr3g470r import api, db -from pyaggr3g470r.models import User, Article +from pyaggr3g470r.models import User, Article, Feed def authenticate(func): """ @@ -68,9 +69,6 @@ class ArticleListAPI(Resource): method_decorators = [authenticate] def __init__(self): - #self.reqparse = reqparse.RequestParser() - #self.reqparse.add_argument('item_id', type = str, location = 'json') - #self.reqparse.add_argument('item_title', type = unicode, location = 'json') super(ArticleListAPI, self).__init__() def get(self): @@ -108,13 +106,6 @@ class ArticleListAPI(Resource): for article in articles] ) - def post(self): - """ - POST method - no sense here. - """ - response = jsonify({'code': 501, 'message': 'POST method not implemented for Article objects.'}) - response.status_code = 501 - return response def put(self): """ @@ -139,6 +130,12 @@ class ArticleAPI(Resource): method_decorators = [authenticate] def __init__(self): + self.reqparse = reqparse.RequestParser() + self.reqparse.add_argument('title', type = unicode, location = 'json') + self.reqparse.add_argument('content', type = unicode, location = 'json') + self.reqparse.add_argument('link', type = unicode, location = 'json') + self.reqparse.add_argument('date', type = str, location = 'json') + self.reqparse.add_argument('feed_id', type = int, location = 'json') super(ArticleAPI, self).__init__() def get(self, id=None): @@ -169,15 +166,37 @@ class ArticleAPI(Resource): for article in result] ) - def post(self): + def post(self, id): """ - POST method - no sense here. + POST method - Create a new article. """ - response = jsonify({'code': 501, 'message': 'POST method not implemented for Article objects.'}) - response.status_code = 501 - return response - - def put(self): + args = self.reqparse.parse_args() + article_dict = {} + for k, v in args.iteritems(): + if v != None: + article_dict[k] = v + else: + return {"message":"missing argument: %s" % (k,)} + article_date = None + try: + article_date = dateutil.parser.parse(article_dict["date"], dayfirst=True) + return {"message":"bad format for the date"} + except: + try: # trying to clean date field from letters + article_date = dateutil.parser.parse(re.sub('[A-z]', '', article_dict["date"], dayfirst=True)) + return {"message":"bad format for the date"} + except: + pass + article = Article(link=article_dict["link"], title=article_dict["title"], + content=article_dict["content"], readed=False, like=False, + date=article_date, user_id=g.user.id, + feed_id=article_dict["feed_id"]) + feed = Feed.query.filter(Feed.id == article_dict["feed_id"], Feed.user_id == g.user.id).first() + feed.articles.append(article) + db.session.commit() + return {"message":"ok"} + + def put(self, id): """ PUT method - no sense here. """ @@ -185,7 +204,7 @@ class ArticleAPI(Resource): response.status_code = 501 return response - def delete(self): + def delete(self, id): """ DELETE method - no sense here. """ |