mmisp-lib 0.6.1__tar.gz → 0.6.2__tar.gz

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/PKG-INFO

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

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/pyproject.toml

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

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/src/mmisp/api_schemas/galaxy_clusters.py

@@ -5,6 +5,7 @@ from pydantic import BaseModel
 from mmisp.api_schemas.events import AddEditGetEventGalaxyClusterRelation, GetAllEventsGalaxyClusterGalaxy
 from mmisp.api_schemas.galaxy_common import GetAllSearchGalaxiesAttributes
 from mmisp.api_schemas.organisations import GetOrganisationResponse, Organisation
+from mmisp.lib.distribution import DistributionLevels
 
 
 class ExportGalaxyGalaxyElement(BaseModel):
@@ -27,13 +28,13 @@ class GetGalaxyClusterResponse(BaseModel):
     authors: list[str]
     version: str
     distribution: str
-    sharing_group_id: str
+    sharing_group_id: str | None
     org_id: str
     orgc_id: str
     default: bool
     locked: bool
     extends_uuid: str | None = None
-    extends_version: str
+    extends_version: str | None
     published: bool
     deleted: bool
     Galaxy: GetAllSearchGalaxiesAttributes | None = None
@@ -78,3 +79,21 @@ class ExportGalaxyClusterResponse(BaseModel):
     GalaxyClusterRelation: list[AddEditGetEventGalaxyClusterRelation] = []
     Org: Organisation
     Orgc: Organisation
+
+
+class AddGalaxyElement(BaseModel):
+    key: str
+    value: str
+
+
+class AddGalaxyClusterRequest(BaseModel):
+    value: str
+    description: str
+    source: str
+    authors: list[str]
+    distribution: DistributionLevels
+    GalaxyElement: list[AddGalaxyElement]
+
+
+class AddGalaxyClusterResponse(BaseModel):
+    pass

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/src/mmisp/db/models/galaxy_cluster.py

@@ -13,7 +13,7 @@ class GalaxyCluster(Base):
 
     id: Mapped[int] = mapped_column(Integer, primary_key=True, nullable=False)
     uuid: Mapped[str] = mapped_column(String(255), unique=True, default=uuid, index=True)
-    collection_uuid: Mapped[str] = mapped_column(String(255), nullable=False, index=True, default="0")
+    collection_uuid: Mapped[str] = mapped_column(String(255), nullable=False, index=True, default="")
     type: Mapped[str] = mapped_column(String(255), nullable=False, index=True)
     value: Mapped[str] = mapped_column(Text, nullable=False)
     tag_name: Mapped[str] = mapped_column(String(255), nullable=False, default="", index=True)
@@ -25,13 +25,13 @@ class GalaxyCluster(Base):
     authors: Mapped[str] = mapped_column(Text, nullable=False)
     version: Mapped[int] = mapped_column(Integer, default=0, index=True)
     distribution: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
-    sharing_group_id: Mapped[int] = mapped_column(Integer, index=True, default=0)
+    sharing_group_id: Mapped[int] = mapped_column(Integer, index=True, nullable=True, default=None)
     org_id: Mapped[int] = mapped_column(Integer, nullable=False, index=True, default=0)
     orgc_id: Mapped[int] = mapped_column(Integer, nullable=False, index=True, default=0)
     default: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, index=True)
     locked: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
     extends_uuid: Mapped[str] = mapped_column(String(40), index=True)
-    extends_version: Mapped[int] = mapped_column(Integer, default=0, index=True)
+    extends_version: Mapped[int] = mapped_column(Integer, index=True, nullable=True, default=None)
     published: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
     deleted: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
 

mmisp_lib-0.6.2/src/mmisp/lib/distribution.py

@@ -0,0 +1,8 @@
+from enum import IntEnum
+
+
+class DistributionLevels(IntEnum):
+    OWN_ORGANIZATION = 0
+    COMMUNITY = 1
+    CONNECTED_COMMUNITIES = 2
+    ALL_COMMUNITIES = 3

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/src/mmisp/lib/galaxies.py

