Browse Source

Got gunicorn working. Problem with sqlite intiailization to fix.

pull/1/head
Matthew Raymer 1 year ago
parent
commit
ebebd2fce2
  1. 2
      Dockerfile
  2. 158
      app.py
  3. 3
      requirements.txt

2
Dockerfile

@ -38,4 +38,4 @@ COPY --from=builder /usr/local /usr/local
USER myuser
# Start gunicorn with the appropriate options
CMD ["gunicorn", "-b", "0.0.0.0:5000", "--workers=3", "app:app"]
CMD ["gunicorn", "-b", "0.0.0.0:3000", "--log-level=debug", "--workers=3", "app:create_app('default')"]

158
app.py

@ -6,90 +6,90 @@ from pywebpush import webpush, WebPushException
import json
import os
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///webpush.db'
db.init_app(app)
def generate_and_save_vapid_keys():
vapid = Vapid()
vapid.generate_keys()
private_key = vapid.get_private_key().to_pem().decode('utf-8').strip()
public_key = vapid.get_public_key().to_pem().decode('utf-8').strip()
key = VAPIDKey(public_key=public_key, private_key=private_key)
db.session.add(key)
db.session.commit()
@app.before_first_request
def initialize():
if not VAPIDKey.query.first():
generate_and_save_vapid_keys()
def send_push_notification(subscription_info, message, vapid_key):
try:
webpush(
subscription_info=subscription_info,
data=json.dumps(message),
vapid_private_key=vapid_key.private_key,
vapid_claims={
"sub": "mailto:your-email@example.com"
}
)
except WebPushException as e:
print(f"Failed to send notification: {e}")
@app.route('/web-push/vapid', methods=['GET'])
def get_vapid():
key = VAPIDKey.query.first()
if key:
return jsonify(public_key=key.public_key)
return jsonify(error='No VAPID keys found'), 404
@app.route('/web-push/subscribe', methods=['POST'])
def subscribe():
content = request.json
vapid_key = VAPIDKey.query.first()
if not vapid_key:
return jsonify(success=False, error="No VAPID keys available"), 500
subscription = Subscription(endpoint=content['endpoint'],
p256dh=content['keys']['p256dh'],
auth=content['keys']['auth'],
vapid_key_id=vapid_key.id)
db.session.add(subscription)
db.session.commit()
subscription_info = {
"endpoint": subscription.endpoint,
"keys": {
"p256dh": subscription.p256dh,
"auth": subscription.auth
}
}
def create_app(config_name):
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///webpush.db'
db.init_app(app)
def generate_and_save_vapid_keys():
vapid = Vapid()
vapid.generate_keys()
private_key = vapid.get_private_key().to_pem().decode('utf-8').strip()
public_key = vapid.get_public_key().to_pem().decode('utf-8').strip()
key = VAPIDKey(public_key=public_key, private_key=private_key)
db.session.add(key)
db.session.commit()
def initialize():
if not VAPIDKey.query.first():
generate_and_save_vapid_keys()
message = {"title": "Subscription Successful", "body": "Thank you for subscribing!"}
send_push_notification(subscription_info, message, vapid_key)
def send_push_notification(subscription_info, message, vapid_key):
try:
webpush(
subscription_info=subscription_info,
data=json.dumps(message),
vapid_private_key=vapid_key.private_key,
vapid_claims={
"sub": "mailto:your-email@example.com"
}
)
except WebPushException as e:
print(f"Failed to send notification: {e}")
return jsonify(success=True)
@app.route('/web-push/vapid', methods=['GET'])
def get_vapid():
key = VAPIDKey.query.first()
if key:
return jsonify(public_key=key.public_key)
return jsonify(error='No VAPID keys found'), 404
@app.route('/web-push/unsubscribe', methods=['DELETE'])
def unsubscribe():
content = request.json
endpoint = content['endpoint']
subscription = Subscription.query.filter_by(endpoint=endpoint).first()
if subscription:
db.session.delete(subscription)
@app.route('/web-push/subscribe', methods=['POST'])
def subscribe():
content = request.json
vapid_key = VAPIDKey.query.first()
if not vapid_key:
return jsonify(success=False, error="No VAPID keys available"), 500
subscription = Subscription(endpoint=content['endpoint'],
p256dh=content['keys']['p256dh'],
auth=content['keys']['auth'],
vapid_key_id=vapid_key.id)
db.session.add(subscription)
db.session.commit()
return jsonify(success=True, message="Subscription deleted successfully")
else:
return jsonify(success=False, error="Subscription not found"), 404
subscription_info = {
"endpoint": subscription.endpoint,
"keys": {
"p256dh": subscription.p256dh,
"auth": subscription.auth
}
}
message = {"title": "Subscription Successful", "body": "Thank you for subscribing!"}
send_push_notification(subscription_info, message, vapid_key)
return jsonify(success=True)
@app.route('/web-push/unsubscribe', methods=['DELETE'])
def unsubscribe():
content = request.json
endpoint = content['endpoint']
subscription = Subscription.query.filter_by(endpoint=endpoint).first()
if subscription:
db.session.delete(subscription)
db.session.commit()
return jsonify(success=True, message="Subscription deleted successfully")
else:
return jsonify(success=False, error="Subscription not found"), 404
if __name__ == '__main__':
app.run(debug=True)
return app

3
requirements.txt

@ -1,4 +1,5 @@
flask
flask>=2.0.0
flask_sqlalchemy
py_vapid
pywebpush
gunicorn

Loading…
Cancel
Save