Auto-Garden/main.py

260 lines
10 KiB
Python

from machine import Pin
import time
import network
import json
import gc # ADD THIS - for garbage collection
# Initialize pins (LED light onboard)
led = Pin("LED", Pin.OUT)
led.low()
# Hard reset WiFi interface before connecting
print("Initializing WiFi...")
try:
wlan = network.WLAN(network.STA_IF)
wlan.deinit()
time.sleep(2)
print("WiFi interface reset complete")
except Exception as e:
print(f"WiFi reset warning: {e}")
# Import after WiFi reset
from scripts.networking import connect_wifi
from scripts.discord_webhook import send_discord_message
from scripts.monitors import TemperatureMonitor, WiFiMonitor, ACMonitor, HeaterMonitor, run_monitors
from scripts.temperature_sensor import TemperatureSensor
from scripts.air_conditioning import ACController
from scripts.heating import HeaterController
from scripts.web_server import TempWebServer
from scripts.scheduler import ScheduleMonitor # NEW: Import scheduler for time-based temp changes
# ===== START: Configuration Loading =====
# Load saved settings from config.json file on Pico
def load_config():
"""Load configuration from config.json file."""
try:
with open('config.json', 'r') as f:
config = json.load(f)
print("Loaded saved settings from config.json")
return config
except:
# If file doesn't exist or is corrupted, use defaults
print("No saved config found, using defaults")
return {
'ac_target': 77.0, # Default AC target temp
'ac_swing': 1.0, # Default AC tolerance (+/- degrees)
'heater_target': 80.0, # Default heater target temp
'heater_swing': 2.0, # Default heater tolerance (+/- degrees)
'schedules': [], # No schedules by default
'schedule_enabled': False # Schedules disabled by default
}
# Load configuration from file
config = load_config()
# ===== END: Configuration Loading =====
# ===== START: WiFi Connection =====
# Connect to WiFi using credentials from secrets.py
wifi = connect_wifi(led)
# Set static IP and print WiFi details
if wifi and wifi.isconnected():
# Configure static IP (easier to bookmark web interface)
static_ip = '192.168.86.43' # Change this to match your network
subnet = '255.255.255.0'
gateway = '192.168.86.1' # Usually your router IP
dns = '192.168.86.1' # Usually your router IP
# Apply static IP configuration
wifi.ifconfig((static_ip, subnet, gateway, dns))
time.sleep(1)
# Print WiFi details for debugging
ifconfig = wifi.ifconfig()
print("\n" + "="*50)
print("WiFi Connected Successfully!")
print("="*50)
print(f"IP Address: {ifconfig[0]}")
print(f"Subnet Mask: {ifconfig[1]}")
print(f"Gateway: {ifconfig[2]}")
print(f"DNS Server: {ifconfig[3]}")
print(f"Web Interface: http://{ifconfig[0]}")
print("="*50 + "\n")
# Send startup notification to Discord
send_discord_message(f"Pico W online at http://{ifconfig[0]}")
# ===== START: NTP Time Sync =====
# Sync time with internet time server (required for schedules to work correctly)
# Without this, the Pico's clock starts at 2021 on every reboot
try:
import ntptime
ntptime.settime() # Downloads current time from pool.ntp.org
print("Time synced with NTP server")
except Exception as e:
print("Failed to sync time: {}".format(e))
# ===== END: NTP Time Sync =====
else:
# WiFi connection failed
print("\n" + "="*50)
print("WiFi Connection Failed!")
print("="*50 + "\n")
# ===== END: WiFi Connection =====
# ===== START: Web Server Setup =====
# Start web server for monitoring and control (accessible at http://192.168.86.43)
web_server = TempWebServer(port=80)
web_server.start()
# ===== END: Web Server Setup =====
# ===== START: Sensor Configuration =====
# Define all temperature sensors and their alert thresholds
SENSOR_CONFIG = {
'inside': {
'pin': 10, # GPIO pin for DS18B20 sensor
'label': 'Inside', # Display name
'alert_high': 80.0, # Send alert if temp > 80°F
'alert_low': 70.0 # Send alert if temp < 70°F
},
'outside': {
'pin': 11, # GPIO pin for DS18B20 sensor
'label': 'Outside', # Display name
'alert_high': 85.0, # Send alert if temp > 85°F
'alert_low': 68.0 # Send alert if temp < 68°F
}
}
# Initialize sensors based on configuration
def get_configured_sensors():
"""Return dictionary of configured sensor instances."""
sensors = {}
for key, config in SENSOR_CONFIG.items():
sensors[key] = TemperatureSensor(pin=config['pin'], label=config['label'])
return sensors
# Create sensor instances
sensors = get_configured_sensors()
# ===== END: Sensor Configuration =====
# ===== START: AC Controller Setup =====
# Set up air conditioning relay controller
ac_controller = ACController(
relay_pin=15, # GPIO pin connected to AC relay
min_run_time=30, # Minimum seconds AC must run before turning off
min_off_time=5 # Minimum seconds AC must be off before turning on again
)
# Create AC monitor (automatically controls AC based on temperature)
ac_monitor = ACMonitor(
ac_controller=ac_controller,
temp_sensor=sensors['inside'], # Use inside sensor for AC control
target_temp=config['ac_target'], # Target temp from config.json
temp_swing=config['ac_swing'], # Tolerance (+/- degrees)
interval=30 # Check temperature every 30 seconds
)
# ===== END: AC Controller Setup =====
# ===== START: Heater Controller Setup =====
# Set up heating relay controller
heater_controller = HeaterController(
relay_pin=16, # GPIO pin connected to heater relay
min_run_time=30, # Minimum seconds heater must run before turning off
min_off_time=5 # Minimum seconds heater must be off before turning on again
)
# Create heater monitor (automatically controls heater based on temperature)
heater_monitor = HeaterMonitor(
heater_controller=heater_controller,
temp_sensor=sensors['inside'], # Use inside sensor for heater control
target_temp=config['heater_target'], # Target temp from config.json
temp_swing=config['heater_swing'], # Tolerance (+/- degrees)
interval=30 # Check temperature every 30 seconds
)
# ===== END: Heater Controller Setup =====
# ===== START: Schedule Monitor Setup =====
# Create schedule monitor (automatically changes temp targets based on time of day)
schedule_monitor = ScheduleMonitor(
ac_monitor=ac_monitor, # Pass AC monitor to control
heater_monitor=heater_monitor, # Pass heater monitor to control
config=config, # Pass config with schedules
interval=60 # Check schedule every 60 seconds
)
# ===== END: Schedule Monitor Setup =====
# ===== START: Print Current Settings =====
# Display loaded configuration for debugging
print("\n" + "="*50)
print("Current Climate Control Settings:")
print("="*50)
print(f"AC Target: {config['ac_target']}°F ± {config['ac_swing']}°F")
print(f"Heater Target: {config['heater_target']}°F ± {config['heater_swing']}°F")
print(f"Schedule: {'Enabled' if config.get('schedule_enabled') else 'Disabled'}")
if config.get('schedules'):
print(f"Schedules: {len(config.get('schedules', []))} configured")
print("="*50 + "\n")
# ===== END: Print Current Settings =====
# ===== START: Monitor Setup =====
# Set up all monitoring systems (run in order during main loop)
monitors = [
# WiFi monitor: Checks connection, reconnects if needed, blinks LED
WiFiMonitor(wifi, led, interval=5, reconnect_cooldown=60),
# Schedule monitor: Changes temp targets based on time of day
schedule_monitor,
# AC monitor: Automatically turns AC on/off based on temperature
ac_monitor,
# Heater monitor: Automatically turns heater on/off based on temperature
heater_monitor,
# Inside temperature monitor: Logs temps, sends alerts if out of range
TemperatureMonitor(
sensor=sensors['inside'],
label=SENSOR_CONFIG['inside']['label'],
check_interval=10, # Check temp every 10 seconds
report_interval=30, # Log to CSV every 30 seconds
alert_high=SENSOR_CONFIG['inside']['alert_high'], # High temp alert threshold
alert_low=SENSOR_CONFIG['inside']['alert_low'], # Low temp alert threshold
log_file="/temp_logs.csv", # CSV file path
send_alerts_to_separate_channel=True # Use separate Discord channel
),
# Outside temperature monitor: Logs temps, sends alerts if out of range
TemperatureMonitor(
sensor=sensors['outside'],
label=SENSOR_CONFIG['outside']['label'],
check_interval=10, # Check temp every 10 seconds
report_interval=30, # Log to CSV every 30 seconds
alert_high=SENSOR_CONFIG['outside']['alert_high'], # High temp alert threshold
alert_low=SENSOR_CONFIG['outside']['alert_low'], # Low temp alert threshold
log_file="/temp_logs.csv", # CSV file path
send_alerts_to_separate_channel=False # Use main Discord channel
),
]
# ===== END: Monitor Setup =====
print("Starting monitoring loop...")
print("Press Ctrl+C to stop\n")
# ===== START: Main Loop =====
# Main monitoring loop (runs forever until Ctrl+C)
while True:
# Run all monitors (each checks if it's time to run via should_run())
run_monitors(monitors)
# Check for incoming web requests (non-blocking)
# Pass schedule_monitor so web interface can reload config when schedules change
web_server.check_requests(sensors, ac_monitor, heater_monitor, schedule_monitor)
# ===== START: Garbage Collection =====
# Free up unused memory to prevent fragmentation
gc.collect()
# ===== END: Garbage Collection =====
# Small delay to prevent CPU overload (0.1 seconds = 10 loops per second)
time.sleep(0.1)
# ===== END: Main Loop =====