moat-kv 0.70.14__py3-none-any.whl → 0.70.18__py3-none-any.whl

moat/kv/model.py

@@ -39,7 +39,9 @@ class Node:
     tick: int = None
     _present: RangeSet = None  # I have these as valid data. Superset of ``._deleted``.
     _deleted: RangeSet = None  # I have these as no-longer-valid data
-    _reported: RangeSet = None  # somebody else reported these missing data for this node
+    _reported: RangeSet = (
+        None  # somebody else reported these missing data for this node
+    )
     _superseded: RangeSet = None  # I know these once existed, but no more.
     entries: dict = None
     tock: int = 0  # tock when node was last observed
@@ -536,7 +538,9 @@ class NodeEvent:
 class UpdateEvent:
     """Represents an event which updates something."""
 
-    def __init__(self, event: NodeEvent, entry: "Entry", new_value, old_value=NotGiven, tock=None):
+    def __init__(
+        self, event: NodeEvent, entry: "Entry", new_value, old_value=NotGiven, tock=None
+    ):
         self.event = event
         self.entry = entry
         self.new_value = new_value
@@ -875,8 +879,12 @@ class Entry:
 
         if evt.event == self.chain:
             if self._data != evt.new_value:
-                logger.error("Diff %r: has\n%r\nbut should have\n%r\n",
-                        evt.event,self._data, evt.new_value)
+                logger.error(
+                    "Diff %r: has\n%r\nbut should have\n%r\n",
+                    evt.event,
+                    self._data,
+                    evt.new_value,
+                )
             return
 
         if hasattr(evt, "new_value"):
@@ -896,7 +904,9 @@ class Entry:
                 if not loading:
                     logger.warning("*** inconsistency ***")
                     logger.warning("Node: %s", self.path)
-                    logger.warning("Current: %s :%s: %r", self.chain, self.tock, self._data)
+                    logger.warning(
+                        "Current: %s :%s: %r", self.chain, self.tock, self._data
+                    )
                     logger.warning("New: %s :%s: %r", evt.event, evt.tock, evt_val)
                 if evt.tock < self.tock:
                     if not loading:
@@ -924,7 +934,9 @@ class Entry:
             n.seen(t, self)
         await self.updated(evt)
 
-    async def walk(self, proc, acl=None, max_depth=-1, min_depth=0, _depth=0, full=False):
+    async def walk(
+        self, proc, acl=None, max_depth=-1, min_depth=0, _depth=0, full=False
+    ):
         """
         Call coroutine ``proc`` on this node and all its children).
 
@@ -948,7 +960,9 @@ class Entry:
             if k is None and not full:
                 continue
             a = acl.step(k) if acl is not None else None
-            await v.walk(proc, acl=a, max_depth=max_depth, min_depth=min_depth, _depth=_depth)
+            await v.walk(
+                proc, acl=a, max_depth=max_depth, min_depth=min_depth, _depth=_depth
+            )
 
     def serialize(self, chop_path=0, nchain=2, conv=None):
         """Serialize this entry for msgpack.

moat/kv/obj/__init__.py

@@ -16,7 +16,7 @@ try:
 except ImportError:
     from async_generator import asynccontextmanager
 
-from moat.util import NoLock, NotGiven, Path, PathLongener, combine_dict
+from moat.util import NoLock, NotGiven, Path, PathLongener, combine_dict, yload
 
 __all__ = ["ClientEntry", "AttrClientEntry", "ClientRoot"]
 
@@ -261,7 +261,9 @@ class ClientEntry:
             self.chain = r.chain
             return r
 
-    async def delete(self, _locked=False, nchain=0, chain=True, wait=False, recursive=False):
+    async def delete(
+        self, _locked=False, nchain=0, chain=True, wait=False, recursive=False
+    ):
         """Delete this node's value.
 
         This is a coroutine.
@@ -417,7 +419,9 @@ class MirrorRoot(ClientEntry):
             self._seen = dict()
 
     @classmethod
-    async def as_handler(cls, client, cfg=None, key="prefix", subpath=(), name=None, **kw):
+    async def as_handler(
+        cls, client, cfg=None, key="prefix", subpath=(), name=None, **kw
+    ):
         """Return an (or "the") instance of this class.
 
         The handler is created if it doesn't exist.
