This commit is contained in:
20
app/app.py
20
app/app.py
@ -4,24 +4,26 @@
|
|||||||
|
|
||||||
from flask import Flask, render_template, request
|
from flask import Flask, render_template, request
|
||||||
from flask_wtf.csrf import CSRFProtect
|
from flask_wtf.csrf import CSRFProtect
|
||||||
from app.send_page import send_page
|
|
||||||
|
|
||||||
|
from app.send_page import send_page
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
csrf = CSRFProtect(app)
|
csrf = CSRFProtect(app)
|
||||||
|
|
||||||
@app.route("/", methods=['GET'])
|
|
||||||
|
@app.route("/", methods=["GET"])
|
||||||
def index():
|
def index():
|
||||||
"""Returns index template."""
|
"""Returns index template."""
|
||||||
return render_template('index.html', status='')
|
return render_template("index.html", status="")
|
||||||
|
|
||||||
@app.route("/", methods=['POST'])
|
|
||||||
|
@app.route("/", methods=["POST"])
|
||||||
def send():
|
def send():
|
||||||
"""POST function"""
|
"""POST function"""
|
||||||
result = send_page(
|
result = send_page(
|
||||||
name=request.form.get('name'),
|
name=request.form.get("name"),
|
||||||
email=request.form.get('email'),
|
email=request.form.get("email"),
|
||||||
message=request.form.get('message')
|
message=request.form.get("message"),
|
||||||
)
|
)
|
||||||
status = 'success' if result[0] else 'fail'
|
status = "success" if result[0] else "fail"
|
||||||
return render_template('index.html', status=status)
|
return render_template("index.html", status=status)
|
||||||
|
@ -3,33 +3,31 @@
|
|||||||
"""Send messages via the Pushover API."""
|
"""Send messages via the Pushover API."""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
|
|
||||||
def send_page(name: str, email: str, message: str) -> tuple:
|
def send_page(name: str, email: str, message: str) -> tuple:
|
||||||
"""POST to the Pushover API."""
|
"""POST to the Pushover API."""
|
||||||
api_url = "https://api.pushover.net/1/messages.json"
|
api_url = "https://api.pushover.net/1/messages.json"
|
||||||
api_token = os.getenv('PUSHOVER_API_TOKEN')
|
api_token = os.getenv("PUSHOVER_API_TOKEN")
|
||||||
user_key = os.getenv('PUSHOVER_USER_KEY')
|
user_key = os.getenv("PUSHOVER_USER_KEY")
|
||||||
|
|
||||||
full_msg = f"Name: {name}\nEmail: {email}\n\nMessage: {message}"
|
full_msg = f"Name: {name}\nEmail: {email}\n\nMessage: {message}"
|
||||||
|
|
||||||
payload = {
|
payload = {
|
||||||
'token': api_token,
|
"token": api_token,
|
||||||
'user': user_key,
|
"user": user_key,
|
||||||
'title': f"ePage from {name}",
|
"title": f"ePage from {name}",
|
||||||
'message': full_msg,
|
"message": full_msg,
|
||||||
'priority': "1",
|
"priority": "1",
|
||||||
'sound': 'cosmic'
|
"sound": "cosmic",
|
||||||
}
|
}
|
||||||
|
|
||||||
req = requests.post(
|
req = requests.post(
|
||||||
api_url,
|
api_url, json=payload, headers={"Content-Type": "application/json"}, timeout=5
|
||||||
json=payload,
|
|
||||||
headers={'Content-Type': 'application/json'},
|
|
||||||
timeout=5
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if req.status_code == 200 and req.json().get('status') == 1:
|
if req.status_code == 200 and req.json().get("status") == 1:
|
||||||
return (True, None)
|
return (True, None)
|
||||||
return (False, req.json())
|
return (False, req.json())
|
||||||
|
@ -3,6 +3,7 @@
|
|||||||
"""PyTest unit tests."""
|
"""PyTest unit tests."""
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from app.app import app, csrf
|
from app.app import app, csrf
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
"""Tests for app/app.py"""
|
"""Tests for app/app.py"""
|
||||||
|
|
||||||
|
|
||||||
from tests import client # pragma: no cover
|
from tests import client # pragma: no cover
|
||||||
|
|
||||||
|
|
||||||
def test_index(client) -> None:
|
def test_index(client) -> None:
|
||||||
"""Ensure that the index page is loaded correctly."""
|
"""Ensure that the index page is loaded correctly."""
|
||||||
req = client.get('/')
|
req = client.get("/")
|
||||||
assert req.status_code == 200 and "ePage" in req.text
|
assert req.status_code == 200 and "ePage" in req.text
|
||||||
|
@ -7,9 +7,5 @@ from app.send_page import send_page
|
|||||||
|
|
||||||
def test_send_page_no_env() -> None:
|
def test_send_page_no_env() -> None:
|
||||||
"""Ensure the API returns an error if no API key specified."""
|
"""Ensure the API returns an error if no API key specified."""
|
||||||
result = send_page(
|
result = send_page(name="Unit Test", email="none@none.com", message="Unit Test")
|
||||||
name='Unit Test',
|
assert not result[0] and result[1].get("token") == "invalid"
|
||||||
email='none@none.com',
|
|
||||||
message='Unit Test'
|
|
||||||
)
|
|
||||||
assert not result[0] and result[1].get('token') == 'invalid'
|
|
||||||
|
Reference in New Issue
Block a user