You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
import monero.address
|
|
from quart import Blueprint, render_template, request, flash, redirect
|
|
from quart_auth import login_required, current_user
|
|
|
|
from lws.models import Wallet, User
|
|
|
|
|
|
bp = Blueprint('wallet', 'wallet')
|
|
|
|
|
|
# @bp.route("/")
|
|
|
|
# accept_requests: {"type": "import"|"create", "addresses":[...]}
|
|
# add_account: {"address": ..., "key": ...}
|
|
# list_accounts: {}
|
|
# list_requests: {}
|
|
# modify_account_status: {"status": "active"|"hidden"|"inactive", "addresses":[...]}
|
|
# reject_requests: {"type": "import"|"create", "addresses":[...]}
|
|
# rescan: {"height":..., "addresses":[...]}
|
|
|
|
@bp.route("/wallet/add", methods=["GET", "POST"])
|
|
@login_required
|
|
async def add():
|
|
form = await request.form
|
|
if form:
|
|
name = form.get("name", "")
|
|
description = form.get("description", "")
|
|
address = form.get("address", "")
|
|
view_key = form.get("view_key", "")
|
|
restore_height = form.get("restore_height", 0)
|
|
valid_view_key = False
|
|
if not address:
|
|
await flash("must provide an address")
|
|
return redirect("/wallet/add")
|
|
if not view_key:
|
|
await flash("must provide a view_key")
|
|
return redirect("/wallet/add")
|
|
try:
|
|
_a = monero.address.Address(address)
|
|
valid_view_key = _a.check_private_view_key(view_key)
|
|
except ValueError:
|
|
await flash("Invalid Monero address")
|
|
return redirect("/wallet/add")
|
|
if not valid_view_key:
|
|
await flash("Invalid view key provided for address")
|
|
return redirect("/wallet/add")
|
|
wallet = Wallet.create(
|
|
name=name,
|
|
description=description,
|
|
address=address,
|
|
view_key=view_key,
|
|
restore_height=restore_height,
|
|
user=User.get(current_user.auth_id)
|
|
)
|
|
if not name:
|
|
wallet.name = f"wallet-{id}"
|
|
wallet.add_wallet_lws()
|
|
await flash("wallet added")
|
|
return redirect(f"/wallet/{wallet.id}/rescan")
|
|
return await render_template("wallet/add.html")
|
|
|
|
|
|
@bp.route("/wallet/<id>")
|
|
@login_required
|
|
async def show(id):
|
|
wallet = Wallet.select().where(Wallet.id == id).first()
|
|
if not wallet:
|
|
await flash("wallet does not exist")
|
|
return redirect("/")
|
|
return await render_template(
|
|
"wallet/show.html",
|
|
wallet=wallet
|
|
)
|
|
|
|
@bp.route("/wallet/<id>/rescan")
|
|
@login_required
|
|
async def rescan(id):
|
|
wallet = Wallet.select().where(Wallet.id == id).first()
|
|
if not wallet:
|
|
await flash("wallet does not exist")
|
|
return redirect("/")
|
|
wallet.rescan()
|
|
return redirect(f"/wallet/{id}")
|
|
|
|
|
|
@bp.route("/wallet/<id>/disable")
|
|
@login_required
|
|
async def disable(id):
|
|
wallet = Wallet.select().where(Wallet.id == id).first()
|
|
if not wallet:
|
|
await flash("wallet does not exist")
|
|
return redirect("/")
|
|
wallet.set_active(False)
|
|
return redirect(f"/wallet/{id}")
|
|
|
|
|
|
@bp.route("/wallet/<id>/enable")
|
|
@login_required
|
|
async def enable(id):
|
|
wallet = Wallet.select().where(Wallet.id == id).first()
|
|
if not wallet:
|
|
await flash("wallet does not exist")
|
|
return redirect("/")
|
|
wallet.set_active(True)
|
|
return redirect(f"/wallet/{id}")
|
|
|