oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make nodes only accept requests when they're on

Browse Source
This commit is contained in:
Marek Wolan
2024-03-09 20:47:57 +00:00
parent 289b5c548a
commit 31ae4672ac
6 changed files with 164 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/primaite/interface/request.py
View File

@@ -1,6 +1,6 @@
from typing import Dict, ForwardRef, Literal
from pydantic import BaseModel, ConfigDict, validate_call
from pydantic import BaseModel, ConfigDict, StrictBool, validate_call
RequestResponse = ForwardRef("RequestResponse")
"""This makes it possible to type-hint RequestResponse.from_bool return type."""
@@ -9,7 +9,7 @@ RequestResponse = ForwardRef("RequestResponse")
class RequestResponse(BaseModel):
"""Schema for generic request responses."""
model_config = ConfigDict(extra="forbid")
model_config = ConfigDict(extra="forbid", strict=True)
"""Cannot have extra fields in the response. Anything custom goes into the data field."""
status: Literal["pending", "success", "failure", "unreachable"] = "pending"
@@ -29,7 +29,7 @@ class RequestResponse(BaseModel):
@classmethod
@validate_call
def from_bool(cls, status_bool: bool) -> RequestResponse:
def from_bool(cls, status_bool: StrictBool) -> RequestResponse:
"""
Construct a basic request response from a boolean.