async def delete_policy(self, policy_id: bytes):
request = starbelly_pb2.Request(
request_id=next(self.request_id),
delete_policy=starbelly_pb2.RequestDeletePolicy(
policy_id=policy_id),
)
return await self.send_request(request)
async def get_schedule(self, schedule_id: bytes) -> starbelly_pb2.Schedule:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
get_schedule=starbelly_pb2.RequestGetSchedule(
schedule_id=schedule_id),
)
return await self.send_request(request)
async def delete_domain_login(self, domain_login_id: bytes):
request = starbelly_pb2.Request(
request_id=next(self.request_id),
delete_domain_login=starbelly_pb2.RequestDeleteDomainLogin(
domain_login_id=domain_login_id),
)
return await self.send_request(request)
async def delete_schedule(self, schedule_id: bytes):
request = starbelly_pb2.Request(
request_id=next(self.request_id),
delete_schedule=starbelly_pb2.RequestDeleteSchedule(
schedule_id=schedule_id),
)
return await self.send_request(request)
async def get_captcha_solver(self, solver_id: bytes):
request = starbelly_pb2.Request(
request_id=next(self.request_id),
get_captcha_solver=starbelly_pb2.RequestGetCaptchaSolver(
solver_id=solver_id),
)
return await self.send_request(request)
async def get_domain_login(
self, domain_login_id: bytes) -> starbelly_pb2.DomainLogin:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
get_domain_login=starbelly_pb2.RequestGetDomainLogin(
domain_login_id=domain_login_id),
)
return await self.send_request(request)
async def unsubscribe(self, subscripion_id: int):
request = starbelly_pb2.Request(
request_id=next(self.request_id),
unsubscribe=starbelly_pb2.RequestUnsubscribe(
subscripion_id=subscripion_id),
)
# TODO: implemement unsubscription
response = await self.send_request(request)
async def subscribe_task_monitor(
self, period: int = None) -> starbelly_pb2.ResponseNewSubscription:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
subscribe_task_monitor=starbelly_pb2.RequestSubscribeTaskMonitor(
period=period),
)
async for event in self.subscribe(request):
yield event
async def list_captcha_solvers(
self,
page: int = None) -> starbelly_pb2.ResponseListCaptchaSolvers:
page = starbelly_pb2.Page(limit=limit, offset=offset)
request = starbelly_pb2.Request(
request_id=next(self.request_id),
list_captcha_solvers=starbelly_pb2.RequestListCaptchaSolvers(
page=page),
)
return await self.send_request(request)
async def subscribe_resource_monitor(
self,
history: int = None) -> starbelly_pb2.ResponseNewSubscription:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
subscribe_resource_monitor=starbelly_pb2.
RequestSubscribeResourceMonitor(history=history),
)
async for event in self.subscribe(request):
yield event
async def list_rate_limits(
self,
offset: int = None,
limit: int = None) -> starbelly_pb2.ResponseListRateLimits:
page = starbelly_pb2.Page(limit=limit, offset=offset)
request = starbelly_pb2.Request(
request_id=next(self.request_id),
list_rate_limits=starbelly_pb2.RequestListRateLimits(page=page),
)
return await self.send_request(request)
async def subscribe_job_status(
self,
min_interval: float = None
) -> starbelly_pb2.ResponseNewSubscription:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
subscribe_job_status=starbelly_pb2.RequestSubscribeJobStatus(
min_interval=min_interval),
)
async for event in self.subscribe(request):
yield event
async def performance_profile(
self,
duration: float = None,
sort_by: str = None,
top_n: int = None) -> starbelly_pb2.ResponsePerformanceProfile:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
performance_profile=starbelly_pb2.RequestPerformanceProfile(
duration=duration, sort_by=sort_by, top_n=top_n),
)
return await self.send_request(request)
async def list_domain_logins(
self,
offset: int = None,
limit: int = None) -> starbelly_pb2.ResponseListDomainLogins:
page = starbelly_pb2.Page(limit=limit, offset=offset)
request = starbelly_pb2.Request(
request_id=next(self.request_id),
list_domain_logins=starbelly_pb2.RequestListDomainLogins(
page=page),
)
return await self.send_request(request)
async def delete_captcha_solver(self, solver_id: bytes):
"""Delete captcha solver
Args:
solver_id (bytes): The solver ID
Returns:
[type]: [description]
"""
request = starbelly_pb2.Request(
request_id=next(self.request_id),
delete_captcha_solver=starbelly_pb2.RequestDeleteCaptchaSolver(
solver_id=solver_id),
)
return await self.send_request(request)
async def subscribe_job_sync(
self,
job_id: bytes,
sync_token: bytes = None,
compression_ok: bool = True
) -> starbelly_pb2.ResponseNewSubscription:
# TODO: Fixme - this returns no data and the request times out.
request = starbelly_pb2.Request(
request_id=next(self.request_id),
subscribe_job_sync=starbelly_pb2.RequestSubscribeJobSync(
job_id=job_id,
sync_token=sync_token,
compression_ok=compression_ok),
)
async for event in self.subscribe(request):
yield event
async def list_jobs(
self,
offset: int = None,
limit: int = None,
started_after: str = None,
tag: str = None,
schedule_id: bytes = None,
) -> starbelly_pb2.ResponseListJobs:
page = starbelly_pb2.Page(limit=limit, offset=offset)
request = starbelly_pb2.Request(
request_id=next(self.request_id),
list_jobs=starbelly_pb2.RequestListJobs(
page=page,
started_after=started_after,
tag=tag,
schedule_id=schedule_id,
),
)
return await self.send_request(request)
async def set_job(
self,
run_state: starbelly_pb2.JobRunState = None,
policy_id: bytes = None,
seeds: list = None,
name: str = None,
job_id: bytes = None,
tags: list = [],
) -> starbelly_pb2.ResponseNewJob:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
set_job=starbelly_pb2.RequestSetJob(
job_id=job_id,
run_state=run_state,
policy_id=policy_id,
seeds=seeds,
name=name,
tags=tags,
),
)
return await self.send_request(request)
async def get_job_items(
self,
job_id: bytes,
include_success: bool = True,
include_error: bool = True,
include_exception: bool = True,
compression_ok: bool = True,
offset: int = None,
limit: int = None,
) -> starbelly_pb2.ResponseListItems:
page = starbelly_pb2.Page(limit=limit, offset=offset)
request = starbelly_pb2.Request(
request_id=next(self.request_id),
get_job_items=starbelly_pb2.RequestGetJobItems(
job_id=job_id,
include_success=include_success,
include_error=include_error,
include_exception=include_exception,
compression_ok=compression_ok,
page=page,
),
)
return await self.send_request(request)
async def get_policy(self, policy_id: bytes) -> starbelly_pb2.Policy:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
get_policy=starbelly_pb2.RequestGetPolicy(policy_id=policy_id),
)
return await self.send_request(request)
async def delete_job(self, job_id: bytes):
request = starbelly_pb2.Request(
request_id=next(self.request_id),
delete_job=starbelly_pb2.RequestDeleteJob(job_id=job_id),
)
return await self.send_request(request)
async def get_job(self, job_id: bytes) -> starbelly_pb2.Job:
request = starbelly_pb2.Request(
request_id=next(self.request_id),
get_job=starbelly_pb2.RequestGetJob(job_id=job_id),
)
return await self.send_request(request)
