From 5dc0e64bcdb856284c42cdf6812b4ee4ec210dd8 Mon Sep 17 00:00:00 2001
From: Trent Larson <trent@trentlarson.com>
Date: Mon, 11 Dec 2023 17:53:58 -0700
Subject: [PATCH] update test file for arguments

---
 webpush.py | 29 +++++++++++++++++++++++------
 1 file changed, 23 insertions(+), 6 deletions(-)

diff --git a/webpush.py b/webpush.py
index 92838c1..8f5a17a 100644
--- a/webpush.py
+++ b/webpush.py
@@ -1,9 +1,13 @@
 # Test subscription creation
+#
+# Setup with `pip install -r requirements.txt` and then `source venv/bin/activate`
 
 from pywebpush import webpush, WebPushException
 
-import json
 import base64
+import datetime
+import json
+import sys
 
 # these below will only work for the browser that acquired the subscription info.  You will need to extract those from our console OR the db.
 
@@ -12,9 +16,11 @@ import base64
 # sVR_s8J4JHv3h4ZmvemL5w
 
 
+
 private_key_hex = "308187020100301306072a8648ce3d020106082a8648ce3d030107046d306b0201010420ac868a9588a69ec9626db857caae42b0d654288abf73b0d8f1a6b43fff093508a1440342000466da453c1e793d7e21ab63b334e80f96715aa97e578639b5fe8092f8752fa6e1f9182324846c70bf1a3480411ba787e652be8049a36a14294681f745c4c4c4f7"
 
 private_key_der = bytes.fromhex(private_key_hex)
+
 private_key_base64 = base64.b64encode(private_key_der).decode()
 
 subscription_info = {
@@ -25,18 +31,29 @@ subscription_info = {
     }
 }
 
-data = json.dumps({"title": "test", "message": "here is a message"})
+# set subscription_info from arg 1 (sqlite results)
+if len(sys.argv) > 1:
+    argument = sys.argv[1]
+    parts = argument.split('|')
+    if len(parts) > 0:
+        subscription_info['endpoint'] = parts[0]
+        if len(parts) > 1:
+            subscription_info['keys']['p256dh'] = parts[1]
+            if len(parts) > 2:
+                subscription_info['keys']['auth'] = parts[2]
+
+now = datetime.datetime.now().isoformat()
+data = json.dumps({"title": "test", "message": f"Message at {now}"})
 
 try:
-    webpush(subscription_info,
+    result = webpush(subscription_info,
             data,
-#            vapid_private_key="./private_key.pem",
-#            vapid_private_key="MIGHAgEAMBMGByqGSM49AgEGCCqGSM49AwEHBG0wawIBAQQgrIaKlYimnslibbhXyq5CsNZUKIq/c7DY8aa0P/8JNQihRANCAARm2kU8Hnk9fiGrY7M06A+WcVqpfleGObX+gJL4dS+m4fkYIySEbHC/GjSAQRunh+ZSvoBJo2oUKUaB90XExMT3",
             vapid_private_key=private_key_base64,
             vapid_claims={"sub": "mailto:matthew.raymer@gmail.com"})
+    print(f"Result from remote service: {result}")
 except WebPushException as ex:
     print(f"An error occurred: {ex}")
     # Check if there is a response from the remote service.
     if ex.response:
         response_data = ex.response.json()
-        print(f"Response from remote service: {response_data}")
+        print(f"Error response from remote service: {response_data}")