vysion 2.1.0__py3-none-any.whl → 2.1.2__py3-none-any.whl

vysion/client/client.py

@@ -27,6 +27,7 @@ import requests
 from vysion.client.error import APIError
 from vysion.dto import (
     Error,
+    ErrorCode,
     DocumentHit,
     VysionResponse,
     RansomwareHit,
@@ -148,7 +149,7 @@ def vysion_error_manager(method) -> Union[VysionResponse, Error]:
             return Error(code=e.code, message=e.message)
         except Exception as e:
             LOGGER.error(e)
-            return Error()
+            return Error(code=ErrorCode.INTERNAL_SERVER_ERROR, message=str(e))
 
     return manage
 
@@ -330,6 +331,7 @@ class Client(BaseClient):
         country: str = None,
         language: Language = None,
         sector: str = None,
+        group: str = None,
     ) -> VysionResponse[RansomwareHit]:
         url = self._build_api_url__(
             "victim/search",
@@ -342,6 +344,7 @@ class Client(BaseClient):
             country=country,
             language=language,
             sector=sector,
+            group=group,
         )
 
         result = VysionResponse[RansomwareHit].model_validate(self._make_request(url))
@@ -412,12 +415,9 @@ class Client(BaseClient):
             username=username,
         )
 
-        print(url)
-
         result = VysionResponse[ImMessageHit].model_validate(self._make_request(url))
         return result.data
 