@@ -430,6 +434,19 @@ class MirrorRoot(ClientEntry):
             d.append(cfg)
 
         defcfg = client._cfg.get(cls.CFG)
+        if not defcfg:
+            # seems we didn't load the class' default config yet.
+            import inspect
+            from pathlib import Path as _Path
+
+            md = inspect.getmodule(cls)
+            try:
+                f = (_Path(md.__file__).parent / "_config.yaml").open("r")
+            except EnvironmentError:
+                pass
+            else:
+                with f:
+                    defcfg = yload(f, attr=True).get("kv",{}).get(cls.CFG)
         if cfg:
             if defcfg:
                 cfg = combine_dict(cfg, defcfg)
@@ -444,7 +461,9 @@ class MirrorRoot(ClientEntry):
             name = str(Path("_moat.kv", client.name, cls.CFG, *subpath))
 
         def make():
-            return client.mirror(cfg[key] + subpath, root_type=cls, need_wait=True, cfg=cfg, **kw)
+            return client.mirror(
+                cfg[key] + subpath, root_type=cls, need_wait=True, cfg=cfg, **kw
+            )
 
         return await client.unique_helper(name, factory=make)
 
@@ -532,7 +551,9 @@ class MirrorRoot(ClientEntry):
                                 pass
 
                             # update entry
-                            entry.chain = None if val is NotGiven else r.get("chain", None)
+                            entry.chain = (
+                                None if val is NotGiven else r.get("chain", None)
+                            )
                             await entry.set_value(val)
 
                             if val is NotGiven and not entry:

moat/kv/obj/command.py

@@ -54,7 +54,9 @@ class _InvSub:
     def id_arg(self, proc):
         if self.id_name is None:
             return proc
-        return click.argument(self.id_name, type=self.id_typ, callback=self.id_cb, nargs=1)(proc)
+        return click.argument(
+            self.id_name, type=self.id_typ, callback=self.id_cb, nargs=1
+        )(proc)
 
     def apply_aux(self, proc):
         for t in self.aux:
@@ -110,9 +112,7 @@ def std_command(cli, *a, **kw):
             \b
             Use '… {tname} -' to list all entries.
             Use '… {tname} NAME' to show details of a single entry.
-            """.format(
-            tname=tname, tlname=tinv.long_name
-        ),
+            """.format(tname=tname, tlname=tinv.long_name),
     )
     @click.argument("name", type=str, nargs=1)
     @click.pass_context
@@ -123,7 +123,9 @@ def std_command(cli, *a, **kw):
 
         if name == "-":
             if ctx.invoked_subcommand is not None:
-                raise click.BadParameter("The name '-' triggers a list and precludes subcommands.")
+                raise click.BadParameter(
+                    "The name '-' triggers a list and precludes subcommands."
+                )
             cnt = 0
             for n in this(obj).all_children if tinv.list_recursive else this(obj):
                 cnt += 1
@@ -150,7 +152,9 @@ def std_command(cli, *a, **kw):
                                 else:
                                     vv = "-"
                             elif isinstance(vv, dict):
-                                vv = " ".join("%s=%s" % (x, y) for x, y in sorted(vv.items()))
+                                vv = " ".join(
+                                    "%s=%s" % (x, y) for x, y in sorted(vv.items())
+                                )
                             print("%s %s %s" % (k, kk, vv), file=obj.stdout)
                     else:
                         print("%s %s" % (k, v), file=obj.stdout)

moat/kv/runner.py

@@ -24,7 +24,14 @@ from moat.util import (
     spawn,
 )
 
