diff options
Diffstat (limited to 'bitbake/lib/prserv/client.py')
-rw-r--r-- | bitbake/lib/prserv/client.py | 46 |
1 files changed, 34 insertions, 12 deletions
diff --git a/bitbake/lib/prserv/client.py b/bitbake/lib/prserv/client.py index 6b81356fac..9f5794c433 100644 --- a/bitbake/lib/prserv/client.py +++ b/bitbake/lib/prserv/client.py | |||
@@ -6,45 +6,67 @@ | |||
6 | 6 | ||
7 | import logging | 7 | import logging |
8 | import bb.asyncrpc | 8 | import bb.asyncrpc |
9 | from . import create_async_client | ||
9 | 10 | ||
10 | logger = logging.getLogger("BitBake.PRserv") | 11 | logger = logging.getLogger("BitBake.PRserv") |
11 | 12 | ||
12 | class PRAsyncClient(bb.asyncrpc.AsyncClient): | 13 | class PRAsyncClient(bb.asyncrpc.AsyncClient): |
13 | def __init__(self): | 14 | def __init__(self): |
14 | super().__init__('PRSERVICE', '1.0', logger) | 15 | super().__init__("PRSERVICE", "1.0", logger) |
15 | 16 | ||
16 | async def getPR(self, version, pkgarch, checksum): | 17 | async def getPR(self, version, pkgarch, checksum, history=False): |
17 | response = await self.invoke( | 18 | response = await self.invoke( |
18 | {'get-pr': {'version': version, 'pkgarch': pkgarch, 'checksum': checksum}} | 19 | {"get-pr": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "history": history}} |
19 | ) | 20 | ) |
20 | if response: | 21 | if response: |
21 | return response['value'] | 22 | return response["value"] |
23 | |||
24 | async def test_pr(self, version, pkgarch, checksum, history=False): | ||
25 | response = await self.invoke( | ||
26 | {"test-pr": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "history": history}} | ||
27 | ) | ||
28 | if response: | ||
29 | return response["value"] | ||
30 | |||
31 | async def test_package(self, version, pkgarch): | ||
32 | response = await self.invoke( | ||
33 | {"test-package": {"version": version, "pkgarch": pkgarch}} | ||
34 | ) | ||
35 | if response: | ||
36 | return response["value"] | ||
37 | |||
38 | async def max_package_pr(self, version, pkgarch): | ||
39 | response = await self.invoke( | ||
40 | {"max-package-pr": {"version": version, "pkgarch": pkgarch}} | ||
41 | ) | ||
42 | if response: | ||
43 | return response["value"] | ||
22 | 44 | ||
23 | async def importone(self, version, pkgarch, checksum, value): | 45 | async def importone(self, version, pkgarch, checksum, value): |
24 | response = await self.invoke( | 46 | response = await self.invoke( |
25 | {'import-one': {'version': version, 'pkgarch': pkgarch, 'checksum': checksum, 'value': value}} | 47 | {"import-one": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "value": value}} |
26 | ) | 48 | ) |
27 | if response: | 49 | if response: |
28 | return response['value'] | 50 | return response["value"] |
29 | 51 | ||
30 | async def export(self, version, pkgarch, checksum, colinfo): | 52 | async def export(self, version, pkgarch, checksum, colinfo, history=False): |
31 | response = await self.invoke( | 53 | response = await self.invoke( |
32 | {'export': {'version': version, 'pkgarch': pkgarch, 'checksum': checksum, 'colinfo': colinfo}} | 54 | {"export": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "colinfo": colinfo, "history": history}} |
33 | ) | 55 | ) |
34 | if response: | 56 | if response: |
35 | return (response['metainfo'], response['datainfo']) | 57 | return (response["metainfo"], response["datainfo"]) |
36 | 58 | ||
37 | async def is_readonly(self): | 59 | async def is_readonly(self): |
38 | response = await self.invoke( | 60 | response = await self.invoke( |
39 | {'is-readonly': {}} | 61 | {"is-readonly": {}} |
40 | ) | 62 | ) |
41 | if response: | 63 | if response: |
42 | return response['readonly'] | 64 | return response["readonly"] |
43 | 65 | ||
44 | class PRClient(bb.asyncrpc.Client): | 66 | class PRClient(bb.asyncrpc.Client): |
45 | def __init__(self): | 67 | def __init__(self): |
46 | super().__init__() | 68 | super().__init__() |
47 | self._add_methods('getPR', 'importone', 'export', 'is_readonly') | 69 | self._add_methods("getPR", "test_pr", "test_package", "max_package_pr", "importone", "export", "is_readonly") |
48 | 70 | ||
49 | def _get_async_client(self): | 71 | def _get_async_client(self): |
50 | return PRAsyncClient() | 72 | return PRAsyncClient() |