diff --git a/assemblyline/odm/models/service.py b/assemblyline/odm/models/service.py index 8076e21f3..42d266ab6 100644 --- a/assemblyline/odm/models/service.py +++ b/assemblyline/odm/models/service.py @@ -1,9 +1,13 @@ from __future__ import annotations + from typing import Optional as Opt from assemblyline import odm from assemblyline.common import forge -from assemblyline.common.constants import DEFAULT_SERVICE_ACCEPTS, DEFAULT_SERVICE_REJECTS +from assemblyline.common.constants import ( + DEFAULT_SERVICE_ACCEPTS, + DEFAULT_SERVICE_REJECTS, +) Classification = forge.get_classification() @@ -94,8 +98,8 @@ class UpdateSource(odm.Model): description="Update check interval, in seconds, for this source") ignore_cache: bool = odm.Boolean(default=False, description="Ignore source caching and forcefully fetch from source") - post_data: dict = odm.Mapping(odm.Any(), default={}, index=False, store=False, - description="Data that's sent in a POST request (`fetch_method=\"POST\"`)") + data: str = odm.Optional(odm.Text(), index=False, store=False, + description="Data that's sent in a POST request (`fetch_method=\"POST\"`)") diff --git a/assemblyline/odm/models/service_delta.py b/assemblyline/odm/models/service_delta.py index dfee952c7..707b0eb02 100644 --- a/assemblyline/odm/models/service_delta.py +++ b/assemblyline/odm/models/service_delta.py @@ -1,8 +1,7 @@ from __future__ import annotations from assemblyline import odm -from assemblyline.odm.models.service import SIGNATURE_DELIMITERS, FETCH_METHODS - +from assemblyline.odm.models.service import FETCH_METHODS, SIGNATURE_DELIMITERS REF_DEPENDENCY_CONFIG = "Refer to:
[Service - DependencyConfig](../service/#dependencyconfig)" REF_DOCKER_CONFIG = "Refer to:
[Service - DockerConfig](../service/#dockerconfig)" @@ -58,7 +57,7 @@ class UpdateSourceDelta(odm.Model): configuration = odm.Optional(odm.Mapping(odm.Any(), default={}), description=REF_UPDATE_SOURCE) update_interval = odm.Optional(odm.Integer(min=0), description=REF_UPDATE_SOURCE) ignore_cache = odm.Optional(odm.Boolean(default=False), description=REF_UPDATE_SOURCE) - post_data = odm.Optional(odm.Mapping(odm.Any(), default={}), description=REF_UPDATE_SOURCE) + data = odm.Optional(odm.Text(), description=REF_UPDATE_SOURCE) @ odm.model(index=False, store=False)