56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from lnbits.helpers import urlsafe_short_hash
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class RelayStatus(BaseModel):
|
|
num_sent_events: Optional[int] = 0
|
|
num_received_events: Optional[int] = 0
|
|
error_counter: Optional[int] = 0
|
|
error_list: Optional[list] = []
|
|
notice_list: Optional[list] = []
|
|
|
|
|
|
class Relay(BaseModel):
|
|
id: Optional[str] = None
|
|
url: Optional[str] = None
|
|
active: Optional[bool] = None
|
|
|
|
connected: Optional[bool] = Field(default=None, no_database=True)
|
|
connected_string: Optional[str] = Field(default=None, no_database=True)
|
|
status: Optional[RelayStatus] = Field(default=None, no_database=True)
|
|
|
|
ping: Optional[int] = Field(default=None, no_database=True)
|
|
|
|
def _init__(self):
|
|
if not self.id:
|
|
self.id = urlsafe_short_hash()
|
|
|
|
|
|
class RelayDb(BaseModel):
|
|
id: str
|
|
url: str
|
|
active: Optional[bool] = True
|
|
|
|
|
|
class TestMessage(BaseModel):
|
|
sender_private_key: Optional[str]
|
|
reciever_public_key: str
|
|
message: str
|
|
|
|
|
|
class TestMessageResponse(BaseModel):
|
|
private_key: str
|
|
public_key: str
|
|
event_json: str
|
|
|
|
|
|
class Config(BaseModel):
|
|
private_ws: bool = True
|
|
public_ws: bool = False
|
|
|
|
|
|
class UserConfig(BaseModel):
|
|
owner_id: str
|
|
extra: Config = Config()
|