pangeacyber/pangea-python

Official Pangea Python Monorepo

pangea-python's Issues

Add support for Pydantic 2.0

ValidationError: 1 validation error for URLReputationResult

Using: pangea-sdk 2.2.1

It appears the Python SDK may not be properly validating results for URLs that are not malicious.

from pangea.config import PangeaConfig
from pangea.services import UrlIntel
from pangea.tools import logger_set_pangea_config
config = PangeaConfig(domain='aws.us.pangea.cloud')
intel = UrlIntel("redacted", config=config, logger_name="intel")

## tftp://harry/vpn/ECTW/beta837-vpn-confg

params= dict(url="tftp://harry/vpn/ECTW/beta837-vpn-confg", provider='crowdstrike', verbose=True, raw=True)
>>> response = intel.reputation(**params)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/opt/pangea/lib/python3.10/site-packages/pangea/services/intel.py", line 685, in reputation
    return self.request.post("v1/reputation", URLReputationResult, data=input.dict(exclude_none=True))
  File "/opt/pangea/lib/python3.10/site-packages/pangea/request.py", line 114, in post
    pangea_response = PangeaResponse(requests_response, result_class=result_class)
  File "/opt/pangea/lib/python3.10/site-packages/pangea/response.py", line 122, in __init__
    self.result_class(**self.raw_result)
  File "/opt/pangea/lib/python3.10/site-packages/pydantic/main.py", line 341, in __init__
    raise validation_error
pydantic.error_wrappers.ValidationError: 1 validation error for URLReputationResult
data -> category
  none is not an allowed value (type=type_error.none.not_allowed)

http://113.235.101.11

>>> params= dict(url="http://113.235.101.11", provider='crowdstrike', verbose=True, raw=True)
>>> response = intel.reputation(**params)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/opt/pangea/lib/python3.10/site-packages/pangea/services/intel.py", line 685, in reputation
    return self.request.post("v1/reputation", URLReputationResult, data=input.dict(exclude_none=True))
  File "/opt/pangea/lib/python3.10/site-packages/pangea/request.py", line 114, in post
    pangea_response = PangeaResponse(requests_response, result_class=result_class)
  File "/opt/pangea/lib/python3.10/site-packages/pangea/response.py", line 122, in __init__
    self.result_class(**self.raw_result)
  File "/opt/pangea/lib/python3.10/site-packages/pydantic/main.py", line 341, in __init__
    raise validation_error
pydantic.error_wrappers.ValidationError: 1 validation error for URLReputationResult
data -> category
  none is not an allowed value (type=type_error.none.not_allowed)

http://113.235.101.11:54384

>>> params= dict(url="http://113.235.101.11:54384", provider='crowdstrike', verbose=True, raw=True)
>>> response = intel.reputation(**params)
>>> response.summary
'URL was found: malicious with score 100'