feat: automatically import extension blueprints

This commit is contained in:
Eneko Illarramendi 2020-02-19 20:52:34 +01:00
parent 7212bad783
commit 30cf2e9f99
4 changed files with 133 additions and 106 deletions

View File

@ -6,6 +6,10 @@ end_of_line = lf
insert_final_newline = true
trim_trailing_whitespace = true
[*.{html,js,json}]
indent_size = 2
indent_style = space
[*.md]
trim_trailing_whitespace = false

View File

@ -1,3 +1,4 @@
import importlib
import json
import requests
import uuid
@ -9,12 +10,17 @@ from lnurl import Lnurl, LnurlWithdrawResponse
from . import bolt11
from .core import core_app
from .db import init_databases, open_db
from .extensions.withdraw import withdraw_ext
from .helpers import ExtensionManager, megajson
from .settings import WALLET, DEFAULT_USER_WALLET_NAME, FEE_RESERVE
app = Flask(__name__)
valid_extensions = [ext for ext in ExtensionManager().extensions if ext.is_valid]
# optimization & security
# -----------------------
Talisman(
app,
content_security_policy={
@ -32,20 +38,41 @@ Talisman(
},
)
# blueprints / extensions
# -----------------------
app.register_blueprint(core_app)
for ext in valid_extensions:
try:
ext_module = importlib.import_module(f"lnbits.extensions.{ext.code}")
app.register_blueprint(getattr(ext_module, f"{ext.code}_ext"), url_prefix=f"/{ext.code}")
except Exception:
raise ImportError(f"Please make sure that the extension `{ext.code}` follows conventions.")
# filters
app.jinja_env.globals["EXTENSIONS"] = [ext for ext in ExtensionManager().extensions if ext.is_valid]
# -------
app.jinja_env.globals["EXTENSIONS"] = valid_extensions
app.jinja_env.filters["megajson"] = megajson
# blueprints
app.register_blueprint(core_app)
app.register_blueprint(withdraw_ext, url_prefix="/withdraw")
# init
# ----
@app.before_first_request
def init():
init_databases()
# vvvvvvvvvvvvvvvvvvvvvvvvvvv
# move the rest to `core_app`
# vvvvvvvvvvvvvvvvvvvvvvvvvvv
# vvvvvvvvvvvvvvvvvvvvvvvvvvv
@app.route("/deletewallet")
def deletewallet():
user_id = request.args.get("usr")

View File

@ -1,5 +1,5 @@
{
"name": "LNURLw",
"short_description": "Make LNURL withdraw links.",
"ion_icon": "beer"
"name": "LNURLw",
"short_description": "Make LNURL withdraw links.",
"ion_icon": "beer"
}

View File

@ -4,115 +4,111 @@
{% block messages %}
<a href="#" class="dropdown-toggle" data-toggle="dropdown">
<i class="fa fa-bell-o"></i>
<span class="label label-danger">!</span>
</a>
<ul class="dropdown-menu">
<li class="header"><b>Instant wallet, bookmark to save</b></li>
<li></li>
</ul>
<a href="#" class="dropdown-toggle" data-toggle="dropdown">
<i class="fa fa-bell-o"></i>
<span class="label label-danger">!</span>
</a>
<ul class="dropdown-menu">
<li class="header"><b>Instant wallet, bookmark to save</b></li>
<li></li>
</ul>
{% endblock %}
{% block menuitems %}
<li class="treeview">
<a href="#">
<i class="fa fa-bitcoin"></i> <span>Wallets</span>
<i class="fa fa-angle-left pull-right"></i>
</a>
<ul class="treeview-menu">
{% for w in user_wallets %}
<li>
<a href="{{ url_for('wallet') }}?wal={{ w.id }}&usr={{ w.user }}"
><i class="fa fa-bolt"></i> {{ w.name }}</a
>
</li>
{% endfor %}
<li><a onclick="sidebarmake()">Add a wallet +</a></li>
<div id="sidebarmake"></div>
</ul>
</li>
<li class="active treeview">
<a href="#">
<i class="fa fa-th"></i> <span>Extensions</span>
<i class="fa fa-angle-left pull-right"></i>
</a>
<ul class="treeview-menu">
{% for extension in EXTENSIONS %}
{% if extension.code in user_ext %}
<li>
<a href="{{ url_for(extension.code + '.index') }}?usr={{ user }}"><i class="fa fa-plus"></i> {{ extension.name }}</a>
</li>
{% endif %}
{% endfor %}
<li>
<a href="{{ url_for('extensions') }}?usr={{ user }}">Manager </a></li>
</ul>
</li>
<li class="treeview">
<a href="#">
<i class="fa fa-bitcoin"></i> <span>Wallets</span>
<i class="fa fa-angle-left pull-right"></i>
</a>
<ul class="treeview-menu">
{% for w in user_wallets %}
<li>
<a href="{{ url_for('wallet') }}?wal={{ w.id }}&usr={{ w.user }}"><i class="fa fa-bolt"></i> {{ w.name }}</a>
</li>
{% endfor %}
<li><a onclick="sidebarmake()">Add a wallet +</a></li>
<div id="sidebarmake"></div>
</ul>
</li>
<li class="active treeview">
<a href="#">
<i class="fa fa-th"></i> <span>Extensions</span>
<i class="fa fa-angle-left pull-right"></i>
</a>
<ul class="treeview-menu">
{% for extension in EXTENSIONS %}
{% if extension.code in user_ext %}
<li>
<a href="{{ url_for(extension.code + '.index') }}?usr={{ user }}"><i class="fa fa-plus"></i> {{ extension.name }}</a>
</li>
{% endif %}
{% endfor %}
<li>
<a href="{{ url_for('extensions') }}?usr={{ user }}">Manager</a></li>
</ul>
</li>
{% endblock %}
{% block body %}
<!-- Right side column. Contains the navbar and content of the page -->
<div class="content-wrapper">
<!-- Right side column. Contains the navbar and content of the page -->
<div class="content-wrapper">
<!-- Content Header (Page header) -->
<section class="content-header">
<div id="wonga"></div>
<h1>Wallet <small>Control panel</small></h1>
<ol class="breadcrumb">
<li>
<a href="#"><i class="fa fa-dashboard"></i> Home</a>
</li>
<li class="active">Extensions</li>
</ol>
<!-- Content Header (Page header) -->
<section class="content-header">
<div id="wonga"></div>
<h1>Wallet <small>Control panel</small></h1>
<ol class="breadcrumb">
<li><a href="#"><i class="fa fa-dashboard"></i> Home</a></li>
<li class="active">Extensions</li>
</ol>
<br />
<br />
<br />
<br />
<div class="alert alert-danger alert-dismissable">
<h4>Bookmark to save your wallet. Wallet is in BETA, use with caution.</h4>
</div>
</section>
<div class="alert alert-danger alert-dismissable">
<h4>Bookmark to save your wallet. Wallet is in BETA, use with caution.</h4>
</div>
</section>
<!-- Main content -->
<section class="content">
<!-- Small boxes (Stat box) -->
<div class="row">
<!-- Main content -->
<section class="content">
<!-- Small boxes (Stat box) -->
<div class="row">
{% for extension in EXTENSIONS %}
<div class="col-lg-3 col-xs-6">
<!-- small box -->
<div class="small-box bg-blue">
<div class="inner">
{% if extension.code in user_ext %}
<a href="{{ url_for(extension.code + '.index') }}?usr={{ user }}" style="color: inherit">
{% endif %}
<h3>{{ extension.name }}</h3>
<p>{{ extension.short_description }}</p>
{% if extension.code in user_ext %}
</a>
{% endif %}
</div>
<div class="icon">
<i class="ion ion-{{ extension.ion_icon }}"></i>
</div>
{% if extension.code in user_ext %}
<a href="{{ url_for('extensions') }}?usr={{user}}&disable={{ extension.code }}" class="small-box-footer">Disable <i class="fa fa-arrow-circle-right"></i></a>
{% else %}
<a href="{{ url_for('extensions') }}?usr={{user}}&enable={{ extension.code }}" class="small-box-footer">Enable <i class="fa fa-arrow-circle-right"></i></a>
{% endif %}
</div>
</div>
{% endfor %}
{% for extension in EXTENSIONS %}
<div class="col-lg-3 col-xs-6">
<!-- small box -->
<div class="small-box bg-blue">
<div class="inner">
{% if extension.code in user_ext %}
<a href="{{ url_for(extension.code + '.index') }}?usr={{ user }}" style="color: inherit">
{% endif %}
<h3>{{ extension.name }}</h3>
<p>{{ extension.short_description }}</p>
{% if extension.code in user_ext %}
</a>
{% endif %}
</div>
<div class="icon">
<i class="ion ion-{{ extension.ion_icon }}"></i>
</div>
{% if extension.code in user_ext %}
<a href="{{ url_for('extensions') }}?usr={{user}}&disable={{ extension.code }}" class="small-box-footer">Disable <i class="fa fa-arrow-circle-right"></i></a>
{% else %}
<a href="{{ url_for('extensions') }}?usr={{user}}&enable={{ extension.code }}" class="small-box-footer">Enable <i class="fa fa-arrow-circle-right"></i></a>
{% endif %}
</div>
</div>
{% endfor %}
</div>
<!-- /.content -->
</section>
</div>
</div>
<!-- /.content -->
</section>
</div>
<script>
window.user = {{ user | megajson | safe }}
window.user_wallets = {{ user_wallets | megajson | safe }}
window.user_ext = {{ user_ext | megajson | safe }}
</script>
<script>
window.user = {{ user | megajson | safe }}
window.user_wallets = {{ user_wallets | megajson | safe }}
window.user_ext = {{ user_ext | megajson | safe }}
</script>
{% endblock %}