SelfPrivacy/selfprivacy-rest-api
1
0
Fork 0
mirror of https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git synced 2024-11-24 04:51:27 +00:00
Code Issues Projects Releases Wiki Activity

Trying out

Browse source
This commit is contained in:
Inex Code 2022-08-02 23:12:48 +03:00
parent 8f940e64fd
commit b965ffd96a
1 changed files with 35 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
selfprivacy_api/graphql/subscriptions/jobs.py
View file

@ -8,6 +8,7 @@ from selfprivacy_api.graphql import IsAuthenticated
from selfprivacy_api.jobs import Job, Jobs
@strawberry.type
class ApiJob:
name: str
@ -21,27 +22,49 @@ class ApiJob:
error: typing.Optional[str]
result: typing.Optional[str]
@strawberry.type
class JobSubscription:
@strawberry.subscription()
async def job_subscription(self) -> AsyncGenerator[typing.List[ApiJob], None]:
is_updated = True
def callback(jobs: typing.List[Job]):
nonlocal is_updated
is_updated = True
print("Subscribing to job updates...")
Jobs().add_observer(callback)
yield [
ApiJob(
name=job.name,
description=job.description,
status=job.status.name,
status_text=job.status_text,
progress=job.progress,
created_at=job.created_at,
updated_at=job.updated_at,
finished_at=job.finished_at,
error=job.error,
result=job.result,
)
for job in Jobs().get_jobs()
]
while True:
if is_updated:
is_updated = False
yield [ ApiJob(
name=job.name,
description=job.description,
status=job.status.name,
status_text=job.status_text,
progress=job.progress,
created_at=job.created_at,
updated_at=job.updated_at,
finished_at=job.finished_at,
error=job.error,
result=job.result,
) for job in Jobs().get_jobs() ]
yield [
ApiJob(
name=job.name,
description=job.description,
status=job.status.name,
status_text=job.status_text,
progress=job.progress,
created_at=job.created_at,
updated_at=job.updated_at,
finished_at=job.finished_at,
error=job.error,
result=job.result,
)
for job in Jobs().get_jobs()
]