fix: mypy errors
This commit is contained in:
committed by
Sebastian Geisler
parent
976a3d4e5c
commit
c3e337a319
@@ -1,23 +1,22 @@
|
||||
from requests import get, post
|
||||
from os import getenv
|
||||
from .base import InvoiceResponse, PaymentResponse, PaymentStatus, Wallet
|
||||
from lightning import LightningRpc
|
||||
from lightning import LightningRpc # type: ignore
|
||||
import random
|
||||
|
||||
class CLightningWallet(Wallet):
|
||||
|
||||
def __init__(self):
|
||||
self.l1 = LightningRpc(getenv("CLIGHTNING_RPC"))
|
||||
|
||||
|
||||
def create_invoice(self, amount: int, memo: str = "") -> InvoiceResponse:
|
||||
label = "lbl{}".format(random.random())
|
||||
label = "lbl{}".format(random.random())
|
||||
r = self.l1.invoice(amount*1000, label, memo, exposeprivatechannels=True)
|
||||
ok, checking_id, payment_request, error_message = True, r["payment_hash"], r["bolt11"], None
|
||||
return InvoiceResponse(ok, checking_id, payment_request, error_message)
|
||||
|
||||
def pay_invoice(self, bolt11: str) -> PaymentResponse:
|
||||
r = self.l1.pay(bolt11)
|
||||
ok, checking_id, fee_msat, error_message = True, None, None, None
|
||||
ok, checking_id, fee_msat, error_message = True, None, 0, None
|
||||
return PaymentResponse(ok, checking_id, fee_msat, error_message)
|
||||
|
||||
def get_invoice_status(self, checking_id: str) -> PaymentStatus:
|
||||
@@ -29,8 +28,8 @@ class CLightningWallet(Wallet):
|
||||
def get_payment_status(self, checking_id: str) -> PaymentStatus:
|
||||
r = self.l1.listsendpays(checking_id)
|
||||
if not r.ok:
|
||||
return PaymentStatus(r, None)
|
||||
payments = [p for p in r.json()["payments"] if p["payment_hash"] == payment_hash]
|
||||
return PaymentStatus(None)
|
||||
payments = [p for p in r.json()["payments"] if p["payment_hash"] == checking_id]
|
||||
payment = payments[0] if payments else None
|
||||
statuses = {"UNKNOWN": None, "IN_FLIGHT": None, "SUCCEEDED": True, "FAILED": False}
|
||||
return PaymentStatus(statuses[payment["status"]] if payment else None)
|
||||
|
||||
Reference in New Issue
Block a user