Rename resource_fields in model. Rename resource classes.

This commit is contained in:
Alexis Lahouze 2016-01-13 11:44:17 +01:00
parent 6ee46c9136
commit cc6f7e2a36

View File

@ -32,7 +32,7 @@ from ..fields import Object
from .users import requires_auth from .users import requires_auth
resource_fields = { account_model = {
'id': fields.Integer(default=None), 'id': fields.Integer(default=None),
'name': fields.String, 'name': fields.String,
'authorized_overdraft': fields.Float, 'authorized_overdraft': fields.Float,
@ -58,7 +58,7 @@ date_parser.add_argument('end',
class AccountListResource(Resource): class AccountListResource(Resource):
@requires_auth @requires_auth
@marshal_with_field(fields.List(Object(resource_fields))) @marshal_with_field(fields.List(Object(account_model)))
def get(self): def get(self):
""" """
Returns accounts with their balances. Returns accounts with their balances.
@ -66,7 +66,7 @@ class AccountListResource(Resource):
return Account.query().all(), 200 return Account.query().all(), 200
@requires_auth @requires_auth
@marshal_with_field(Object(resource_fields)) @marshal_with_field(Object(account_model))
def post(self): def post(self):
""" """
Create a new account. Create a new account.
@ -88,7 +88,7 @@ class AccountListResource(Resource):
class AccountResource(Resource): class AccountResource(Resource):
@requires_auth @requires_auth
@marshal_with_field(Object(resource_fields)) @marshal_with_field(Object(account_model))
def get(self, id): def get(self, id):
""" """
Get account. Get account.
@ -105,7 +105,7 @@ class AccountResource(Resource):
return None, 404 return None, 404
@requires_auth @requires_auth
@marshal_with_field(Object(resource_fields)) @marshal_with_field(Object(account_model))
def post(self, id): def post(self, id):
data = parser.parse_args() data = parser.parse_args()
@ -130,7 +130,7 @@ class AccountResource(Resource):
).one(), 200 ).one(), 200
@requires_auth @requires_auth
@marshal_with_field(Object(resource_fields)) @marshal_with_field(Object(account_model))
def delete(self, id): def delete(self, id):
# Need to get the object to update it. # Need to get the object to update it.
account = db.session.query(Account).get(id) account = db.session.query(Account).get(id)
@ -153,23 +153,23 @@ range_parser.add_argument('begin', type=lambda a: dateutil.parser.parse(a))
range_parser.add_argument('end', type=lambda a: dateutil.parser.parse(a)) range_parser.add_argument('end', type=lambda a: dateutil.parser.parse(a))
category_resource_fields = { category_model = {
'category': fields.String, 'category': fields.String,
'expenses': fields.Float, 'expenses': fields.Float,
'revenues': fields.Float 'revenues': fields.Float
} }
class CategoriesResource(Resource): class CategoryResource(Resource):
@requires_auth @requires_auth
@marshal_with_field(fields.List(Object(category_resource_fields))) @marshal_with_field(fields.List(Object(category_model)))
def get(self): def get(self):
data = range_parser.parse_args() data = range_parser.parse_args()
return Operation.get_categories_for_range(**data).all(), 200 return Operation.get_categories_for_range(**data).all(), 200
ohlc_resource_fields = { ohlc_model = {
'operation_date': fields.DateTime(dt_format='iso8601'), 'operation_date': fields.DateTime(dt_format='iso8601'),
'open': fields.Float, 'open': fields.Float,
'high': fields.Float, 'high': fields.Float,
@ -178,14 +178,13 @@ ohlc_resource_fields = {
} }
class SoldsResource(Resource): class OHLCResource(Resource):
@requires_auth @requires_auth
@marshal_with_field(fields.List(Object(ohlc_resource_fields)))
def get(self): def get(self):
data = range_parser.parse_args() data = range_parser.parse_args()
return Operation.get_ohlc_per_day_for_range(**data).all(), 200 return Operation.get_ohlc_per_day_for_range(**data).all(), 200
api.add_resource(CategoriesResource, "/categories") api.add_resource(CategoryResource, "/categories")
api.add_resource(SoldsResource, "/solds") api.add_resource(OHLCResource, "/solds")