-
     @vysion_error_manager
     def im_find_email(
         self,

vysion/client/error.py

@@ -14,7 +14,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 """
-from vysion.dto import Error
+from vysion.dto import ErrorCode
 
 
 class APIError(Exception):
@@ -28,8 +28,8 @@ class APIError(Exception):
 
         super().__init__()
 
-        if code not in [i.value for i in Error.StatusCode]:
-            self.code = Error.StatusCode.UNK
+        if code not in [i.value for i in ErrorCode]:
+            self.code = ErrorCode.INTERNAL_SERVER_ERROR
             self.message = f"{message} (Original code: {code})"
         else:
             self.code = code

vysion/dto/dto.py

@@ -487,13 +487,23 @@ class ImFeedHit(BaseModel):
     network: str
 
 
+class OnionFeedHit(BaseModel):
+    id: str
+    url: URL
+    path: str
+    detectionDate: datetime
+    title: Optional[str] = Field(default_factory=lambda: None)
+    tag: List[Tag] = Field(default_factory=lambda: [])
+    network: Network = Field(default_factory=lambda: Network.tor)
+    
+
 class CryptoFeedHit(BaseModel):
     id: str
     url: str
     detectionDate: datetime
     url: str
     network: str
-    title: str
+    title: Optional[str] = Field(default_factory=lambda: None)
     tag: List[Tag] = Field(default_factory=lambda: [])
     bitcoin_address: Optional[List[BitcoinAddress]] = Field(default_factory=lambda: [])
     polkadot_address: Optional[List[PolkadotAddress]] = Field(default_factory=lambda: [])
@@ -528,6 +538,85 @@ class CryptoFeedHit(BaseModel):
         return values
 
 
+class CryptoTelegramFeedHit(BaseModel):
+    userId: str
+    usernames: Optional[List[str]] = Field(default_factory=lambda: [])
+    firstName: Optional[str] = Field(default_factory=lambda: None)
+    lastName: Optional[str] = Field(default_factory=lambda: None)
+    detectionDate: datetime
+    profilePhoto: Optional[List[str]] = Field(default_factory=lambda: None)
+    bitcoin_address: Optional[List[BitcoinAddress]] = Field(default_factory=lambda: [])
+    polkadot_address: Optional[List[PolkadotAddress]] = Field(default_factory=lambda: [])
+    ethereum_address: Optional[List[EthereumAddress]] = Field(default_factory=lambda: [])
+    monero_address: Optional[List[MoneroAddress]] = Field(default_factory=lambda: [])
+    ripple_address: Optional[List[RippleAddress]] = Field(default_factory=lambda: [])
+    zcash_address: Optional[List[ZcashAddress]] = Field(default_factory=lambda: [])
+
+
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True, 
+        exclude_defaults=True, 
+        validate_assignment=True
+    )
+
+    @model_validator(mode="after")
+    def remove_empty_lists(cls, values):
+        """Remove empty cryptocurrency lists from output"""
+        crypto_fields = [
+            "bitcoin_address", 
+            "polkadot_address", 
+            "ethereum_address", 
+            "monero_address", 
+            "ripple_address", 
+            "zcash_address"
+        ]
+        
+        for field_name in crypto_fields:
+            field_value = getattr(values, field_name, None)
+            if isinstance(field_value, list) and not field_value:
+                setattr(values, field_name, None)
+        
+        return values
+    
+class CryptoDiscordFeedHit(BaseModel):
+    userId: str
+    usernames: Optional[List[str]] = Field(default_factory=lambda: [])
+    creationDate: Optional[datetime] = Field(default_factory=lambda: None)
+    detectionDate: datetime
+    profilePhoto: Optional[List[str]] = Field(default_factory=lambda: None)
+    bitcoin_address: Optional[List[BitcoinAddress]] = Field(default_factory=lambda: [])
+    polkadot_address: Optional[List[PolkadotAddress]] = Field(default_factory=lambda: [])
+    ethereum_address: Optional[List[EthereumAddress]] = Field(default_factory=lambda: [])
+    monero_address: Optional[List[MoneroAddress]] = Field(default_factory=lambda: [])
+    ripple_address: Optional[List[RippleAddress]] = Field(default_factory=lambda: [])
+    zcash_address: Optional[List[ZcashAddress]] = Field(default_factory=lambda: [])
+
+
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True, 
+        exclude_defaults=True, 
+        validate_assignment=True
+    )
+
+    @model_validator(mode="after")
+    def remove_empty_lists(cls, values):
+        """Remove empty cryptocurrency lists from output"""
+        crypto_fields = [
+            "bitcoin_address", 
+            "polkadot_address", 
+            "ethereum_address", 
+            "monero_address", 
+            "ripple_address", 
+            "zcash_address"
+        ]
+        
+        for field_name in crypto_fields:
+            field_value = getattr(values, field_name, None)
+            if isinstance(field_value, list) and not field_value:
+                setattr(values, field_name, None)
+        
+        return values
+
 class RansomFeedHit(BaseModel):
     id: str
     companyName: Optional[str]

vysion/version.py

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

{vysion-2.1.0.dist-info → vysion-2.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vysion
-Version: 2.1.0
+Version: 2.1.2
 Summary: The official Python client library for Vysion
 Home-page: https://vysion.ai
 License: Apache-2.0
@@ -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.1.0](https://pypi.org/project/vysion/)
+Latest version: [2.1.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.1.0.dist-info → vysion-2.1.2.dist-info}/RECORD

@@ -1,9 +1,9 @@
 vysion/__init__.py,sha256=yxJiM-S29q8GB3PBff9tV70P2nvydd8aKLEprLkXw5o,664
 vysion/client/__init__.py,sha256=aVHmBuetPdybp7TgcNzUx4HkxTjEuXYzSdDYiYfFQd0,630
-vysion/client/client.py,sha256=-AUOWTTRU-yJXSPj1c50aI26YFXgix3dDx06iVA2UcQ,15014
-vysion/client/error.py,sha256=-3NGr9ngTGSy-ctqk6S9eMBAvtY3lnjv9i0KiuRjd5g,1158
+vysion/client/client.py,sha256=LShr3t81ACAduxXgdx3_E-qUQTgRcPw7nkGkfulGq00,15112
+vysion/client/error.py,sha256=JK0dsZkyMjmxPtEgO5IStG_N_q8y_-VGbppwWrUtSgM,1166
 vysion/dto/__init__.py,sha256=ct8JxVMfJ0APiOTgr9ju-JIuBlXOrPkx7n2qISSXUts,605
-vysion/dto/dto.py,sha256=etlcurb0a_gSIGLrdZw3yrSAKv2W8phWl2iaXbuDx18,21007
+vysion/dto/dto.py,sha256=ERPY2zbNXgc40SPfmy1gp9-6WaD3x8AeFBzmTFDnMtM,24588
 vysion/dto/tag.py,sha256=whGPsdasgYm8oijrL7MRSMmapzaoCmKNeaJ5ih0aBIk,1503
 vysion/dto/util.py,sha256=AxdMWPdf7xrFtWva5epKxgjI0Bah1GiIw-sEWvIUYb4,4491
 vysion/model/__init__.py,sha256=rJX9eNW9-jQGTgwm97W04jhZa-dV_dSTTTFcbWg73w0,606
@@ -15,8 +15,8 @@ vysion/model/enum/services.py,sha256=fUfZEO0RNNctU4Gzae5TV3c7BU5A_cEwaEb1tzqgcr0
 vysion/taxonomy/__init__.py,sha256=Bc364AYxRCyjIn26-XBPeEDCPe3Hma4r5RAI8R8eblU,635
 vysion/taxonomy/flavours.py,sha256=ubImHBE8v0zhzFy_2YozAQk2SzN1XDQwHAxb2RebtCo,2347
 vysion/taxonomy/taxonomy.py,sha256=mG6vHX5Eyp38AKyJBgfBDqdpEmDMbkv8TA8YUBzko54,12227
-vysion/version.py,sha256=8cOVMSH8g0Y_tOihg-6IyzRRFDqDfZDgTFNEqE0ueOg,610
-vysion-2.1.0.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-vysion-2.1.0.dist-info/METADATA,sha256=lVVsvr8EC9T3BUeQyIsKV9rDsMJMrHNDLcnZa5hYugk,1958
-vysion-2.1.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-vysion-2.1.0.dist-info/RECORD,,
+vysion/version.py,sha256=lQZu579c_lsErwq2SpYLvpx_OBODgJD0siiUhy2loSA,610
+vysion-2.1.2.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+vysion-2.1.2.dist-info/METADATA,sha256=lyiWJtw7ZnTokNOBGvXDiCKHryc9bARnCTwWsfFec34,1958
+vysion-2.1.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+vysion-2.1.2.dist-info/RECORD,,