add log

Author: surajit

logger/__init__.py

@@ -0,0 +1,40 @@
+import json
+import logging
+from functools import wraps
+from settings import BASE_DIR
+
+logger = logging.getLogger(__name__)
+
+LOG_FILE = BASE_DIR.joinpath('data/app.json')
+
+
+async def append_log(data):
+    with open(LOG_FILE, 'a') as file:
+        json.dump(data, file)
+        file.write('\n')
+
+
+def save_log(func):
+    @wraps(func)
+    async def wrapper(*args, **kwargs):
+        try:
+            request = kwargs.get("request")
+            url = request.url
+            method = request.method
+            message = dict(
+                url=str(url),
+                method=method,
+            )
+            if hasattr(request, "data"):
+                data = await request.data()
+                message["data"] = [data]
+            if hasattr(request, 'json'):
+                json_data = await request.json()
+                message["data"] = [json_data]
+            await append_log(message)
+        except Exception as exc:
+            logger.exception(f'Exception occurred {exc}')
+            pass
+        return await func(*args, **kwargs)
+
+    return wrapper

logger/log.py

@@ -8,6 +8,7 @@
 from email_api import email
 from email_api.email import UnauthorizedException
 from email_api.email import BadRequestException
+from logger.log import save_log
 from models.message import MessageType
 from models.message import Carrier
 from sqlalchemy.orm import Session
@@ -22,6 +23,7 @@
 
 
 @router.post("/email/send/")
+@save_log
 async def send_email(request: Request, db: Session = Depends(get_db)):
     try:
         data = await request.json()
@@ -36,3 +38,4 @@ async def send_email(request: Request, db: Session = Depends(get_db)):
     except (UnauthorizedException, BadRequestException, MessageExistException) as exc:
         raise HTTPException(status_code=200, detail=str(exc))
     return response
+

routes/v1/email.py

@@ -19,7 +19,9 @@ def __init__(self, headers, status_code):
 
 
 @mock.patch("routes.v1.email.email.send_email")
-def test_email_send(mock_send_email, client, db):
+@mock.patch("logger.log.append_log")
+def test_email_send(mock_save_log, mock_send_email, client, db):
+    mock_save_log.return_value = "Ok"
     sendgrid_response_mock = SendGridResponseMock(
         headers={"x-message-id": "xvis10203sn"}, status_code=202
     )
@@ -34,25 +36,31 @@ def test_email_send(mock_send_email, client, db):
 
 
 @mock.patch("routes.v1.email.email.send_email")
-def test_email_send_with_sendgrid_unauthorizedexception(mock_send_email, client):
+@mock.patch("logger.log.append_log")
+def test_email_send_with_sendgrid_unauthorizedexception(mock_save_log, mock_send_email, client):
+    mock_save_log.return_value = "Ok"
     mock_send_email.side_effect = UnauthorizedException("Not allowed to access the API")
     response = client.post("/api/v1/email/send/", json=read_json("email_data.json"))
     assert response.status_code == 200
     assert response.json() == {"detail": "Not allowed to access the API"}
 
 
 @mock.patch("routes.v1.email.email.send_email")
-def test_email_send_with_sendgrid_bad_request_exception(mock_send_email, client):
+@mock.patch("logger.log.append_log")
+def test_email_send_with_sendgrid_bad_request_exception(mock_save_log, mock_send_email, client):
+    mock_save_log.return_value = "Ok"
     mock_send_email.side_effect = BadRequestException("Bad request")
     response = client.post("/api/v1/email/send/", json=read_json("email_data.json"))
     assert response.status_code == 200
     assert response.json() == {"detail": "Bad request"}
 
 
 @mock.patch("routes.v1.email.email.send_email")
+@mock.patch("logger.log.append_log")
 def test_email_send_with_db_duplicate_message_id(
-    mock_send_email, client, message
+        mock_save_log, mock_send_email, client, message
 ):
+    mock_save_log.return_value = "Ok"
     sendgrid_response_mock = SendGridResponseMock(
         headers={"x-message-id": "123657ab"}, status_code=202
     )
@@ -61,6 +69,6 @@ def test_email_send_with_db_duplicate_message_id(
     assert response.status_code == 200
     assert response.json() == {
         "detail": "duplicate key value violates unique constraint "
-        '"ix_message_message_id"\nDETAIL:  Key (message_id)=('
-        "123657ab) already exists.\n"
+                  '"ix_message_message_id"\nDETAIL:  Key (message_id)=('
+                  "123657ab) already exists.\n"
     }