vysion 2.0.9__tar.gz → 2.0.11__tar.gz

{vysion-2.0.9 → vysion-2.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vysion
-Version: 2.0.9
+Version: 2.0.11
 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.0.9](https://pypi.org/project/vysion/)
+Latest version: [2.0.11](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.9 → vysion-2.0.11}/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.9](https://pypi.org/project/vysion/)
+Latest version: [2.0.11](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.9 → vysion-2.0.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vysion"
-version = "2.0.9"
+version = "2.0.11"
 description = "The official Python client library for Vysion"
 homepage = "https://vysion.ai"
 repository = "https://github.com/ByronLabs/vysion-py"

{vysion-2.0.9 → vysion-2.0.11}/vysion/client/client.py

@@ -205,27 +205,59 @@ class Client(BaseClient):
         self,
         url: str,
         page: int = 1,
+        page_size: int = 10,
         lte: datetime = None,
         gte: datetime = None,
     ) -> VysionResponse[DocumentHit]:
-        url = self._build_api_url__("document/url", url, page=page, lte=lte, gte=gte)
+        url = self._build_api_url__(
+            "document/url",
+            url,
+            page=page,
+            page_size=page_size,
+            lte=lte,
+            gte=gte,
+        )
 
         result = VysionResponse[DocumentHit].model_validate(self._make_request(url))
         return result.data
 
     @vysion_error_manager
-    def get_tag(self, tag: str) -> VysionResponse[DocumentHit]:
-        url = self._build_api_url__("document/tag", tag)
+    def get_tag(
+        self,
+        tag: str,
+        gte: datetime = None,
+        lte: datetime = None,
+        page: int = 1,
+        page_size: int = 10,
+    ) -> VysionResponse[DocumentHit]:
+        url = self._build_api_url__(
+            "document/tag",
+            tag,
+            gte=gte,
+            lte=lte,
+            page=page,
+            page_size=page_size,
+        )
 
         result = VysionResponse[DocumentHit].model_validate(self._make_request(url))
         return result.data
 
     @vysion_error_manager
     def find_email(
-        self, email: str, page: int = 1, lte: datetime = None, gte: datetime = None
+        self,
+        email: str,
+        page: int = 1,
+        page_size: int = 10,
+        lte: datetime = None,
+        gte: datetime = None
     ) -> VysionResponse[DocumentHit]:
         url = self._build_api_url__(
-            "document/email", email, page=page, lte=lte, gte=gte
+            "document/email",
+            email,
+            page=page,
+            page_size=page_size,
+            lte=lte,
+            gte=gte
         )
 
         result = VysionResponse[DocumentHit].model_validate(self._make_request(url))
@@ -237,6 +269,7 @@ class Client(BaseClient):
         country_code: str,
         phone_number: str,
         page: int = 1,
+        page_size: int = 10,
         lte: datetime = None,
         gte: datetime = None,
     ) -> VysionResponse[DocumentHit]:
@@ -244,6 +277,7 @@ class Client(BaseClient):
             "document/phone/" + country_code,
             phone_number,
             page=page,
+            page_size=page_size,
             lte=lte,
             gte=gte,
         )
@@ -257,11 +291,17 @@ class Client(BaseClient):
         chain: str,
         address: str,
         page: int = 1,
+        page_size: int = 10,
         lte: datetime = None,
         gte: datetime = None,
     ) -> VysionResponse[DocumentHit]:
         url = self._build_api_url__(
-            "document/wallet/" + chain, address, page=page, lte=lte, gte=gte
+            "document/wallet/" + chain,
+            address,
+            page=page,
+            page_size=page_size,
+            lte=lte,
+            gte=gte
         )
 
         result = VysionResponse[DocumentHit].model_validate(self._make_request(url))
@@ -357,6 +397,7 @@ class Client(BaseClient):
         gte: datetime = None,
         lte: datetime = None,
         page: int = 1,
+        page_size: int = 10,
         username: str = None,
     ) -> VysionResponse[ImMessageHit]:
         url = self._build_api_url__(
@@ -365,6 +406,7 @@ class Client(BaseClient):
             gte=gte,
             lte=lte,
             page=page,
+            page_size=page_size,
             username=username,
         )
 
@@ -440,4 +482,4 @@ class RansomwareFeed(DaylyFeed):
             yield self._make_request(url)
 
 
-# TODO /api/v1/feeds
+# TODO /api/v1/feeds

{vysion-2.0.9 → vysion-2.0.11}/vysion/dto/dto.py

@@ -287,10 +287,10 @@ class LanguagePair(BaseModel):
 
 
 class ImMessageHit(BaseModel):
-    userId: Optional[int] = Field(default_factory=lambda: None)
+    userId: Optional[Union[int, str]] = Field(default_factory=lambda: None)
     username: Optional[str] = Field(default_factory=lambda: None)
