mmisp-lib 0.5.0__tar.gz → 0.6.0__tar.gz

{mmisp_lib-0.5.0 → mmisp_lib-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mmisp-lib
-Version: 0.5.0
+Version: 0.6.0
 Requires-Python: >=3.11.0
 Description-Content-Type: text/markdown
 License-File: LICENSE

{mmisp_lib-0.5.0 → mmisp_lib-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mmisp-lib"
-version = "0.5.0"
+version = "0.6.0"
 description = ""
 authors = []
 readme = "README.md"

{mmisp_lib-0.5.0 → mmisp_lib-0.6.0}/src/mmisp/api_schemas/galaxies.py

@@ -2,8 +2,8 @@ from typing import List
 
 from pydantic import BaseModel
 
-import mmisp.api_schemas.organisations
-from mmisp.api_schemas.events import AddEditGetEventGalaxyClusterRelation, GetAllEventsGalaxyClusterGalaxy
+from mmisp.api_schemas.galaxy_clusters import GetGalaxyClusterResponse
+from mmisp.api_schemas.galaxy_common import GetAllSearchGalaxiesAttributes
 
 
 class SearchGalaxiesBody(BaseModel):
@@ -38,31 +38,6 @@ class ExportGalaxyGalaxyElement(BaseModel):
     value: str
 
 
-class GetGalaxyClusterResponse(BaseModel):
-    id: str | None = None
-    uuid: str | None = None
-    collection_uuid: str
-    type: str
-    value: str
-    tag_name: str
-    description: str
-    galaxy_id: str
-    source: str
-    authors: list[str]
-    version: str
-    distribution: str
-    sharing_group_id: str
-    org_id: str
-    orgc_id: str
-    default: bool
-    locked: bool
-    extends_uuid: str
-    extends_version: str
-    published: bool
-    deleted: bool
-    GalaxyElement: list[ExportGalaxyGalaxyElement]
-
-
 class ImportGalaxyBody(BaseModel):
     GalaxyCluster: GetGalaxyClusterResponse
     Galaxy: ImportGalaxyGalaxy
@@ -71,20 +46,6 @@ class ImportGalaxyBody(BaseModel):
         orm_mode = True
 
 
-class GetAllSearchGalaxiesAttributes(BaseModel):
-    id: str
-    uuid: str
-    name: str
-    type: str
-    description: str
-    version: str
-    icon: str
-    namespace: str
-    kill_chain_order: str | None = None
-    enabled: bool
-    local_only: bool
-
-
 class GetAllSearchGalaxiesResponse(BaseModel):
     Galaxy: GetAllSearchGalaxiesAttributes
 
@@ -115,38 +76,6 @@ class GalaxySchema(BaseModel):
         orm_mode = True
 
 
-class ExportGalaxyClusterResponse(BaseModel):
-    id: str
-    uuid: str
-    collection_uuid: str
-    type: str
-    value: str
-    tag_name: str
-    description: str
-    galaxy_id: str
-    source: str
-    authors: list[str]
-    version: str
-    distribution: str
-    sharing_group_id: str
-    org_id: str
-    orgc_id: str
-    default: bool
-    locked: bool
-    extends_uuid: str
-    extends_version: str
-    published: bool
-    deleted: bool
-    GalaxyElement: list[ExportGalaxyGalaxyElement]
-    Galaxy: GetAllEventsGalaxyClusterGalaxy
-    GalaxyClusterRelation: list[AddEditGetEventGalaxyClusterRelation] = []
-    Org: mmisp.api_schemas.organisations.Organisation
-    Orgc: mmisp.api_schemas.organisations.Organisation
-
-    class Config:
-        orm_mode = True
-
-
 class ExportGalaxyAttributes(BaseModel):
     default: bool
     custom: bool | None = None

{mmisp_lib-0.5.0 → mmisp_lib-0.6.0}/src/mmisp/api_schemas/galaxy_clusters.py

@@ -1,10 +1,17 @@
+from datetime import datetime
+
 from pydantic import BaseModel
 
-from mmisp.api_schemas.events.add_edit_get_event_response import AddEditGetEventGalaxyClusterRelation
-from mmisp.api_schemas.events.get_all_events_response import GetAllEventsGalaxyClusterGalaxy
-from mmisp.api_schemas.galaxies import ExportGalaxyGalaxyElement, GetAllSearchGalaxiesAttributes
-from mmisp.api_schemas.organisations import GetOrganisationResponse
-from mmisp.api_schemas.organisations.organisation import Organisation
+from mmisp.api_schemas.events import AddEditGetEventGalaxyClusterRelation, GetAllEventsGalaxyClusterGalaxy
+from mmisp.api_schemas.galaxy_common import GetAllSearchGalaxiesAttributes
+from mmisp.api_schemas.organisations import GetOrganisationResponse, Organisation
+
+
+class ExportGalaxyGalaxyElement(BaseModel):
+    id: str | None = None
+    galaxy_cluster_id: str | None = None
+    key: str
+    value: str
 
 
 class GetGalaxyClusterResponse(BaseModel):
@@ -25,14 +32,23 @@ class GetGalaxyClusterResponse(BaseModel):
     orgc_id: str
     default: bool
     locked: bool
-    extends_uuid: str
+    extends_uuid: str | None = None
     extends_version: str
     published: bool
     deleted: bool
-    Galaxy: GetAllSearchGalaxiesAttributes
+    Galaxy: GetAllSearchGalaxiesAttributes | None = None
     GalaxyElement: list[ExportGalaxyGalaxyElement]
-    Org: GetOrganisationResponse
-    Orgc: GetOrganisationResponse
+    GalaxyClusterRelation: list = []
+    RelationshipInbound: list = []
+    Org: GetOrganisationResponse | None = None
+    Orgc: GetOrganisationResponse | None = None
+
+
+class GalaxyClusterResponse(BaseModel):
+    GalaxyCluster: GetGalaxyClusterResponse
+
+    class Config:
+        json_encoders = {datetime: lambda v: v.strftime("%Y-%m-%d %H:%M:%S")}
 
 
 class ExportGalaxyClusterResponse(BaseModel):

mmisp_lib-0.6.0/src/mmisp/api_schemas/galaxy_common.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel
+
+
+class GetAllSearchGalaxiesAttributes(BaseModel):
+    id: str
+    uuid: str
+    name: str
+    type: str
+    description: str
+    version: str
+    icon: str
+    namespace: str
+    kill_chain_order: str | None = None
+    enabled: bool
+    local_only: bool

mmisp_lib-0.6.0/src/mmisp/lib/galaxies.py

@@ -0,0 +1,17 @@
+import json
+
+
+def json_or_original_string(s: str) -> str | dict | list:
+    try:
+        return json.loads(s)
+    except json.decoder.JSONDecodeError:
+        return s
+
+
+def parse_galaxy_authors(str_authors: str) -> list[str]:
+    parsed_author = json_or_original_string(str_authors)
+    if isinstance(parsed_author, str):
+        parsed_author = [parsed_author]  # force to be a list
+    assert isinstance(parsed_author, list)
+
+    return parsed_author

{mmisp_lib-0.5.0 → mmisp_lib-0.6.0}/src/mmisp_lib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mmisp-lib
-Version: 0.5.0
+Version: 0.6.0
 Requires-Python: >=3.11.0
 Description-Content-Type: text/markdown
 License-File: LICENSE

{mmisp_lib-0.5.0 → mmisp_lib-0.6.0}/src/mmisp_lib.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ src/mmisp/api_schemas/events.py
 src/mmisp/api_schemas/feeds.py
 src/mmisp/api_schemas/galaxies.py
 src/mmisp/api_schemas/galaxy_clusters.py
+src/mmisp/api_schemas/galaxy_common.py
 src/mmisp/api_schemas/jobs.py
 src/mmisp/api_schemas/logs.py
 src/mmisp/api_schemas/noticelists.py
@@ -73,6 +74,7 @@ src/mmisp/lib/__init__.py
 src/mmisp/lib/actions.py
 src/mmisp/lib/attribute_search_filter.py
 src/mmisp/lib/attributes.py
+src/mmisp/lib/galaxies.py
 src/mmisp/lib/logging.py
 src/mmisp/lib/permissions.py
 src/mmisp/lib/py.typed