changed name to myextension

This commit is contained in:
benarc 2024-01-16 16:38:53 +00:00
parent ba7e11e30d
commit 0711f583d6
16 changed files with 221 additions and 225 deletions

View file

@ -19,15 +19,15 @@ from lnbits.decorators import (
require_invoice_key,
)
from . import temp_ext
from . import myextension_ext
from .crud import (
create_temp,
update_temp,
delete_temp,
get_temp,
get_temps
create_myextension,
update_myextension,
delete_myextension,
get_myextension,
get_myextensions
)
from .models import CreateTempData
from .models import CreateMyExtensionData
#######################################
@ -36,65 +36,65 @@ from .models import CreateTempData
## Get all the records belonging to the user
@temp_ext.get("/api/v1/temps", status_code=HTTPStatus.OK)
async def api_temps(
@myextension_ext.get("/api/v1/temps", status_code=HTTPStatus.OK)
async def api_myextensions(
req: Request, all_wallets: bool = Query(False), wallet: WalletTypeInfo = Depends(get_key_type)
):
wallet_ids = [wallet.wallet.id]
if all_wallets:
user = await get_user(wallet.wallet.user)
wallet_ids = user.wallet_ids if user else []
return [temp.dict() for temp in await get_temps(wallet_ids, req)]
return [myextension.dict() for myextension in await get_myextensions(wallet_ids, req)]
## Get a specific record belonging to a user
@temp_ext.put("/api/v1/temps/{temp_id}")
async def api_temp_update(
data: CreateTempData,
temp_id: str,
@myextension_ext.put("/api/v1/temps/{myextension_id}")
async def api_myextension_update(
data: CreateMyExtensionData,
myextension_id: str,
wallet: WalletTypeInfo = Depends(require_admin_key),
):
if not temp_id:
if not myextension_id:
raise HTTPException(
status_code=HTTPStatus.NOT_FOUND, detail="Temp does not exist."
status_code=HTTPStatus.NOT_FOUND, detail="MyExtension does not exist."
)
temp = await get_temp(temp_id)
assert temp, "Temp couldn't be retrieved"
myextension = await get_myextension(myextension_id)
assert myextension, "MyExtension couldn't be retrieved"
if wallet.wallet.id != temp.wallet:
raise HTTPException(status_code=HTTPStatus.FORBIDDEN, detail="Not your Temp.")
temp = await update_temp(temp_id=temp_id, **data.dict())
return temp.dict()
if wallet.wallet.id != myextension.wallet:
raise HTTPException(status_code=HTTPStatus.FORBIDDEN, detail="Not your MyExtension.")
myextension = await update_myextension(myextension_id=myextension_id, **data.dict())
return myextension.dict()
## Create a new record
@temp_ext.post("/api/v1/temps", status_code=HTTPStatus.CREATED)
async def api_temp_create(
data: CreateTempData, wallet: WalletTypeInfo = Depends(get_key_type)
@myextension_ext.post("/api/v1/temps", status_code=HTTPStatus.CREATED)
async def api_myextension_create(
data: CreateMyExtensionData, wallet: WalletTypeInfo = Depends(get_key_type)
):
temp = await create_temp(wallet_id=wallet.wallet.id, data=data)
return temp.dict()
myextension = await create_myextension(wallet_id=wallet.wallet.id, data=data)
return myextension.dict()
## Delete a record
@temp_ext.delete("/api/v1/temps/{temp_id}")
async def api_temp_delete(
temp_id: str, wallet: WalletTypeInfo = Depends(require_admin_key)
@myextension_ext.delete("/api/v1/temps/{myextension_id}")
async def api_myextension_delete(
myextension_id: str, wallet: WalletTypeInfo = Depends(require_admin_key)
):
temp = await get_temp(temp_id)
myextension = await get_myextension(myextension_id)
if not temp:
if not myextension:
raise HTTPException(
status_code=HTTPStatus.NOT_FOUND, detail="Temp does not exist."
status_code=HTTPStatus.NOT_FOUND, detail="MyExtension does not exist."
)
if temp.wallet != wallet.wallet.id:
raise HTTPException(status_code=HTTPStatus.FORBIDDEN, detail="Not your Temp.")
if myextension.wallet != wallet.wallet.id:
raise HTTPException(status_code=HTTPStatus.FORBIDDEN, detail="Not your MyExtension.")
await delete_temp(temp_id)
await delete_myextension(myextension_id)
return "", HTTPStatus.NO_CONTENT
@ -102,26 +102,26 @@ async def api_temp_delete(
## This endpoint creates a payment
@temp_ext.post("/api/v1/temps/payment/{temp_id}", status_code=HTTPStatus.CREATED)
@myextension_ext.post("/api/v1/temps/payment/{myextension_id}", status_code=HTTPStatus.CREATED)
async def api_tpos_create_invoice(
temp_id: str, amount: int = Query(..., ge=1), memo: str = ""
myextension_id: str, amount: int = Query(..., ge=1), memo: str = ""
) -> dict:
temp = await get_temp(temp_id)
myextension = await get_myextension(myextension_id)
if not temp:
if not myextension:
raise HTTPException(
status_code=HTTPStatus.NOT_FOUND, detail="Temp does not exist."
status_code=HTTPStatus.NOT_FOUND, detail="MyExtension does not exist."
)
# we create a payment and add some tags, so tasks.py can grab the payment once its paid
try:
payment_hash, payment_request = await create_invoice(
wallet_id=temp.wallet,
wallet_id=myextension.wallet,
amount=amount,
memo=f"{memo} to {temp.name}" if memo else f"{temp.name}",
memo=f"{memo} to {myextension.name}" if memo else f"{myextension.name}",
extra={
"tag": "temp",
"tag": "myextension",
"tipAmount": tipAmount,
"tempId": tempId,
"amount": amount,