72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
|
#
|
||
|
|
# Copyright BitBake Contributors
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: GPL-2.0-only
|
||
|
|
#
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import bb.asyncrpc
|
||
|
|
|
||
|
|
logger = logging.getLogger("BitBake.PRserv")
|
||
|
|
|
||
|
|
class PRAsyncClient(bb.asyncrpc.AsyncClient):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__("PRSERVICE", "1.0", logger)
|
||
|
|
|
||
|
|
async def getPR(self, version, pkgarch, checksum):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"get-pr": {"version": version, "pkgarch": pkgarch, "checksum": checksum}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return response["value"]
|
||
|
|
|
||
|
|
async def test_pr(self, version, pkgarch, checksum):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"test-pr": {"version": version, "pkgarch": pkgarch, "checksum": checksum}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return response["value"]
|
||
|
|
|
||
|
|
async def test_package(self, version, pkgarch):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"test-package": {"version": version, "pkgarch": pkgarch}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return response["value"]
|
||
|
|
|
||
|
|
async def max_package_pr(self, version, pkgarch):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"max-package-pr": {"version": version, "pkgarch": pkgarch}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return response["value"]
|
||
|
|
|
||
|
|
async def importone(self, version, pkgarch, checksum, value):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"import-one": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "value": value}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return response["value"]
|
||
|
|
|
||
|
|
async def export(self, version, pkgarch, checksum, colinfo):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"export": {"version": version, "pkgarch": pkgarch, "checksum": checksum, "colinfo": colinfo}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return (response["metainfo"], response["datainfo"])
|
||
|
|
|
||
|
|
async def is_readonly(self):
|
||
|
|
response = await self.invoke(
|
||
|
|
{"is-readonly": {}}
|
||
|
|
)
|
||
|
|
if response:
|
||
|
|
return response["readonly"]
|
||
|
|
|
||
|
|
class PRClient(bb.asyncrpc.Client):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
self._add_methods("getPR", "test_pr", "test_package", "importone", "export", "is_readonly")
|
||
|
|
|
||
|
|
def _get_async_client(self):
|
||
|
|
return PRAsyncClient()
|