Convert ws_schemas to pydantic 2.0
This commit is contained in:
@@ -2,15 +2,14 @@ from datetime import datetime
|
|||||||
from typing import Any, Dict, List, Optional, TypedDict
|
from typing import Any, Dict, List, Optional, TypedDict
|
||||||
|
|
||||||
from pandas import DataFrame
|
from pandas import DataFrame
|
||||||
from pydantic import BaseModel
|
from pydantic import ConfigDict, BaseModel
|
||||||
|
|
||||||
from freqtrade.constants import PairWithTimeframe
|
from freqtrade.constants import PairWithTimeframe
|
||||||
from freqtrade.enums.rpcmessagetype import RPCMessageType, RPCRequestType
|
from freqtrade.enums.rpcmessagetype import RPCMessageType, RPCRequestType
|
||||||
|
|
||||||
|
|
||||||
class BaseArbitraryModel(BaseModel):
|
class BaseArbitraryModel(BaseModel):
|
||||||
class Config:
|
model_config = ConfigDict(arbitrary_types_allowed=True)
|
||||||
arbitrary_types_allowed = True
|
|
||||||
|
|
||||||
|
|
||||||
class WSRequestSchema(BaseArbitraryModel):
|
class WSRequestSchema(BaseArbitraryModel):
|
||||||
@@ -27,9 +26,7 @@ class WSMessageSchemaType(TypedDict):
|
|||||||
class WSMessageSchema(BaseArbitraryModel):
|
class WSMessageSchema(BaseArbitraryModel):
|
||||||
type: RPCMessageType
|
type: RPCMessageType
|
||||||
data: Optional[Any] = None
|
data: Optional[Any] = None
|
||||||
|
model_config = ConfigDict(extra='allow')
|
||||||
class Config:
|
|
||||||
extra = 'allow'
|
|
||||||
|
|
||||||
|
|
||||||
# ------------------------------ REQUEST SCHEMAS ----------------------------
|
# ------------------------------ REQUEST SCHEMAS ----------------------------
|
||||||
|
|||||||
Reference in New Issue
Block a user