vysion 2.0.1__tar.gz → 2.0.2__tar.gz

{vysion-2.0.1 → vysion-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vysion
-Version: 2.0.1
+Version: 2.0.2
 Summary: The official Python client library for Vysion
 Home-page: https://vysion.ai
 License: Apache-2.0
@@ -16,7 +16,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: pydantic (>=2.0.1,<3.0.0)
-Requires-Dist: pymisp (==2.4.175)
+Requires-Dist: pymisp (==2.4.194)
 Requires-Dist: requests (>=2.28.1,<3.0.0)
 Requires-Dist: softenum (==1.0.1)
 Project-URL: Documentation, https://developers.vysion.ai
@@ -29,7 +29,7 @@ Welcome to the PyPi webpage for Vysion, our implementation as a Python library t
 
 You can request a demo for the web app or an API-key to use in this library at [vysion.ai](https://vysion.ai).
 
-Latest version: [2.0.1](https://pypi.org/project/vysion/)
+Latest version: [2.0.2](https://pypi.org/project/vysion/)
 
 You can visit [the documentation](https://developers.vysion.ai/?python) for more information on the searches and requests performed with the library or directly on the API.
 

{vysion-2.0.1 → vysion-2.0.2}/README.md

@@ -4,7 +4,7 @@ Welcome to the PyPi webpage for Vysion, our implementation as a Python library t
 
 You can request a demo for the web app or an API-key to use in this library at [vysion.ai](https://vysion.ai).
 
-Latest version: [2.0.1](https://pypi.org/project/vysion/)
+Latest version: [2.0.2](https://pypi.org/project/vysion/)
 
 You can visit [the documentation](https://developers.vysion.ai/?python) for more information on the searches and requests performed with the library or directly on the API.
 

{vysion-2.0.1 → vysion-2.0.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vysion"
-version = "2.0.1"
+version = "2.0.2"
 description = "The official Python client library for Vysion"
 homepage = "https://vysion.ai"
 repository = "https://gitlab.com/byronlabs/vysion/vysion-py"
@@ -21,9 +21,9 @@ classifiers=[
 python = "^3.8.0"
 
 pydantic = "^2.0.1"
-pymisp = "2.4.175"
 softenum = "1.0.1"
 requests = "^2.28.1"
+pymisp = "2.4.194"
 
 [tool.poetry.dev-dependencies]
 pytest = "7.1.2"

{vysion-2.0.1 → vysion-2.0.2}/vysion/client/client.py

@@ -221,6 +221,26 @@ class Client(BaseClient):
         result = self._make_request(url)
         return result.data
 
+    @vysion_error_manager
+    def find_phone(
+        self,
+        country_code: str,
+        phone_number: str,
+        page: int = 1,
+        lte: datetime = None,
+        gte: datetime = None,
+    ) -> dto.VysionResponse:
+        url = self._build_api_url__(
+            "document/phone",
+            country_code + "/" + phone_number,
+            page=page,
+            lte=lte,
+            gte=gte,
+        )
+
+        result = self._make_request(url)
+        return result.data
+
     @vysion_error_manager
     def find_wallet(
         self,
@@ -276,6 +296,13 @@ class Client(BaseClient):
         result = self._make_request(url)
         return result.data
 
+    @vysion_error_manager
+    def get_ransomware_victim(self, document_id: str) -> dto.VysionResponse:
+        url = self._build_api_url__("victim", document_id)
+
+        result = self._make_request(url)
+        return result.data
+
     #
     # Ransomware Stats
     #
@@ -283,10 +310,11 @@ class Client(BaseClient):
     @vysion_error_manager
     def ransomware_countries_stats(
         self,
+        countries: str = None,
         gte: datetime = None,
         lte: datetime = None,
     ) -> dto.VysionResponse[dto.Stat]:
-        url = self._build_api_url__("stats/countries", gte=gte, lte=lte)
+        url = self._build_api_url__("stats/countries", countries=countries, gte=gte, lte=lte)
 
         result = self._make_request(url)
         return result.data
@@ -294,21 +322,11 @@ class Client(BaseClient):
     @vysion_error_manager
     def ransomware_groups_stats(
         self,
+        countries: str = None,
         gte: datetime = None,
         lte: datetime = None,
     ) -> dto.VysionResponse[dto.Stat]:
-        url = self._build_api_url__("stats/groups", gte=gte, lte=lte)
-
-        result = self._make_request(url)
-        return result.data
-
-    @vysion_error_manager
-    def ransomware_attacks_stats(
-        self,
-        gte: datetime = None,
-        lte: datetime = None,
-    ) -> dto.VysionResponse[dto.Stat]:
-        url = self._build_api_url__("stats/attacks", gte=gte, lte=lte)
+        url = self._build_api_url__("stats/groups", countries=countries, gte=gte, lte=lte)
 
         result = self._make_request(url)
         return result.data
@@ -341,9 +359,9 @@ class Client(BaseClient):
 
     @vysion_error_manager
     def get_im_chat(
-        self, platform: str, channelId: str
+        self, platform: str, channelId: str, gte: datetime = None, lte: datetime = None
     ) -> dto.VysionResponse[dto.ImMessageHit]:
-        url = self._build_api_url__("im/" + platform + "/chat/", channelId)
+        url = self._build_api_url__("im/" + platform + "/chat/" + channelId, gte=gte, lte=lte)
 
         result = self._make_request(url)
         return result.data

{vysion-2.0.1 → vysion-2.0.2}/vysion/dto/dto.py

@@ -399,8 +399,9 @@ T = TypeVar("T")
 
 class Result(BaseModel, Generic[T]):
     total: int = 0
-    hits: List[T] = Field(default_factory=lambda: [])
-
+    hits: Union[List[DocumentHit],List[Stat],List[ImChannelHit],List[ImMessageHit],
+                List[ImProfileHit]] = Field(default_factory=lambda: [])
+    
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         if self.total <= 0:

{vysion-2.0.1 → vysion-2.0.2}/vysion/dto/util.py

@@ -33,7 +33,6 @@ class MISPProcessor:
     def parse_hit(self, hit: DocumentHit, ref_attribute: MISPAttribute = None, **_):
         page: Page = hit.page
 
-        # TODO Add more page parameters
         misp_object = MISPObject("vysion-page")
         misp_object.template_uuid = "4d0b66f1-5268-47e0-9d29-f2e4f3db8e7f"
 
@@ -47,7 +46,7 @@ class MISPProcessor:
         misp_object.add_attribute("network", type="text", value=network)
 
         pageTitle = page.pageTitle
-        misp_object.add_attribute("title", type="text", value=pageTitle)
+        misp_object.add_attribute("pageTitle", type="text", value=pageTitle)
 
         url_vysion: URL = "https://app.vysion.ai/document/" + page.id
         misp_object.add_attribute("url_vysion", type="url", value=url_vysion)
@@ -59,7 +58,7 @@ class MISPProcessor:
 
         vysion_reference_id = misp_object.uuid
 
-        # TODO Remove this addition when the vysion-page object works
+        # TODO Add more page parameters into misp objects
         self.misp_event.add_attribute("url", value=url.build())
 
         self.misp_event.add_attribute("domain", value=url.domainName)
@@ -94,9 +93,8 @@ class MISPProcessor:
         misp_object = MISPObject("vysion-ransomware-feed")
         misp_object.template_uuid = "e0bfa994-c184-4894-bfaa-73b1350746e1"
         misp_object["meta-category"] = (
-            "misc"  # TODO Esto se tiene que poder hacer de otra manera... Y sólo es necesario en los feeds
+            "misc"  
         )
-
         misp_object.add_attribute("id", type="text", value=hit.id)
         misp_object.add_attribute("companyName", type="target-org", value=hit.companyName)
         misp_object.add_attribute("companyLink", type="link", value=hit.companyLink)
@@ -123,4 +121,4 @@ class MISPProcessor:
         for hit in result.hits:
             processor(hit, **kwargs)
 
-        return self.misp_event
+        return self.misp_event

{vysion-2.0.1 → vysion-2.0.2}/vysion/version.py

@@ -15,4 +15,4 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-__version__ = "2.0.1"
+__version__ = "2.0.2"