app.py

import asyncio
import websockets
import json
import os
from dotenv import load_dotenv
from flask import Flask, render_template
from flask_sockets import Sockets
import threading
from datetime import datetime, timezone

# Carregar variáveis de ambiente do arquivo .env
load_dotenv()

# Obter a chave da API da variável de ambiente
API_KEY = os.getenv("API_KEY")
if not API_KEY:
    raise ValueError("API_KEY não definida. Verifique o arquivo .env.")


async def connect_ais_stream():

    async with websockets.connect("wss://stream.aisstream.io/v0/stream") as websocket:
        subscribe_message = {"APIKey": API_KEY,  # Required !
                             "BoundingBoxes": [[[-90, -180], [90, 180]]], # Required!
                             #"FiltersShipMMSI": ["368207620", "367719770", "211476060"], # Optional!
                             "FilterMessageTypes": ["PositionReport"]} # Optional!

        subscribe_message_json = json.dumps(subscribe_message)
        await websocket.send(subscribe_message_json)

        async for message_json in websocket:
            message = json.loads(message_json)
            message_type = message["MessageType"]

            if message_type == "PositionReport":
                # the message parameter contains a key of the message type which contains the message itself
                ais_message = message['Message']['PositionReport']
                metaData = message['MetaData'] 
                print(f"[{datetime.now(timezone.utc)}] ShipId: {ais_message['UserID']} Latitude: {ais_message['Latitude']} Longitude: {ais_message['Longitude']} ShipName: {metaData['ShipName']} ")
        print("Encerrado")
        print("")
if __name__ == "__main__":
    asyncio.run(asyncio.run(connect_ais_stream()))