now storing and retrieving last messageId

Author: tblanarik

function_app.py

@@ -1,5 +1,8 @@
 import azure.functions as func
 from azure.data.tables import TableServiceClient
+from azure.data.tables import UpdateMode
+import json
+import datetime
 import logging
 import requests
 import os
@@ -20,7 +23,18 @@ def spotbot(req: func.HttpRequest) -> func.HttpResponse:
         return func.HttpResponse("Invalid JSON", status_code=400)
 
     content = create_content(req_body)
-    return call_target(content)
+    callsign = req_body.get('callsign')
+
+    table = get_table()
+    entity = query_for_entity(table, callsign)
+    messageId = None
+    if is_entity_recent(entity):
+        messageId = entity.metadata['MessageId']
+    response = call_target(content, messageId)
+    messageId = extract_message_id(response)
+    upsert_entity(table, callsign, messageId)
+
+    return response
 
 def create_content(req_body):
     callsign = req_body.get('callsign', 'Unknown')
@@ -36,9 +50,13 @@ def create_content(req_body):
     content = {"content": f"{callsign} | {source} | freq: {frequency} | mode: {mode} | loc: {summitRef}{wwffRef} | {spot_deeplink}", "flags": 4}
     return content
 
-def call_target(content):
+def call_target(content, messageId=None):
     target_url = os.getenv('TARGET_URL')
-    response = requests.post(target_url, json=content)
+    verb = "POST"
+    if messageId is not None:
+        target_url = target_url + f"/messages/{messageId}"
+        verb = "PATCH"
+    response = requests.request(verb, url=target_url, params={"wait": "true"}, json=content)
     return func.HttpResponse(response.text, status_code=response.status_code)
 
 def create_spot_deeplink(source, callsign, wwffRef):
@@ -50,17 +68,35 @@ def create_spot_deeplink(source, callsign, wwffRef):
         case _:
             return ""
 
-def store_in_azure_table(callsign, messageId):
+def get_table():
     connection_string = os.getenv('AzureWebJobsStorage')
     table_name = os.getenv('TABLE_NAME')
     table_service_client = TableServiceClient.from_connection_string(conn_str=connection_string)
-    table_client = table_service_client.create_table(table_name=table_name)
-    
-    entities = table_client.query_entities(f"PartitionKey eq '{callsign}' and RowKey eq '{callsign}'")
-    
-    
-    my_entity = {
+    table_client = table_service_client.get_table_client(table_name=table_name)
+    return table_client
+
+def query_for_entity(table_client, callsign):
+    entities = [ent for ent in table_client.query_entities(f"PartitionKey eq '{callsign}' and RowKey eq '{callsign}'")]
+    if len(entities) > 0:
+        logging.info(f"Entity already exists for {callsign}")
+    return entities[0] if len(entities) > 0 else None
+
+def is_entity_recent(entity):
+    if entity is None:
+        return False
+    ent_time = entity.metadata['timestamp']
+    cur_time = datetime.datetime.now(datetime.timezone.utc)
+    two_hours_in_seconds = 60 * 60 * 2
+    return (cur_time - ent_time).total_seconds() < two_hours_in_seconds
+
+def upsert_entity(table_client, callsign, messageId):
+    entity = {
         u'PartitionKey': callsign,
         u'RowKey': callsign,
         u'MessageId': messageId
-    }
+    }
+    table_client.upsert_entity(mode=UpdateMode.REPLACE, entity=entity)
+
+def extract_message_id(response):
+    resp = json.loads(response.get_body())
+    return resp['id']