-from .actor import ActorState, BrokenState, ClientActor, CompleteState, DetachedState, PartialState
+from .actor import (
+    ActorState,
+    BrokenState,
+    ClientActor,
+    CompleteState,
+    DetachedState,
+    PartialState,
+)
 from .exceptions import ServerError
 from .obj import AttrClientEntry, ClientRoot, MirrorRoot
 
@@ -287,7 +294,9 @@ class CallAdmin:
                         async with slf.client.watch(path, **kw) as watcher:
                             yield watcher
                     else:
-                        raise RuntimeError(f"What should I do with a path marked {path.mark !r}?")
+                        raise RuntimeError(
+                            f"What should I do with a path marked {path.mark !r}?"
+                        )
 
                 with anyio.CancelScope() as sc:
                     slf.scope = sc
@@ -309,7 +318,9 @@ class CallAdmin:
                             elif msg.get("state", "") == "uptodate":
                                 slf.admin._n_watch_seen += 1
                                 if slf.admin._n_watch_seen == slf.admin._n_watch:
-                                    await slf.runner.send_event(ReadyMsg(slf.admin._n_watch_seen))
+                                    await slf.runner.send_event(
+                                        ReadyMsg(slf.admin._n_watch_seen)
+                                    )
 
             def cancel(slf):
                 if slf.scope is None:
@@ -544,7 +555,9 @@ class RunnerEntry(AttrClientEntry):
                 if state.node is not None:
                     raise RuntimeError(f"already running on {state.node}")
                 code = self.root.code.follow(self.code, create=False)
-                data = combine_dict(self.data or {}, code.value.get("default", {}), deep=True)
+                data = combine_dict(
+                    self.data or {}, {} if code.value is NotGiven else code.value.get("default", {}), deep=True
+                )
 
                 if code.is_async:
                     data["_info"] = self._q = create_queue(QLEN)
@@ -561,7 +574,9 @@ class RunnerEntry(AttrClientEntry):
 
                 await state.save(wait=True)
                 if state.node != state.root.name:
-                    raise RuntimeError("Rudely taken away from us.", state.node, state.root.name)
+                    raise RuntimeError(
+                        "Rudely taken away from us.", state.node, state.root.name
+                    )
 
                 data["_self"] = calls = CallAdmin(self, state, data)
                 res = await calls._run(code, data)
@@ -778,7 +793,10 @@ class StateEntry(AttrClientEntry):
             await self.delete()
             return
 
-        if self.node is None or self.node != self.root.name:
+        if self.node is None:
+            return
+        if self.node != self.root.name:
+            self.root.runner.get_node(self.node)
             return
 
         self.stopped = time.time()
@@ -834,7 +852,10 @@ class StateEntry(AttrClientEntry):
             run.scope.cancel()
         elif n is not None:
             logger.warning(
-                "Runner %s at %r: running but node is %s", self.root.name, self.subpath, n
+                "Runner %s at %r: running but node is %s",
+                self.root.name,
+                self.subpath,
+                n,
             )
 
         await run.root.trigger_rescan()
@@ -940,7 +961,9 @@ class _BaseRunnerRoot(ClientRoot):
             cfg_ = client._cfg["runner"]
         else:
             cfg_ = combine_dict(cfg, client._cfg["runner"])
-        return await super().as_handler(client, subpath=subpath, _subpath=subpath, cfg=cfg_, **kw)
+        return await super().as_handler(
+            client, subpath=subpath, _subpath=subpath, cfg=cfg_, **kw
+        )
 
     async def run_starting(self):
         from .code import CodeRoot
@@ -1073,7 +1096,9 @@ class AnyRunnerRoot(_BaseRunnerRoot):
     def __init__(self, *a, **kw):
         super().__init__(*a, **kw)
         self.group = (
-            P(self.client.config.server["root"]) + P(self._cfg["name"]) | "any" | self._path[-1]
+            P(self.client.config.server["root"]) + P(self._cfg["name"])
+            | "any"
+            | self._path[-1]
         )
 
     def get_node(self, name):
