Skip to content

Commit f995dcb

Browse files
committed
Added missed datadome.py and improved code clarity
1 parent 2ccb40d commit f995dcb

File tree

3 files changed

+25
-3
lines changed

3 files changed

+25
-3
lines changed

capmonster_python/datadome.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
from .capmonster import UserAgent
2+
3+
4+
class DataDomeTask(UserAgent):
5+
def __init__(self, client_key):
6+
super(DataDomeTask, self).__init__(client_key)
7+
8+
def create_task(self, website_url: str, metadata: object):
9+
data = {
10+
"client_key": self._client_key,
11+
"task": {
12+
"type": "CustomTask",
13+
"class": "DataDome",
14+
"websiteURL": website_url,
15+
"metadata": metadata,
16+
}
17+
}
18+
data, is_user_agent = self._add_user_agent(data)
19+
return self._make_request("createTask", data=data).get("taskId")

capmonster_python/recaptcha_v2_enterprise.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
from .capmonster import Proxy, UserAgent
21
from typing import Union
2+
from .capmonster import Proxy, UserAgent
3+
34

45
class RecaptchaV2EnterpriseTask(UserAgent, Proxy):
56
def __init__(self, client_key):
67
super(RecaptchaV2EnterpriseTask, self).__init__(client_key)
78

89
def create_task(self, website_url: str, website_key: str,
9-
enterprise_payload = None, api_domain: str = None,
10+
enterprise_payload=None, api_domain: str = None,
1011
cookies: Union[dict, list, str] = None,
1112
no_cache: bool = None):
1213
data = {

capmonster_python/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def checker(f):
1717
@wraps(f)
1818
def wrap(*args, **kwargs):
1919
rf = f(*args, **kwargs)
20-
if type(rf) == dict:
20+
if type(rf) is dict:
2121
if rf.get("errorId") == 0:
2222
return rf
2323
else:
@@ -29,5 +29,7 @@ def wrap(*args, **kwargs):
2929
error_code="CAPMONSTER_API_ERROR",
3030
error_description="Sometimes can be happen if capmonster_python "
3131
"servers there is too much intensity")
32+
3233
return wrap
34+
3335
return checker

0 commit comments

Comments
 (0)