aiolabs/lnbits
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Format with yapf

Browse source
This commit is contained in:
Fitti 2021-07-07 01:28:39 +02:00
parent 435787ad93
commit 7c1ff1ff37
4 changed files with 131 additions and 195 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lnbits/extensions/twitchalerts/__init__.py
View file

@ -3,13 +3,10 @@ from lnbits.db import Database
db = Database("ext_twitchalerts") db = Database("ext_twitchalerts")
twitchalerts_ext: Blueprint = Blueprint( twitchalerts_ext: Blueprint = Blueprint("twitchalerts",
"twitchalerts",
__name__, __name__,
static_folder="static", static_folder="static",
template_folder="templates" template_folder="templates")
)
from .views_api import * # noqa from .views_api import * # noqa
from .views import * # noqa from .views import * # noqa

98
lnbits/extensions/twitchalerts/crud.py
View file

@ -67,17 +67,7 @@ async def create_donation(
) )
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
""", """,
( (id, wallet, name, message, cur_code, sats, amount, service, posted),
id,
wallet,
name,
message,
cur_code,
sats,
amount,
service,
posted
),
) )
return await get_donation(id) return await get_donation(id)
@ -89,15 +79,11 @@ async def post_donation(donation_id: str) -> tuple:
""" """
donation = await get_donation(donation_id) donation = await get_donation(donation_id)
if not donation: if not donation:
return ( return (jsonify({"message":
jsonify({"message": "Donation not found!"}), "Donation not found!"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
if donation.posted: if donation.posted:
return ( return (jsonify({"message": "Donation has already been posted!"}),
jsonify({"message": "Donation has already been posted!"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
service = await get_service(donation.service) service = await get_service(donation.service)
if service.servicename == "Streamlabs": if service.servicename == "Streamlabs":
url = "https://streamlabs.com/api/v1.0/donations" url = "https://streamlabs.com/api/v1.0/donations"
@ -114,23 +100,14 @@ async def post_donation(donation_id: str) -> tuple:
print(response.json()) print(response.json())
status = [s for s in list(HTTPStatus) if s == response.status_code][0] status = [s for s in list(HTTPStatus) if s == response.status_code][0]
elif service.servicename == "StreamElements": elif service.servicename == "StreamElements":
return ( return (jsonify({"message": "StreamElements not yet supported!"}),
jsonify({"message": "StreamElements not yet supported!"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
else: else:
return ( return (jsonify({"message":
jsonify({"message": "Unsopported servicename"}), "Unsopported servicename"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST await db.execute("UPDATE Donations SET posted = 1 WHERE id = ?",
) (donation_id, ))
await db.execute( return (jsonify(response.json()), status)
"UPDATE Donations SET posted = 1 WHERE id = ?",
(donation_id,)
)
return (
jsonify(response.json()),
status
)
async def create_service( async def create_service(
@ -182,24 +159,18 @@ async def get_service(service_id: int,
streamer via typos like 2 -> 3. streamer via typos like 2 -> 3.
""" """
if by_state: if by_state:
row = await db.fetchone( row = await db.fetchone("SELECT * FROM Services WHERE state = ?",
"SELECT * FROM Services WHERE state = ?", (by_state, ))
(by_state,)
)
else: else:
row = await db.fetchone( row = await db.fetchone("SELECT * FROM Services WHERE id = ?",
"SELECT * FROM Services WHERE id = ?", (service_id, ))
(service_id,)
)
return Service.from_row(row) if row else None return Service.from_row(row) if row else None
async def get_services(wallet_id: str) -> Optional[list]: async def get_services(wallet_id: str) -> Optional[list]:
"""Return all services belonging assigned to the wallet_id""" """Return all services belonging assigned to the wallet_id"""
rows = await db.fetchall( rows = await db.fetchall("SELECT * FROM Services WHERE wallet = ?",
"SELECT * FROM Services WHERE wallet = ?", (wallet_id, ))
(wallet_id,)
)
return [Service.from_row(row) for row in rows] if rows else None return [Service.from_row(row) for row in rows] if rows else None
@ -237,49 +208,40 @@ async def service_add_token(service_id, token):
return False return False
await db.execute( await db.execute(
"UPDATE Services SET authenticated = 1, token = ? where id = ?", "UPDATE Services SET authenticated = 1, token = ? where id = ?",
(token, service_id,), (
token,
service_id,
),
) )
return True return True
async def delete_service(service_id: int) -> None: async def delete_service(service_id: int) -> None:
"""Delete a Service and all corresponding Donations""" """Delete a Service and all corresponding Donations"""
await db.execute( await db.execute("DELETE FROM Services WHERE id = ?", (service_id, ))
"DELETE FROM Services WHERE id = ?", rows = await db.fetchall("SELECT * FROM Donations WHERE service = ?",
(service_id,) (service_id, ))
)
rows = await db.fetchall(
"SELECT * FROM Donations WHERE service = ?",
(service_id,)
)
for row in rows: for row in rows:
await delete_donation(row["id"]) await delete_donation(row["id"])
async def get_donation(donation_id: str) -> Optional[Donation]: async def get_donation(donation_id: str) -> Optional[Donation]:
"""Return a Donation""" """Return a Donation"""
row = await db.fetchone( row = await db.fetchone("SELECT * FROM Donations WHERE id = ?",
"SELECT * FROM Donations WHERE id = ?", (donation_id, ))
(donation_id,)
)
return Donation.from_row(row) if row else None return Donation.from_row(row) if row else None
async def get_donations(wallet_id: str) -> Optional[list]: async def get_donations(wallet_id: str) -> Optional[list]:
"""Return all Donations assigned to wallet_id""" """Return all Donations assigned to wallet_id"""
rows = await db.fetchall( rows = await db.fetchall("SELECT * FROM Donations WHERE wallet = ?",
"SELECT * FROM Donations WHERE wallet = ?", (wallet_id, ))
(wallet_id,)
)
return [Donation.from_row(row) for row in rows] if rows else None return [Donation.from_row(row) for row in rows] if rows else None
async def delete_donation(donation_id: str) -> None: async def delete_donation(donation_id: str) -> None:
"""Delete a Donation and its corresponding statspay charge""" """Delete a Donation and its corresponding statspay charge"""
await db.execute( await db.execute("DELETE FROM Donations WHERE id = ?", (donation_id, ))
"DELETE FROM Donations WHERE id = ?",
(donation_id,)
)
await delete_charge(donation_id) await delete_charge(donation_id)

12
lnbits/extensions/twitchalerts/migrations.py
View file

@ -1,7 +1,6 @@
async def m001_initial(db): async def m001_initial(db):
await db.execute( await db.execute("""
"""
CREATE TABLE IF NOT EXISTS Services ( CREATE TABLE IF NOT EXISTS Services (
id INTEGER PRIMARY KEY AUTOINCREMENT, id INTEGER PRIMARY KEY AUTOINCREMENT,
state TEXT NOT NULL, state TEXT NOT NULL,
@ -14,11 +13,9 @@ async def m001_initial(db):
authenticated BOOLEAN NOT NULL, authenticated BOOLEAN NOT NULL,
token TEXT token TEXT
); );
""" """)
)
await db.execute( await db.execute("""
"""
CREATE TABLE IF NOT EXISTS Donations ( CREATE TABLE IF NOT EXISTS Donations (
id TEXT PRIMARY KEY, id TEXT PRIMARY KEY,
wallet TEXT NOT NULL, wallet TEXT NOT NULL,
@ -31,5 +28,4 @@ async def m001_initial(db):
posted BOOLEAN NOT NULL, posted BOOLEAN NOT NULL,
FOREIGN KEY(service) REFERENCES Services(id) FOREIGN KEY(service) REFERENCES Services(id)
); );
""" """)
)

183
lnbits/extensions/twitchalerts/views_api.py
View file

@ -6,22 +6,11 @@ from lnbits.core.crud import get_wallet, get_user
from lnbits.utils.exchange_rates import btc_price from lnbits.utils.exchange_rates import btc_price
from . import twitchalerts_ext from . import twitchalerts_ext
from .crud import ( from .crud import (get_charge_details, get_service_redirect_uri,
get_charge_details, create_donation, post_donation, get_donation, get_donations,
get_service_redirect_uri, delete_donation, create_service, get_service, get_services,
create_donation, authenticate_service, update_donation, update_service,
post_donation, delete_service)
get_donation,
get_donations,
delete_donation,
create_service,
get_service,
get_services,
authenticate_service,
update_donation,
update_service,
delete_service
)
from ..satspay.crud import create_charge, get_charge from ..satspay.crud import create_charge, get_charge
@ -29,14 +18,30 @@ from ..satspay.crud import create_charge, get_charge
@api_check_wallet_key("invoice") @api_check_wallet_key("invoice")
@api_validate_post_request( @api_validate_post_request(
schema={ schema={
"twitchuser": {"type": "string", "required": True}, "twitchuser": {
"client_id": {"type": "string", "required": True}, "type": "string",
"client_secret": {"type": "string", "required": True}, "required": True
"wallet": {"type": "string", "required": True}, },
"servicename": {"type": "string", "required": True}, "client_id": {
"onchain": {"type": "string"} "type": "string",
"required": True
},
"client_secret": {
"type": "string",
"required": True
},
"wallet": {
"type": "string",
"required": True
},
"servicename": {
"type": "string",
"required": True
},
"onchain": {
"type": "string"
} }
) })
async def api_create_service(): async def api_create_service():
"""Create a service, which holds data about how/where to post donations""" """Create a service, which holds data about how/where to post donations"""
service = await create_service(**g.data) service = await create_service(**g.data)
@ -64,15 +69,12 @@ async def api_get_access(service_id):
} }
endpoint_url = "https://streamlabs.com/api/v1.0/authorize/?" endpoint_url = "https://streamlabs.com/api/v1.0/authorize/?"
querystring = "&".join( querystring = "&".join(
[f"{key}={value}" for key, value in params.items()] [f"{key}={value}" for key, value in params.items()])
)
redirect_url = endpoint_url + querystring redirect_url = endpoint_url + querystring
return redirect(redirect_url) return redirect(redirect_url)
else: else:
return ( return (jsonify({"message":
jsonify({"message": "Service does not exist!"}), "Service does not exist!"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
@twitchalerts_ext.route("/api/v1/authenticate/<service_id>", methods=["GET"]) @twitchalerts_ext.route("/api/v1/authenticate/<service_id>", methods=["GET"])
@ -86,31 +88,36 @@ async def api_authenticate_service(service_id):
state = request.args.get('state') state = request.args.get('state')
service = await get_service(service_id) service = await get_service(service_id)
if service.state != state: if service.state != state:
return ( return (jsonify({"message":
jsonify({"message": "State doesn't match!"}), "State doesn't match!"}), HTTPStatus.BAD_Request)
HTTPStatus.BAD_Request
)
redirect_uri = request.scheme + "://" + request.headers["Host"] redirect_uri = request.scheme + "://" + request.headers["Host"]
redirect_uri += f"/twitchalerts/api/v1/authenticate/{service_id}" redirect_uri += f"/twitchalerts/api/v1/authenticate/{service_id}"
url, success = await authenticate_service(service_id, code, redirect_uri) url, success = await authenticate_service(service_id, code, redirect_uri)
if success: if success:
return redirect(url) return redirect(url)
else: else:
return ( return (jsonify({"message": "Service already authenticated!"}),
jsonify({"message": "Service already authenticated!"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
@twitchalerts_ext.route("/api/v1/donations", methods=["POST"]) @twitchalerts_ext.route("/api/v1/donations", methods=["POST"])
@api_validate_post_request( @api_validate_post_request(
schema={ schema={
"name": {"type": "string"}, "name": {
"sats": {"type": "integer", "required": True}, "type": "string"
"service": {"type": "integer", "required": True}, },
"message": {"type": "string"} "sats": {
"type": "integer",
"required": True
},
"service": {
"type": "integer",
"required": True
},
"message": {
"type": "string"
} }
) })
async def api_create_donation(): async def api_create_donation():
"""Take data from donation form and return satspay charge""" """Take data from donation form and return satspay charge"""
# Currency is hardcoded while frotnend is limited # Currency is hardcoded while frotnend is limited
@ -143,18 +150,16 @@ async def api_create_donation():
amount=amount, amount=amount,
service=g.data["service"], service=g.data["service"],
) )
return ( return (jsonify({"redirect_url": f"/satspay/{charge.id}"}), HTTPStatus.OK)
jsonify({"redirect_url": f"/satspay/{charge.id}"}),
HTTPStatus.OK
)
@twitchalerts_ext.route("/api/v1/postdonation", methods=["POST"]) @twitchalerts_ext.route("/api/v1/postdonation", methods=["POST"])
@api_validate_post_request( @api_validate_post_request(schema={
schema={ "id": {
"id": {"type": "string", "required": True}, "type": "string",
} "required": True
) },
})
async def api_post_donation(): async def api_post_donation():
"""Post a paid donation to Stremalabs/StreamElements. """Post a paid donation to Stremalabs/StreamElements.
@ -165,10 +170,8 @@ async def api_post_donation():
if charge and charge.paid: if charge and charge.paid:
return await post_donation(donation_id) return await post_donation(donation_id)
else: else:
return ( return (jsonify({"message":
jsonify({"message": "Not a paid charge!"}), "Not a paid charge!"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
@twitchalerts_ext.route("/api/v1/services", methods=["GET"]) @twitchalerts_ext.route("/api/v1/services", methods=["GET"])
@ -181,9 +184,8 @@ async def api_get_services():
new_services = await get_services(wallet_id) new_services = await get_services(wallet_id)
services += new_services if new_services else [] services += new_services if new_services else []
return ( return (
jsonify([ jsonify([service._asdict()
service._asdict() for service in services for service in services] if services else []),
] if services else []),
HTTPStatus.OK, HTTPStatus.OK,
) )
@ -200,9 +202,8 @@ async def api_get_donations():
new_donations = await get_donations(wallet_id) new_donations = await get_donations(wallet_id)
donations += new_donations if new_donations else [] donations += new_donations if new_donations else []
return ( return (
jsonify([ jsonify([donation._asdict()
donation._asdict() for donation in donations for donation in donations] if donations else []),
] if donations else []),
HTTPStatus.OK, HTTPStatus.OK,
) )
@ -215,23 +216,17 @@ async def api_update_donation(donation_id=None):
donation = await get_donation(donation_id) donation = await get_donation(donation_id)
if not donation: if not donation:
return ( return (jsonify({"message": "Donation does not exist."}),
jsonify({"message": "Donation does not exist."}), HTTPStatus.NOT_FOUND)
HTTPStatus.NOT_FOUND
)
if donation.wallet != g.wallet.id: if donation.wallet != g.wallet.id:
return ( return (jsonify({"message":
jsonify({"message": "Not your donation."}), "Not your donation."}), HTTPStatus.FORBIDDEN)
HTTPStatus.FORBIDDEN
)
donation = await update_donation(donation_id, **g.data) donation = await update_donation(donation_id, **g.data)
else: else:
return ( return (jsonify({"message":
jsonify({"message": "No donation ID specified"}), "No donation ID specified"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
return jsonify(donation._asdict()), HTTPStatus.CREATED return jsonify(donation._asdict()), HTTPStatus.CREATED
@ -243,23 +238,17 @@ async def api_update_service(service_id=None):
service = await get_service(service_id) service = await get_service(service_id)
if not service: if not service:
return ( return (jsonify({"message":
jsonify({"message": "Service does not exist."}), "Service does not exist."}), HTTPStatus.NOT_FOUND)
HTTPStatus.NOT_FOUND
)
if service.wallet != g.wallet.id: if service.wallet != g.wallet.id:
return ( return (jsonify({"message":
jsonify({"message": "Not your service."}), "Not your service."}), HTTPStatus.FORBIDDEN)
HTTPStatus.FORBIDDEN
)
service = await update_service(service_id, **g.data) service = await update_service(service_id, **g.data)
else: else:
return ( return (jsonify({"message":
jsonify({"message": "No service ID specified"}), "No service ID specified"}), HTTPStatus.BAD_REQUEST)
HTTPStatus.BAD_REQUEST
)
return jsonify(service._asdict()), HTTPStatus.CREATED return jsonify(service._asdict()), HTTPStatus.CREATED
@ -269,15 +258,11 @@ async def api_delete_donation(donation_id):
"""Delete the donation with the given donation_id""" """Delete the donation with the given donation_id"""
donation = await get_donation(donation_id) donation = await get_donation(donation_id)
if not donation: if not donation:
return ( return (jsonify({"message":
jsonify({"message": "No donation with this ID!"}), "No donation with this ID!"}), HTTPStatus.NOT_FOUND)
HTTPStatus.NOT_FOUND
)
if donation.wallet != g.wallet.id: if donation.wallet != g.wallet.id:
return ( return (jsonify({"message": "Not authorized to delete this donation!"
jsonify({"message": "Not authorized to delete this donation!"}), }), HTTPStatus.FORBIDDEN)
HTTPStatus.FORBIDDEN
)
await delete_donation(donation_id) await delete_donation(donation_id)
return "", HTTPStatus.NO_CONTENT return "", HTTPStatus.NO_CONTENT
@ -289,15 +274,11 @@ async def api_delete_service(service_id):
"""Delete the service with the given service_id""" """Delete the service with the given service_id"""
service = await get_service(service_id) service = await get_service(service_id)
if not service: if not service:
return ( return (jsonify({"message":
jsonify({"message": "No service with this ID!"}), "No service with this ID!"}), HTTPStatus.NOT_FOUND)
HTTPStatus.NOT_FOUND
)
if service.wallet != g.wallet.id: if service.wallet != g.wallet.id:
return ( return (jsonify({"message": "Not authorized to delete this service!"}),
jsonify({"message": "Not authorized to delete this service!"}), HTTPStatus.FORBIDDEN)
HTTPStatus.FORBIDDEN
)
await delete_service(service_id) await delete_service(service_id)
return "", HTTPStatus.NO_CONTENT return "", HTTPStatus.NO_CONTENT