traffic_sensor.py

import random
import time
from datetime import datetime



def generate_traffic_flow(anomaly=False):

    now = datetime.now()
    hour = now.hour

    
    if 7 <= hour <= 9:  
        cars_range = (60, 90)
        speed_range = (20, 35)

    elif 17 <= hour <= 19:  
        cars_range = (70, 100)
        speed_range = (15, 30)

    elif 22 <= hour or hour <= 5:  
        cars_range = (5, 20)
        speed_range = (50, 70)

    else:  # normal daytime traffic
        cars_range = (20, 50)
        speed_range = (35, 50)

    # Anomaly (gridlock scenario)
    if anomaly:
        cars_per_minute = random.randint(100, 150)
        avg_speed_kmh = round(random.uniform(3, 10), 2)
    else:
        cars_per_minute = random.randint(*cars_range)
        avg_speed_kmh = round(random.uniform(*speed_range), 2)

    data = {
        "timestamp": now.isoformat(),
        "sensor_type": "traffic_flow",
        "cars_per_minute": cars_per_minute,
        "avg_speed_kmh": avg_speed_kmh
    }

    return data



def sensor_stream():

    while True:

        # 30% chance of anomaly
        anomaly = random.random() < 0.3

        traffic_flow = generate_traffic_flow(anomaly)

        print(traffic_flow)
        print("-" * 50)

        time.sleep(2)  # new sensor reading every 2 seconds



sensor_stream()