"""Stream Mac addresses"""
import re
import asyncio
import subprocess
from fastapi import FastAPI, Response
from fastapi.staticfiles import StaticFiles
from fastapi.responses import StreamingResponse, PlainTextResponse
app = FastAPI()
from fastapi.middleware.cors import CORSMiddleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
async def stream_mac_addresses():
"""Stream Mac Addresses"""
pipe = await asyncio.create_subprocess_shell('tcpdump -i en0 -n -e',
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
mac_regex = re.compile(r"(([0-9a-fA-F]:?){12})")
while True:
line = await pipe.stdout.readline()
line = line.decode('utf-8').strip()
macs = mac_regex.findall(line)
if macs:
yield f'data: {macs[0][0]}\n\n'
# New - mount static files
app.mount("/static", StaticFiles(directory="."), name="static")
@app.get("/hello")
def read_root():
"""Hello world"""
return {"Hello": "World"}
@app.get("/stream", response_class=PlainTextResponse)
async def read_stream():
return StreamingResponse(stream_mac_addresses())
