40 lines
1 KiB
Python
40 lines
1 KiB
Python
# Data models for your extension
|
|
|
|
from typing import Optional
|
|
|
|
from fastapi import Request
|
|
from lnurl.core import encode as lnurl_encode
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class CreateMyExtensionData(BaseModel):
|
|
name: str
|
|
lnurlpayamount: int
|
|
lnurlwithdrawamount: int
|
|
wallet: Optional[str] = None
|
|
total: int = 0
|
|
|
|
|
|
class MyExtension(BaseModel):
|
|
id: str
|
|
wallet: str
|
|
lnurlpayamount: int
|
|
name: str
|
|
lnurlwithdrawamount: int
|
|
total: int
|
|
|
|
def lnurlpay(self, req: Request) -> str:
|
|
url = req.url_for("myextension.api_lnurl_pay", myextension_id=self.id)
|
|
url_str = str(url)
|
|
if url.netloc.endswith(".onion"):
|
|
url_str = url_str.replace("https://", "http://")
|
|
|
|
return lnurl_encode(url_str)
|
|
|
|
def lnurlwithdraw(self, req: Request) -> str:
|
|
url = req.url_for("myextension.api_lnurl_withdraw", myextension_id=self.id)
|
|
url_str = str(url)
|
|
if url.netloc.endswith(".onion"):
|
|
url_str = url_str.replace("https://", "http://")
|
|
|
|
return lnurl_encode(url_str)
|