@@ -1151,7 +1176,7 @@ class AnyRunnerRoot(_BaseRunnerRoot):
         self._stale_times.append(cur)
         if self._stale_times[0] > cur - self.max_age:
             return
-        if len(self._stale_times) < 5:
+        if len(self._stale_times) <= 2*self._cfg["actor"]["n_hosts"]+1:
             return
         cut = self._stale_times.pop(0)
 
@@ -1253,7 +1278,9 @@ class SingleRunnerRoot(_BaseRunnerRoot):
         async with anyio.create_task_group() as tg:
             age_q = create_queue(1)
 
-            async with ClientActor(self.client, self.name, topic=self.group, cfg=self._cfg) as act:
+            async with ClientActor(
+                self.client, self.name, topic=self.group, cfg=self._cfg
+            ) as act:
                 self._act = act
                 tg.start_soon(self._age_notifier, age_q)
                 await self.spawn(self._run_now)
@@ -1303,7 +1330,9 @@ class AllRunnerRoot(SingleRunnerRoot):
     def __init__(self, *a, **kw):
         super().__init__(*a, **kw)
         self.group = (
-            P(self.client.config.server["root"]) + P(self._cfg["name"]) | "all" | self._path[-1]
+            P(self.client.config.server["root"]) + P(self._cfg["name"])
+            | "all"
+            | self._path[-1]
         )
 
     async def _state_runner(self):

moat/kv/server.py

@@ -151,7 +151,9 @@ class StreamCommand:
         self.client.in_stream[self.seq] = self
         self.qlen = self.client.qlen
         if self.qlen:
-            self.qr = DelayedRead(self.qlen, get_seq=self._get_seq, send_ack=self._send_ack)
+            self.qr = DelayedRead(
+                self.qlen, get_seq=self._get_seq, send_ack=self._send_ack
+            )
             self.dw = DelayedWrite(self.qlen)
         else:
             self.qr = create_queue(1)
@@ -225,7 +227,9 @@ class StreamCommand:
                     await self.send(**res)
             except Exception as exc:
                 if not isinstance(exc, CancelledError):
-                    self.client.logger.exception("ERS%d %r", self.client._client_nr, self.msg)
+                    self.client.logger.exception(
+                        "ERS%d %r", self.client._client_nr, self.msg
+                    )
                 await self.send(error=repr(exc))
             finally:
                 with anyio.move_on_after(2, shield=True):
@@ -475,7 +479,11 @@ class SCmd_get_tree(StreamCommand):
         if root is None:
             root = client.root
             entry, acl = root.follow_acl(
-                msg.path, create=False, nulls_ok=client.nulls_ok, acl=client.acl, acl_key="e"
+                msg.path,
+                create=False,
+                nulls_ok=client.nulls_ok,
+                acl=client.acl,
+                acl_key="e",
             )
         else:
             entry, _ = root.follow_acl(msg.path, create=False, nulls_ok=client.nulls_ok)
@@ -560,7 +568,9 @@ class SCmd_watch(StreamCommand):
                                 return
                             if entry.tock < tock:
                                 res = entry.serialize(
-                                    chop_path=client._chop_path, nchain=nchain, conv=conv
+                                    chop_path=client._chop_path,
+                                    nchain=nchain,
+                                    conv=conv,
                                 )
                                 shorter(res)
                                 if not acl.allows("r"):
@@ -697,7 +707,9 @@ class ServerClient:
         with anyio.CancelScope() as s:
             self.tasks[seq] = s
             if "chain" in msg:
-                msg.chain = NodeEvent.deserialize(msg.chain, cache=self.server.node_cache)
+                msg.chain = NodeEvent.deserialize(
+                    msg.chain, cache=self.server.node_cache
+                )
 
             fn = None
             if msg.get("state", "") != "start":
@@ -912,7 +924,9 @@ class ServerClient:
             if msg.get("nchain", 0):
                 entry["chain"] = None
         else:
-            entry = entry.serialize(chop_path=-1, nchain=msg.get("nchain", 0), conv=self.conv)
+            entry = entry.serialize(
+                chop_path=-1, nchain=msg.get("nchain", 0), conv=self.conv
+            )
         return entry
 
     async def cmd_set_value(self, msg, **kw):
@@ -941,7 +955,9 @@ class ServerClient:
         entry, acl = root.follow_acl(msg.path, acl=acl, acl_key="W", nulls_ok=_nulls_ok)
         if root is self.root and "match" in self.metaroot:
             try:
-                self.metaroot["match"].check_value(None if value is NotGiven else value, entry)
+                self.metaroot["match"].check_value(
+                    None if value is NotGiven else value, entry
+                )
             except ClientError:
                 raise
             except Exception as exc:
@@ -952,7 +968,11 @@ class ServerClient:
         send_prev = True
         nchain = msg.get("nchain", 1)
 
-        if msg.get("idem", False) and type(entry.data) is type(value) and entry.data == value:
+        if (
+            msg.get("idem", False)
+            and type(entry.data) is type(value)
+            and entry.data == value
+        ):
             res = attrdict(tock=entry.tock, changed=False)
             if nchain > 0:
                 res.chain = entry.chain.serialize(nchain=nchain)
@@ -960,7 +980,9 @@ class ServerClient:
 
         if "prev" in msg:
             if entry.data != msg.prev:
-                raise ClientError(f"Data is {entry.data !r} not {msg.prev !r} at {msg.path}")
+                raise ClientError(
+                    f"Data is {entry.data !r} not {msg.prev !r} at {msg.path}"
+                )
             send_prev = False
         if "chain" in msg:
             if msg.chain is None:
@@ -987,7 +1009,9 @@ class ServerClient:
         async with self.server.next_event() as event:
             await entry.set_data(
                 event,
-                NotGiven if value is NotGiven else self.conv.dec_value(value, entry=entry),
+                NotGiven
+                if value is NotGiven
+                else self.conv.dec_value(value, entry=entry),
                 server=self.server,
                 tock=self.server.tock,
             )