-    channelId: Optional[int] = Field(default_factory=lambda: None)
-    messageId: str
+    channelId: Optional[Union[int, str]] = Field(default_factory=lambda: None)
+    messageId: Union[int, str]
     message: Optional[str] = Field(default_factory=lambda: None)
     channelTitle: Optional[str] = Field(default_factory=lambda: None)
     languages: Optional[List[LanguagePair]] = Field(default_factory=lambda: None)
@@ -298,22 +298,35 @@ class ImMessageHit(BaseModel):
     sha256sum: Optional[str] = None
     media: Optional[str] = Field(default_factory=lambda: None)
     detectionDate: datetime
-
-    @field_validator("messageId")
-    def validate_messageId(cls, v: int) -> int:
-        if not v:
-            raise ValueError("MessageId field cannot be empty")
-        return v
-
+    serverId: Optional[Union[int, str]] = Field(default_factory=lambda: None) #Discord Exclusive
+    serverTitle: Optional[str] = Field(default_factory=lambda: None) #Discord Exclusive
+    platform: Optional[str] = Field(default_factory=lambda: None) #Discord Exclusive
+    
+class ImMessageCardHit(BaseModel):
+    userId: Optional[Union[int, str]] = Field(default_factory=lambda: None)
+    username: Optional[str] = Field(default_factory=lambda: None)
+    channelId: Optional[int] = Field(default_factory=lambda: None)
+    messageId: Union[int, str]
+    message: Optional[str] = Field(default_factory=lambda: None)
+    channelTitle: Optional[str] = Field(default_factory=lambda: None)
+    languages: Optional[List[LanguagePair]] = Field(default_factory=lambda: None)
+    detectionDate: datetime
+    serverId: Optional[Union[int, str]] = Field(default_factory=lambda: None) #Discord Exclusive
+    serverTitle: Optional[str] = Field(default_factory=lambda: None) #Discord Exclusive
+    platform: Optional[str] = Field(default_factory=lambda: None)
 
 class ImProfileHit(BaseModel):
-    userId: int
+    userId: Union[int, str]
     usernames: Optional[List[str]] = Field(default_factory=lambda: None)
     firstName: Optional[List[str]] = Field(default_factory=lambda: None)
     lastName: Optional[List[str]] = Field(default_factory=lambda: None)
     detectionDate: datetime
     profilePhoto: Optional[List[str]] = Field(default_factory=lambda: None)
-
+    bot: Optional[bool] = Field(default_factory=lambda: None) #Discord Exclusive
+    discordLink: Optional[List[str]] = Field(default_factory=lambda: None) #Discord Exclusive
+    discriminator: Optional[List[int]] = Field(default_factory=lambda: None) #Discord Exclusive
+    platform: Optional[str] = Field(default_factory=lambda: None)
+    
     @field_validator("userId")
     def validate_userId(cls, v: int) -> int:
         if not v:
@@ -328,18 +341,36 @@ class ImProfileHit(BaseModel):
 
 
 class ImChannelHit(BaseModel):
-    channelId: int
+    channelId: Union[int, str]
     channelTitles: Optional[List[str]] = Field(default_factory=lambda: None)
     detectionDate: datetime
     creationDate: datetime
-    channelPhoto: Optional[List[str]] = Field(default_factory=lambda: None)
+    channelPhoto: Optional[List[str]] = Field(default_factory=lambda: None) #Telegram Exclusive
+    serverId: Optional[Union[int, str]] = Field(default_factory=lambda: None) #Discord Exclusive
+    serverTitle: Optional[List[str]] = Field(default_factory=lambda: None) #Discord Exclusive
+    platform: Optional[str] = Field(default_factory=lambda: None)
+    
+    @field_validator("detectionDate")
+    def validate_detectionDate(cls, v: datetime) -> datetime:
+        if not v:
+            raise ValueError("DetectionDate field cannot be empty")
+        return v
 
-    @field_validator("channelId")
-    def validate_channelId(cls, v: int) -> int:
+    @field_validator("creationDate")
+    def validate_creationDate(cls, v: datetime) -> datetime:
         if not v:
-            raise ValueError("channelId field cannot be empty")
+            raise ValueError("creationDate field cannot be empty")
         return v
 
+class ImServerHit(BaseModel):
+    serverId: Union[int, str]
+    serverTitles: Optional[List[str]] = Field(default_factory=lambda: None)
+    detectionDate: datetime
+    creationDate: datetime
+    serverPhoto: Optional[List[str]] = Field(default_factory=lambda: None)
+    memberCount: Optional[int] = Field(default_factory=lambda: None)
+    discordLink: Optional[List[str]] = Field(default_factory=lambda: None)
+
     @field_validator("detectionDate")
     def validate_detectionDate(cls, v: datetime) -> datetime:
         if not v:
@@ -352,7 +383,6 @@ class ImChannelHit(BaseModel):
             raise ValueError("creationDate field cannot be empty")
         return v
 
-
 class ImFeedHit(BaseModel):
     id: str
     telegram: List[str]

{vysion-2.0.9 → vysion-2.0.11}/vysion/version.py

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