occheung
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

browser: add update method to dataset controller

This commit is contained in:
Florian Agbuya 2023-06-06 18:59:01 +08:00 committed by Sébastien Bourdeauducq
parent bf46ce4a92
commit 7ad32d903a
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
artiq/browser/datasets.py
View File

@ -28,21 +28,34 @@ class DatasetCtl:
self.master_host = master_host
self.master_port = master_port
async def set(self, key, value, persist=None):
logger.info("Uploading dataset '%s' to master...", key)
async def _execute_rpc(self, op_name, key_or_mod, value=None, persist=None):
logger.info("Starting %s operation on %s", op_name, key_or_mod)
try:
remote = RPCClient()
await remote.connect_rpc(self.master_host, self.master_port,
"master_dataset_db")
try:
await remote.set(key, value, persist)
if op_name == "set":
await remote.set(key_or_mod, value, persist)
elif op_name == "update":
await remote.update(key_or_mod)
else:
logger.error("Invalid operation: %s", op_name)
return
finally:
remote.close_rpc()
except:
logger.error("Failed uploading dataset '%s'",
key, exc_info=True)
logger.error("Failed %s operation on %s", op_name,
key_or_mod, exc_info=True)
else:
logger.info("Finished uploading dataset '%s'", key)
logger.info("Finished %s operation on %s", op_name,
key_or_mod)
async def set(self, key, value, persist=None):
await self._execute_rpc("set", key, value, persist)
async def update(self, mod):
await self._execute_rpc("update", mod)
class DatasetsDock(QtWidgets.QDockWidget):