32 lines
846 B
Python
32 lines
846 B
Python
|
"""Shared code between dmypy.py and dmypy_server.py.
|
||
|
|
|
||
|
|
This should be pretty lightweight and not depend on other mypy code (other than ipc).
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
from typing import Any, Final
|
||
|
|
|
||
|
|
from mypy.ipc import IPCBase
|
||
|
|
|
||
|
|
DEFAULT_STATUS_FILE: Final = ".dmypy.json"
|
||
|
|
|
||
|
|
|
||
|
|
def receive(connection: IPCBase) -> Any:
|
||
|
|
"""Receive JSON data from a connection until EOF.
|
||
|
|
|
||
|
|
Raise OSError if the data received is not valid JSON or if it is
|
||
|
|
not a dict.
|
||
|
|
"""
|
||
|
|
bdata = connection.read()
|
||
|
|
if not bdata:
|
||
|
|
raise OSError("No data received")
|
||
|
|
try:
|
||
|
|
data = json.loads(bdata.decode("utf8"))
|
||
|
|
except Exception as e:
|
||
|
|
raise OSError("Data received is not valid JSON") from e
|
||
|
|
if not isinstance(data, dict):
|
||
|
|
raise OSError(f"Data received is not a dict ({type(data)})")
|
||
|
|
return data
|