refactor: extract simpler methods

This commit is contained in:
Vlad Stan 2023-03-06 18:01:05 +02:00
parent 35298a4f44
commit 1b317b1b9b
2 changed files with 46 additions and 52 deletions

View file

@ -256,6 +256,7 @@ class OrderContact(BaseModel):
nostr: Optional[str]
phone: Optional[str]
email: Optional[str]
address: Optional[str]
class PartialOrder(BaseModel):
@ -307,15 +308,3 @@ class PaymentRequest(BaseModel):
id: str
message: Optional[str]
payment_options: List[PaymentOption]
def to_nostr_event(self, author_pubkey: str, to_pubkey: str) -> NostrEvent:
event = NostrEvent(
pubkey=author_pubkey,
created_at=round(time.time()),
kind=4,
tags=[["p", to_pubkey]],
content=json.dumps(self.dict(), separators=(",", ":"), ensure_ascii=False),
)
event.id = event.event_id
return event

View file

@ -89,20 +89,22 @@ async def handle_message(msg: str):
if type.upper() == "EVENT":
event = NostrEvent(**event)
if event.kind == 4:
await handle_nip04_message(public_key, event)
except Exception as ex:
logger.warning(ex)
async def handle_nip04_message(public_key: str, event: NostrEvent):
merchant = await get_merchant_by_pubkey(public_key)
assert merchant, f"Merchant not found for public key '{public_key}'"
clear_text_msg = merchant.decrypt_message(event.content, event.pubkey)
dm_resp = await handle_dirrect_message(
event.pubkey, event.id, clear_text_msg
)
dm_resp = await handle_dirrect_message(event.pubkey, event.id, clear_text_msg)
if dm_resp:
dm_event = merchant.build_dm_event(dm_resp, event.pubkey)
await publish_nostr_event(dm_event)
except Exception as ex:
logger.warning(ex)
async def handle_dirrect_message(
from_pubkey: str, event_id: str, msg: str
@ -110,18 +112,28 @@ async def handle_dirrect_message(
order, text_msg = order_from_json(msg)
try:
if order:
### check that event_id not parsed already
order["pubkey"] = from_pubkey
order["event_id"] = event_id
partial_order = PartialOrder(**order)
partial_order.validate_order()
assert len(partial_order.items) != 0, "Order has no items. Order: " + msg
return await handle_new_order(PartialOrder(**order))
else:
print("### text_msg", text_msg)
return None
except Exception as ex:
logger.warning(ex)
return None
first_product_id = partial_order.items[0].product_id
wallet_id = await get_wallet_for_product(first_product_id)
async def handle_new_order(order: PartialOrder):
### check that event_id not parsed already
order.validate_order()
assert (
wallet_id
), f"Cannot find wallet id for product id: {first_product_id}"
len(order.items) != 0
), f"Order has no items. Order: '{order.id}' ({order.event_id})"
first_product_id = order.items[0].product_id
wallet_id = await get_wallet_for_product(first_product_id)
assert wallet_id, f"Cannot find wallet id for product id: {first_product_id}"
wallet = await get_wallet(wallet_id)
assert wallet, f"Cannot find wallet for product id: {first_product_id}"
@ -133,18 +145,11 @@ async def handle_dirrect_message(
headers={
"X-Api-Key": wallet.adminkey,
},
json=order,
json=order.dict(),
)
resp.raise_for_status()
data = resp.json()
return (
json.dumps(data, separators=(",", ":"), ensure_ascii=False)
if data
else None
)
else:
print("### text_msg", text_msg)
return None
except Exception as ex:
logger.warning(ex)
if data:
return json.dumps(data, separators=(",", ":"), ensure_ascii=False)
return None