@@ -1004,7 +1028,11 @@ class ServerClient:
         You usually do this via a stream command.
         """
         msg = UpdateEvent.deserialize(
-            self.root, msg, nulls_ok=self.nulls_ok, conv=self.conv, cache=self.server._nodes
+            self.root,
+            msg,
+            nulls_ok=self.nulls_ok,
+            conv=self.conv,
+            cache=self.server._nodes,
         )
         res = await msg.entry.apply(msg, server=self, root=self.root)
         if res is None:
@@ -1066,10 +1094,15 @@ class ServerClient:
             res = 0
             if entry.data is not None and acl.allows("d"):
                 async with self.server.next_event() as event:
-                    evt = await entry.set_data(event, NotGiven, server=self, tock=self.server.tock)
+                    evt = await entry.set_data(
+                        event, NotGiven, server=self, tock=self.server.tock
+                    )
                     if nchain:
                         r = evt.serialize(
-                            chop_path=self._chop_path, nchain=nchain, with_old=True, conv=self.conv
+                            chop_path=self._chop_path,
+                            nchain=nchain,
+                            with_old=True,
+                            conv=self.conv,
                         )
                         r["seq"] = seq
                         r.pop("new_value", None)  # always None
@@ -1119,7 +1152,9 @@ class ServerClient:
         if msg.typ is None:
             val.pop("current", None)
         elif msg.typ not in a or not len(a[msg.typ]["user"].keys()):
-            raise RuntimeError("You didn't configure this method yet:" + repr((msg.typ, vars(a))))
+            raise RuntimeError(
+                "You didn't configure this method yet:" + repr((msg.typ, vars(a)))
+            )
         else:
             val["current"] = msg.typ
         msg.value = val
@@ -1203,7 +1238,9 @@ class ServerClient:
                             await evt.wait()
                     except Exception as exc:
                         msg = {"error": str(exc)}
-                        if isinstance(exc, ClientError):  # pylint doesn't seem to see this, so …:
+                        if isinstance(
+                            exc, ClientError
+                        ):  # pylint doesn't seem to see this, so …:
                             msg["etype"] = exc.etype  # pylint: disable=no-member  ### YES IT HAS
                         else:
                             self.logger.exception(
@@ -1241,7 +1278,12 @@ class _RecoverControl:
     _id = 0
 
     def __init__(
-        self, server, scope, prio, local_history, sources  # pylint:disable=redefined-outer-name
+        self,
+        server,
+        scope,
+        prio,
+        local_history,
+        sources,  # pylint:disable=redefined-outer-name
     ):
         self.server = server
         self.scope = scope
@@ -1431,13 +1473,17 @@ class Server:
                 yield n
             except BaseException as exc:
                 if n is not None:
-                    self.logger.warning("Deletion %s %d due to %r", self.node, n.tick, exc)
+                    self.logger.warning(
+                        "Deletion %s %d due to %r", self.node, n.tick, exc
+                    )
                     self.node.report_deleted(
-                        RangeSet((nt,)), self  # pylint: disable=used-before-assignment
+                        RangeSet((nt,)),
+                        self,  # pylint: disable=used-before-assignment
                     )
                     with anyio.move_on_after(2, shield=True):
                         await self._send_event(
-                            "info", dict(node="", tick=0, deleted={self.node.name: (nt,)})
+                            "info",
+                            dict(node="", tick=0, deleted={self.node.name: (nt,)}),
                         )
                 raise
             finally:
@@ -1693,7 +1739,9 @@ class Server:
         """
         Process an update message: deserialize it and apply the result.
         """
-        msg = UpdateEvent.deserialize(self.root, msg, cache=self.node_cache, nulls_ok=True)
+        msg = UpdateEvent.deserialize(
+            self.root, msg, cache=self.node_cache, nulls_ok=True
+        )
         await msg.entry.apply(msg, server=self, root=self.paranoid_root)
 
     async def user_info(self, msg):
@@ -1967,7 +2015,9 @@ class Server:
         try:
             # First try to read the host name from the meta-root's
             # "hostmap" entry, if any.
-            hme = self.root.follow(Path(None, "hostmap", host), create=False, nulls_ok=True)
+            hme = self.root.follow(
+                Path(None, "hostmap", host), create=False, nulls_ok=True
+            )
             if hme.data is NotGiven:
                 raise KeyError(host)
         except KeyError:
@@ -2064,7 +2114,11 @@ class Server:
 
                     pl = PathLongener(())
                     res = await client._request(
-                        "get_tree", iter=True, from_server=self.node.name, nchain=-1, path=()
+                        "get_tree",
+                        iter=True,
+                        from_server=self.node.name,
+                        nchain=-1,
+                        path=(),
                     )
                     async for r in res:
                         pl(r)
@@ -2118,7 +2172,9 @@ class Server:
                 if len(self.fetch_missing):
                     self.fetch_running = False
                     for nm in self.fetch_missing:
-                        self.logger.error("Sync: missing: %s %s", nm.name, nm.local_missing)
+                        self.logger.error(
+                            "Sync: missing: %s %s", nm.name, nm.local_missing
+                        )
                     await self.spawn(self.do_send_missing)
                 if self.force_startup or not len(self.fetch_missing):
                     if self.node.tick is None:
@@ -2298,7 +2354,9 @@ class Server:
         if prio is None:
             await anyio.sleep(clock * (1 + self._actor.random / 3))
         else:
-            await anyio.sleep(clock * (1 - (1 / (1 << prio)) / 2 - self._actor.random / 5))
+            await anyio.sleep(
+                clock * (1 - (1 / (1 << prio)) / 2 - self._actor.random / 5)
+            )
 
         self.logger.debug("SendMissingGo %s %s", prio, self.sending_missing)
         while self.sending_missing:
@@ -2309,7 +2367,10 @@ class Server:
             deleted = {}
             for n in nodes:
                 self.logger.debug(
-                    "SendMissingGo %s %r %r", n.name, n.remote_missing, n.local_superseded
+                    "SendMissingGo %s %r %r",
+                    n.name,
+                    n.remote_missing,
+                    n.local_superseded,
                 )
                 k = n.remote_missing & n.local_superseded
                 for r in n.remote_missing & n.local_present:
@@ -2370,12 +2431,18 @@ class Server:
                         await self.tock_seen(m.tock)
                     else:
                         m.tock = self.tock
-                    m = UpdateEvent.deserialize(self.root, m, cache=self.node_cache, nulls_ok=True)
+                    m = UpdateEvent.deserialize(
+                        self.root, m, cache=self.node_cache, nulls_ok=True
+                    )
                     await self.tock_seen(m.tock)
-                    await m.entry.apply(m, server=self, root=self.paranoid_root, loading=True)
+                    await m.entry.apply(
+                        m, server=self, root=self.paranoid_root, loading=True
+                    )
                 elif "info" in m:
                     await self._process_info(m["info"])
-                elif "nodes" in m or "known" in m or "deleted" in m or "tock" in m:  # XXX LEGACY
+                elif (
+                    "nodes" in m or "known" in m or "deleted" in m or "tock" in m
+                ):  # XXX LEGACY
                     await self._process_info(m)
                 else:
                     self.logger.warning("Unknown message in stream: %s", repr(m))
@@ -2501,7 +2568,11 @@ class Server:
             self._savers.append(state)
             try:
                 await self.save_stream(
-                    path=path, stream=stream, done=done, done_val=s, save_state=save_state
+                    path=path,
+                    stream=stream,
+                    done=done,
+                    done_val=s,
+                    save_state=save_state,
                 )
             except EnvironmentError as err:
                 if done is None:
@@ -2511,7 +2582,9 @@ class Server:
                 with anyio.CancelScope(shield=True):
                     sd.set()
 
-    async def run_saver(self, path: str = None, stream=None, save_state=False, wait: bool = True):
+    async def run_saver(
+        self, path: str = None, stream=None, save_state=False, wait: bool = True
+    ):
         """
         Start a task that continually saves to disk.
 
