Convert fields in API response to camelCase

Author: recrsn

Pipfile

@@ -4,7 +4,7 @@ url = "https://pypi.org/simple"
 verify_ssl = true
 
 [dev-packages]
-
+pytest = "*"
 
 [packages]
 flask = "*"

Pipfile.lock

@@ -1,6 +1,5 @@
 import json
 
-from dotenv import load_dotenv
 from flask import Flask, url_for
 
 from . import filters, settings, logger
@@ -11,7 +10,6 @@
 from .services import login_manager
 from .services.session_interface import CustomSessionInterface
 
-load_dotenv()
 logger.setup()
 
 app = Flask(__name__)

snowflake/app.py

@@ -0,0 +1,20 @@
+from ..marshmallow import marshmallow
+
+
+def camelcase(s):
+    parts = iter(s.split("_"))
+    return next(parts) + "".join(i.title() for i in parts)
+
+
+class BaseSchema(marshmallow.Schema):
+    # noinspection PyMethodMayBeStatic
+    def on_bind_field(self, field_name, field_obj):
+        field_obj.data_key = camelcase(field_obj.data_key or field_name)
+
+
+class BaseSQLAlchemySchema(marshmallow.SQLAlchemySchema, BaseSchema):
+    pass
+
+
+class BaseSQLAlchemyAutoSchema(marshmallow.SQLAlchemyAutoSchema, BaseSchema):
+    pass

snowflake/schemas/base.py

@@ -1,14 +1,14 @@
 from marshmallow.fields import String, DateTime, Nested
 
+from .base import BaseSchema
 from .user import UserSchema
-from ..marshmallow import marshmallow
 
 
-class LoginSchema(marshmallow.Schema):
+class LoginSchema(BaseSchema):
     token = String()
 
 
-class LoginResponseSchema(marshmallow.Schema):
+class LoginResponseSchema(BaseSchema):
     token = String()
     expiry = DateTime()
     refresh_token = String()

snowflake/schemas/login.py

@@ -1,12 +1,7 @@
-from marshmallow_sqlalchemy.fields import Nested
-
-from .user import UserSchema
-from ..marshmallow import marshmallow
+from .base import BaseSQLAlchemyAutoSchema
 from ..models import Notification
 
 
-class NotificationSchema(marshmallow.SQLAlchemyAutoSchema):
+class NotificationSchema(BaseSQLAlchemyAutoSchema):
     class Meta:
         model = Notification
-
-    user = Nested(UserSchema)

snowflake/schemas/notification.py

@@ -1,13 +1,14 @@
 from marshmallow.fields import List
 from marshmallow_sqlalchemy.fields import Nested
 
+from .base import BaseSQLAlchemySchema
 from .fields import UserByUsername
 from .user import UserSchema
 from ..marshmallow import marshmallow
 from ..models import OneOnOne, OneOnOneActionItem
 
 
-class OneOnOneActionItemSchema(marshmallow.SQLAlchemySchema):
+class OneOnOneActionItemSchema(BaseSQLAlchemySchema):
     class Meta:
         model = OneOnOneActionItem
 
@@ -18,7 +19,7 @@ class Meta:
     created_by = Nested(UserSchema)
 
 
-class CreateOrEditOneOnOneActionItemSchema(marshmallow.SQLAlchemySchema):
+class CreateOrEditOneOnOneActionItemSchema(BaseSQLAlchemySchema):
     class Meta:
         model = OneOnOneActionItem
         load_instance = True
@@ -27,7 +28,7 @@ class Meta:
     content = marshmallow.auto_field()
 
 
-class OneOnOneSchema(marshmallow.SQLAlchemySchema):
+class OneOnOneSchema(BaseSQLAlchemySchema):
     class Meta:
         model = OneOnOne
 
@@ -41,7 +42,7 @@ class GetOneOnOneSchema(OneOnOneSchema):
     action_items = List(Nested(OneOnOneActionItemSchema))
 
 
-class CreateOneOnOneSchema(marshmallow.SQLAlchemySchema):
+class CreateOneOnOneSchema(BaseSQLAlchemySchema):
     class Meta:
         model = OneOnOne
         load_instance = True

snowflake/schemas/one_on_one.py

@@ -1,8 +1,8 @@
-from ..marshmallow import marshmallow
+from .base import BaseSQLAlchemyAutoSchema
 from ..models import User
 
 
-class UserSchema(marshmallow.SQLAlchemyAutoSchema):
+class UserSchema(BaseSQLAlchemyAutoSchema):
     class Meta:
         model = User
         exclude = ("id",)

snowflake/schemas/user.py

@@ -1,8 +1,11 @@
 import os
 
 import requests
+from dotenv import load_dotenv
 from flask import Flask
 
+load_dotenv()
+
 GOOGLE_DISCOVERY_URL = "https://accounts.google.com/.well-known/openid-configuration"
 REDIS_URL = os.getenv('REDIS_URL')
 TOKEN_VALIDITY_SECS = int(os.getenv('TOKEN_VALIDITY_SECS'))