@@ -15,3 +15,7 @@ def parse_galaxy_authors(str_authors: str) -> list[str]:
     assert isinstance(parsed_author, list)
 
     return parsed_author
+
+
+def galaxy_tag_name(galaxy_type: str, galaxy_cluster_uuid: str) -> str:
+    return f'misp-galaxy:{galaxy_type}="{galaxy_cluster_uuid}"'

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/src/mmisp/tests/fixtures.py

@@ -13,6 +13,7 @@ from mmisp.db.models.attribute import Attribute
 from mmisp.db.models.galaxy import Galaxy
 from mmisp.db.models.galaxy_cluster import GalaxyCluster, GalaxyElement
 from mmisp.db.models.tag import Tag
+from mmisp.lib.galaxies import galaxy_tag_name
 from mmisp.util.crypto import hash_secret
 from mmisp.util.uuid import uuid
 
@@ -460,10 +461,6 @@ async def auth_key(db, site_admin_user):
     await db.commit()
 
 
-def galaxy_tag_name_from_uuid(galaxy_cluster_uuid):
-    return f'misp-galaxy:test="{galaxy_cluster_uuid}"'
-
-
 @pytest.fixture
 def galaxy_cluster_collection_uuid():
     return uuid()
@@ -512,7 +509,7 @@ async def test_default_galaxy(db, galaxy_default_cluster_one_uuid, galaxy_defaul
         collection_uuid="",
         type="test galaxy type",
         value="test",
-        tag_name=galaxy_tag_name_from_uuid(galaxy_default_cluster_one_uuid),
+        tag_name=galaxy_tag_name("test galaxy type", galaxy_default_cluster_one_uuid),
         description="test",
         galaxy_id=galaxy.id,
         source="me",
@@ -534,7 +531,7 @@ async def test_default_galaxy(db, galaxy_default_cluster_one_uuid, galaxy_defaul
         collection_uuid="",
         type="test galaxy type",
         value="test",
-        tag_name=galaxy_tag_name_from_uuid(galaxy_default_cluster_two_uuid),
+        tag_name=galaxy_tag_name("test galaxy type", galaxy_default_cluster_two_uuid),
         description="test",
         galaxy_id=galaxy.id,
         source="me",
@@ -624,7 +621,7 @@ async def test_galaxy(db, instance_owner_org, galaxy_cluster_one_uuid, galaxy_cl
         collection_uuid="",
         type="test galaxy type",
         value="test",
-        tag_name=galaxy_tag_name_from_uuid(galaxy_cluster_one_uuid),
+        tag_name=galaxy_tag_name("test galaxy type", galaxy_cluster_one_uuid),
         description="test",
         galaxy_id=galaxy.id,
         source="me",
@@ -646,7 +643,7 @@ async def test_galaxy(db, instance_owner_org, galaxy_cluster_one_uuid, galaxy_cl
         collection_uuid="",
         type="test galaxy type",
         value="test",
-        tag_name=galaxy_tag_name_from_uuid(galaxy_cluster_two_uuid),
+        tag_name=galaxy_tag_name("test galaxy type", galaxy_cluster_two_uuid),
         description="test",
         galaxy_id=galaxy.id,
         source="me",
@@ -716,7 +713,7 @@ async def test_galaxy(db, instance_owner_org, galaxy_cluster_one_uuid, galaxy_cl
 @pytest_asyncio.fixture()
 async def galaxy_cluster_one_tag(db, galaxy_cluster_one_uuid):
     tag = Tag(
-        name=galaxy_tag_name_from_uuid(galaxy_cluster_one_uuid),
+        name=galaxy_tag_name("test galaxy type", galaxy_cluster_one_uuid),
         colour="#123456",
         exportable=True,
         hide_tag=False,

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/src/mmisp_lib.egg-info/PKG-INFO

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

{mmisp_lib-0.6.1 → mmisp_lib-0.6.2}/src/mmisp_lib.egg-info/SOURCES.txt

@@ -74,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/distribution.py
 src/mmisp/lib/fallbacks.py
 src/mmisp/lib/galaxies.py
 src/mmisp/lib/logging.py