@@ -2532,7 +2605,13 @@ class Server:
         res = None
         if path is not None:
             await self.spawn(
-                partial(self._saver, path=path, stream=stream, save_state=save_state, done=done)
+                partial(
+                    self._saver,
+                    path=path,
+                    stream=stream,
+                    save_state=save_state,
+                    done=done,
+                )
             )
             if wait:
                 res = await done.get()
@@ -2653,7 +2732,9 @@ class Server:
                 assert self.node.tick is None
                 self.node.tick = 0
                 async with self.next_event() as event:
-                    await self.root.set_data(event, self._init, tock=self.tock, server=self)
+                    await self.root.set_data(
+                        event, self._init, tock=self.tock, server=self
+                    )
 
             await self.spawn(self._sigterm)
 

moat/kv/types.py

@@ -306,7 +306,9 @@ class CodecEntry(Entry):
                     ) from exc
                 else:
                     if r != w:
-                        raise ValueError(f"Decoding at {self.path}: {v!r} got {r!r}, not {w!r}")
+                        raise ValueError(
+                            f"Decoding at {self.path}: {v!r} got {r!r}, not {w!r}"
+                        )
 
         if value is not None and value.encode is not None:
             if not value["out"]:
@@ -321,7 +323,9 @@ class CodecEntry(Entry):
                     ) from exc
                 else:
                     if r != w:
-                        raise ValueError(f"Encoding at {self.path}: {v!r} got {r!r}, not {w!r}")
+                        raise ValueError(
+                            f"Encoding at {self.path}: {v!r} got {r!r}, not {w!r}"
+                        )
 
         await super().set(value)
         self._enc = enc