add option to use custom fields in post request

Author: markbeep

README.md

@@ -84,6 +84,7 @@ Notifications depend on [Apprise](https://github.com/caronc/apprise) or [Gotify]
 2. Add the Apprise URL or the path to your gotify instance with `/message` appended, i.e.: `http://gotify:8080/message`.
 3. For gofity, add the API key as a header: `{"Authentication": "Bearer <your token>"}`.
 4. Configure the remaining settings. **The event variables are case sensitive**.
+5. `Additional POST fields` allow you to add extra values that are sent along in the POST request to the notification service. This allows for changing the priority for gotify notifications or changing the look of apprise notifications. **Event variables also work in keys and values!**
 
 ### OpenID Connect
 

alembic/versions/304eed96f8ed_add_additional_fields.py

@@ -0,0 +1,39 @@
+"""add additional fields
+
+Revision ID: 304eed96f8ed
+Revises: 0fa71b2e5d30
+Create Date: 2025-04-24 15:29:26.686894
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = "304eed96f8ed"
+down_revision: Union[str, None] = "0fa71b2e5d30"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("notification", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "additional_fields", sa.JSON(), nullable=True, server_default="{}"
+            )
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("notification", schema=None) as batch_op:
+        batch_op.drop_column("additional_fields")
+
+    # ### end Alembic commands ###

app/internal/models.py

@@ -8,6 +8,8 @@
 import pydantic
 from sqlmodel import JSON, Column, DateTime, Field, SQLModel, UniqueConstraint, func
 
+from app.util import json_type
+
 
 class BaseModel(SQLModel):
     pass
@@ -201,13 +203,17 @@ class EventEnum(str, Enum):
 class NotificationServiceEnum(str, Enum):
     apprise = "apprise"
     gotify = "gotify"
+    custom = "custom"
 
 
 class Notification(BaseModel, table=True):
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
     name: str
     url: str
     headers: dict[str, str] = Field(default_factory=dict, sa_column=Column(JSON))
+    additional_fields: dict[str, json_type.JSON] = Field(
+        default_factory=dict, sa_column=Column(JSON)
+    )
     event: EventEnum
     service: NotificationServiceEnum
     title_template: str
@@ -217,3 +223,7 @@ class Notification(BaseModel, table=True):
     @property
     def serialized_headers(self):
         return json.dumps(self.headers)
+
+    @property
+    def serialized_additional_fields(self):
+        return json.dumps(self.additional_fields)

app/internal/notifications.py

@@ -1,3 +1,4 @@
+import json
 import logging
 from typing import Optional
 
@@ -11,50 +12,42 @@
     Notification,
     NotificationServiceEnum,
 )
+from app.util import json_type
 from app.util.db import open_session
 
 logger = logging.getLogger(__name__)
 
 
 def replace_variables(
-    title_template: str,
-    body_template: str,
+    template: str,
     username: Optional[str] = None,
     book_title: Optional[str] = None,
     book_authors: Optional[str] = None,
     book_narrators: Optional[str] = None,
     event_type: Optional[str] = None,
     other_replacements: dict[str, str] = {},
 ):
-    title = title_template
-    body = body_template
-
     if username:
-        title = title.replace("{eventUser}", username)
-        body = body.replace("{eventUser}", username)
+        template = template.replace("{eventUser}", username)
     if book_title:
-        title = title.replace("{bookTitle}", book_title)
-        body = body.replace("{bookTitle}", book_title)
+        template = template.replace("{bookTitle}", book_title)
     if book_authors:
-        title = title.replace("{bookAuthors}", book_authors)
-        body = body.replace("{bookAuthors}", book_authors)
+        template = template.replace("{bookAuthors}", book_authors)
     if book_narrators:
-        title = title.replace("{bookNarrators}", book_narrators)
-        body = body.replace("{bookNarrators}", book_narrators)
+        template = template.replace("{bookNarrators}", book_narrators)
     if event_type:
-        title = title.replace("{eventType}", event_type)
-        body = body.replace("{eventType}", event_type)
+        template = template.replace("{eventType}", event_type)
 
     for key, value in other_replacements.items():
-        title = title.replace(f"{{{key}}}", value)
-        body = body.replace(f"{{{key}}}", value)
+        template = template.replace(f"{{{key}}}", value)
 
-    return title, body
+    return template
 
 
 async def _send(
     title: str,
     body: str,
+    additional_fields: dict[str, json_type.JSON],
     notification: Notification,
     client_session: ClientSession,
 ):
@@ -63,13 +56,29 @@ async def _send(
             body_key = "message"
         case NotificationServiceEnum.apprise:
             body_key = "body"
+        case NotificationServiceEnum.custom:
+            body_key = ""
 
-    async with client_session.post(
-        notification.url,
-        json={
+    if notification.service == NotificationServiceEnum.custom:
+        json_body = {}
+    else:
+        json_body: dict[str, json_type.JSON] = {
             "title": title,
             body_key: body,
-        },
+        }
+
+    for key, value in additional_fields.items():
+        if key in json_body.keys():
+            logger.warning(
+                f"Key '{key}' already exists in the JSON body but is passed as additional field. Overwriting with value: {value}"
+            )
+        json_body[key] = value
+
+    print(json_body)
+
+    async with client_session.post(
+        notification.url,
+        json=json_body,
         headers=notification.headers,
     ) as response:
         response.raise_for_status()
@@ -95,8 +104,16 @@ async def send_notification(
             book_authors = ",".join(book.authors)
             book_narrators = ",".join(book.narrators)
 
-    title, body = replace_variables(
+    title = replace_variables(
         notification.title_template,
+        requester_username,
+        book_title,
+        book_authors,
+        book_narrators,
+        notification.event.value,
+        other_replacements,
+    )
+    body = replace_variables(
         notification.body_template,
         requester_username,
         book_title,
@@ -105,13 +122,24 @@ async def send_notification(
         notification.event.value,
         other_replacements,
     )
+    additional_fields: dict[str, json_type.JSON] = json.loads(
+        replace_variables(
+            json.dumps(notification.additional_fields),
+            requester_username,
+            book_title,
+            book_authors,
+            book_narrators,
+            notification.event.value,
+            other_replacements,
+        )
+    )
 
     logger.info(
         f"Sending notification to {notification.url} with title: '{title}', event type: {notification.event.value}"
     )
 
     async with ClientSession() as client_session:
-        return await _send(title, body, notification, client_session)
+        return await _send(title, body, additional_fields, notification, client_session)
 
 
 async def send_all_notifications(
@@ -144,23 +172,45 @@ async def send_manual_notification(
 ):
     """Send a notification for manual book requests"""
     try:
-        title, body = replace_variables(
+        book_authors = ",".join(book.authors)
+        book_narrators = ",".join(book.narrators)
+
+        title = replace_variables(
             notification.title_template,
+            requester_username,
+            book.title,
+            book_authors,
+            book_narrators,
+            notification.event.value,
+            other_replacements,
+        )
+        body = replace_variables(
             notification.body_template,
             requester_username,
             book.title,
-            ",".join(book.authors),
-            ",".join(book.narrators),
+            book_authors,
+            book_narrators,
             notification.event.value,
             other_replacements,
         )
+        additional_fields: dict[str, json_type.JSON] = json.loads(
+            replace_variables(
+                json.dumps(notification.additional_fields),
+                requester_username,
+                book.title,
+                book_authors,
+                book_narrators,
+                notification.event.value,
+                other_replacements,
+            )
+        )
 
         logger.info(
             f"Sending manual notification to {notification.url} with title: '{title}', event type: {notification.event.value}"
         )
 
         async with ClientSession() as client_session:
-            await _send(title, body, notification, client_session)
+            await _send(title, body, additional_fields, notification, client_session)
 
     except Exception as e:
         logger.error("Failed to send notification", e)