wind_speed.py

import random
import time
import csv
import os
from datetime import datetime

# Weilong & Pradosh main contribution

def simulate_wind_speed():
    wind_speed = random.uniform(0, 10)
    return wind_speed

def create_directory_if_not_exists(directory):
    if not os.path.exists(directory):
        try:
            os.makedirs(directory)
        except FileExistsError:
            pass  # Directory already exists, ignore the error

def save_wind_data(directory, timestamp, wind_speed):
    # Determine the file name based on the elapsed time within the current hour
    elapsed_minutes = timestamp.minute
    file_name = f"wind_speed{15 * (elapsed_minutes // 15)}.csv"
    
    file_path = os.path.join(directory, file_name)
    with open(file_path, 'a', newline='') as csvfile:
        writer = csv.writer(csvfile)
        if os.stat(file_path).st_size == 0:
            writer.writerow(["Timestamp", "Wind Speed (m/s)"])
        writer.writerow([timestamp, wind_speed])

# Set the duration for the script to run (15 minutes)
duration = 15 * 60  # 15 minutes in seconds
start_time = time.time()

while True:
    # Check if 15 minutes have passed
    if time.time() - start_time > duration:
        break

    # Get the current timestamp
    current_timestamp = datetime.now()

    # Create directories for each day and hour
    day_directory = current_timestamp.strftime("%Y-%m-%d")
    hour_directory = current_timestamp.strftime("%H")
    directory_to_write = os.path.join("/users/pgrad/singhr6/group24/data", day_directory, hour_directory)
    create_directory_if_not_exists("/users/pgrad/singhr6/group24/data")
    create_directory_if_not_exists(os.path.join("/users/pgrad/singhr6/group24/data", day_directory))
    create_directory_if_not_exists(directory_to_write)

    # Generate simulated wind speed reading
    wind_speed = simulate_wind_speed()

    # Save the simulated wind speed reading with timestamp to the CSV file
    save_wind_data(directory_to_write, current_timestamp, wind_speed)

    # Print the simulated wind speed reading with timestamp (optional)
    print(f"{current_timestamp} - Simulated Wind Speed: {wind_speed} m/s")

    # Wait for 30 seconds before generating the next reading
    time.sleep(30)

# Print a message indicating the end of the script
print("Script completed after 15 minutes.")