48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
from sqlite3 import Row
|
|
from typing import List, Optional
|
|
|
|
from lnbits.helpers import urlsafe_short_hash
|
|
from pydantic import BaseModel
|
|
|
|
|
|
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
|
|
connected: Optional[bool] = None
|
|
connected_string: Optional[str] = None
|
|
status: Optional[RelayStatus] = None
|
|
active: Optional[bool] = None
|
|
ping: Optional[int] = None
|
|
|
|
def _init__(self):
|
|
if not self.id:
|
|
self.id = urlsafe_short_hash()
|
|
|
|
@classmethod
|
|
def from_row(cls, row: Row) -> "Relay":
|
|
return cls(**dict(row))
|
|
|
|
|
|
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
|