simplehandler.py

from aiocache import Cache
from fastapi import FastAPI
from pydantic import BaseModel
from fastapi import Response
import hashlib
import segment.analytics as analytics

analytics.write_key = 'REDACTED'


class EventItem(BaseModel):
  userId: str
  eventName: str

class EventCache(BaseModel):
  hasCompiled: bool


# @klotho::persist {
#   id = "seenCache"
# }
eventCache = Cache(Cache.MEMORY)

# @klotho::expose {
#   id = "signalapi"
#   target = "public"
# }
app = FastAPI()

@app.post("/event")
async def handleSignalEvent(eventItem: EventItem, response: Response):
    userHash = hashlib.md5(eventItem.userId.encode()).hexdigest()
    userEvent = EventCache(hasCompiled=True)
    userEvent.hasCompiled = True 
    eventsFound = await eventCache.get(userHash)
    if eventsFound is None:
        analytics.track(eventItem.userId, 'FIRST_COMPILE')
        await eventCache.set(userHash, userEvent)
        return {"message": "FIRST_COMPILE"}
    else:
        return {"message": "ALREADY_COMPILED"}