SensorShare_Receiver2.py

#!/usr/bin/env python3
"""
SensorShare - MQTT Sensor Receiver
Runs on another Raspberry Pi to receive and display sensor data
Updated to include force sensor data and day/night mode
"""

import paho.mqtt.client as mqtt
import json
import time
from datetime import datetime
from collections import defaultdict

# MQTT Configuration
MQTT_BROKER = "192.168.0.202"  # Rotator-PI's IP address
MQTT_PORT = 1883
BASE_TOPIC = "sensors/rotator"

class SensorDataReceiver:
    def __init__(self, broker, port):
        self.client = mqtt.Client()
        self.broker = broker
        self.port = port
        self.connected = False
        self.sensor_data = defaultdict(dict)
        self.last_update = defaultdict(str)
        self.force_sensor_state = {"timecycle": False, "mode": "night"}
        
        # Set up callbacks
        self.client.on_connect = self.on_connect
        self.client.on_disconnect = self.on_disconnect
        self.client.on_message = self.on_message
    
    def on_connect(self, client, userdata, flags, rc):
        if rc == 0:
            print(f"✅ Connected to MQTT broker at {self.broker}:{self.port}")
            self.connected = True
            
            # Subscribe to all sensor topics
            client.subscribe(f"{BASE_TOPIC}/light/#")      # All light sensors
            client.subscribe(f"{BASE_TOPIC}/uvb/#")        # All UVB sensors
            client.subscribe(f"{BASE_TOPIC}/timecycle/#")  # Force sensor timecycle
            client.subscribe(f"{BASE_TOPIC}/mode/#")       # Force sensor day/night mode
            client.subscribe(f"{BASE_TOPIC}/summary")      # Summary data
            print(f"📡 Subscribed to topics:")
            print(f"   - {BASE_TOPIC}/light/# (light sensors)")
            print(f"   - {BASE_TOPIC}/uvb/# (UVB sensors)")
            print(f"   - {BASE_TOPIC}/timecycle/# (force sensor timecycle)")
            print(f"   - {BASE_TOPIC}/mode/# (force sensor day/night mode)")
            print(f"   - {BASE_TOPIC}/summary (summary data)")
        else:
            print(f"❌ Failed to connect to MQTT broker, return code: {rc}")
    
    def on_disconnect(self, client, userdata, rc):
        print("🔌 Disconnected from MQTT broker")
        self.connected = False
    
    def on_message(self, client, userdata, msg):
        """Handle incoming MQTT messages"""
        try:
            topic = msg.topic
            payload = msg.payload.decode('utf-8')
            
            print(f"\n📨 Received message on topic: {topic}")
            
            # Parse JSON data
            data = json.loads(payload)
            
            if topic.endswith('/summary'):
                # Summary data
                print("📊 Summary data received:")
                for sensor_id, value in data['sensors'].items():
                    print(f"   {sensor_id}: {value}")
                print(f"   Total sensors: {data['sensor_count']}")
                print(f"   Timestamp: {data['timestamp']}")
            else:
                # Individual sensor data
                topic_parts = topic.split('/')
                sensor_type = topic_parts[-2]  # light, uvb, timecycle, or mode
                sensor_id = topic_parts[-1]    # outside_dome, inside_dome, or force_a
                
                if sensor_type in ['timecycle', 'mode']:
                    # Force sensor data
                    self.force_sensor_state[sensor_type] = data.get('value', 'N/A')
                    full_sensor_id = f"{sensor_type}_{sensor_id}"
                    self.sensor_data[full_sensor_id] = data
                    self.last_update[full_sensor_id] = datetime.now().strftime("%H:%M:%S")
                    
                    value = data.get('value', 'N/A')
                    unit = data.get('unit', 'N/A')
                    print(f"🔘 {full_sensor_id}: {value} {unit}")
                else:
                    # Light or UVB sensor data
                    full_sensor_id = f"{sensor_type}_{sensor_id}"
                    self.sensor_data[full_sensor_id] = data
                    self.last_update[full_sensor_id] = datetime.now().strftime("%H:%M:%S")
                    
                    value = data.get('value', 'N/A')
                    unit = data.get('unit', 'N/A')
                    print(f"💡 {full_sensor_id}: {value} {unit}")
            
            # Display current status
            self.display_status()
            
        except json.JSONDecodeError as e:
            print(f"❌ Error parsing JSON: {e}")
        except Exception as e:
            print(f"❌ Error processing message: {e}")
    
    def get_motor_status(self):
        """Determine motor status based on current sensor data"""
        day_mode = self.force_sensor_state.get('mode', 'night') == 'day'
        light_outside = None
        
        # Get outside light level
        for sensor_id, data in self.sensor_data.items():
            if sensor_id == 'light_outside_dome':
                light_outside = data.get('value')
                break
        
        if not day_mode:
            return "🌙 Night Mode: Motor Inactive"
        
        if light_outside is None:
            return "⚠️ Day Mode: No Light Data"
        
        if light_outside > 19000:
            return f"☀️ Day Mode: Motor Clockwise (Light: {light_outside} lux > 19000)"
        elif light_outside < 2000:
            return f"☀️ Day Mode: Motor Counter-Clockwise (Light: {light_outside} lux < 2000)"
        else:
            return f"☀️ Day Mode: Motor Stopped (Light: {light_outside} lux, 2000-19000 range)"
    
    def display_status(self):
        """Display current sensor status"""
        print("\n" + "="*70)
        print("📊 CURRENT SENSOR STATUS")
        print("="*70)
        
        # Display force sensor status
        timecycle = self.force_sensor_state.get('timecycle', False)
        mode = self.force_sensor_state.get('mode', 'night')
        mode_icon = "☀️" if mode == 'day' else "🌙"
        timecycle_icon = "✅" if timecycle else "❌"
        
        print("🔘 FORCE SENSOR STATUS:")
        print("-" * 40)
        print(f"   {mode_icon} Mode: {mode.upper()}")
        print(f"   {timecycle_icon} TimeCycle: {timecycle}")
        print()
        
        # Display motor status
        motor_status = self.get_motor_status()
        print("🔄 MOTOR STATUS:")
        print("-" * 40)
        print(f"   {motor_status}")
        print()
        
        if not self.sensor_data:
            print("⏳ Waiting for sensor data...")
            return
        
        # Group sensors by type
        light_sensors = {k: v for k, v in self.sensor_data.items() if k.startswith('light_')}
        uvb_sensors = {k: v for k, v in self.sensor_data.items() if k.startswith('uvb_')}
        force_sensors = {k: v for k, v in self.sensor_data.items() if k.startswith(('timecycle_', 'mode_'))}
        
        # Display light sensors
        if light_sensors:
            print("💡 LIGHT SENSORS:")
            print("-" * 40)
            for sensor_id, data in light_sensors.items():
                value = data.get('value', 'N/A')
                unit = data.get('unit', 'N/A')
                timestamp = data.get('timestamp', 'N/A')
                last_update = self.last_update.get(sensor_id, 'N/A')
                
                location = "Outside Dome" if "outside" in sensor_id else "Inside Dome"
                
                # Add status indicators for outside light
                if "outside" in sensor_id and value != 'N/A':
                    if value > 19000:
                        status = "🔴 HIGH (Motor Clockwise)"
                    elif value < 2000:
                        status = "🔵 LOW (Motor Counter-Clockwise)"
                    else:
                        status = "🟡 MEDIUM (Motor Stopped)"
                else:
                    status = ""
                
                print(f"📍 {location}")
                print(f"   💡 Light Level: {value} {unit} {status}")
                print(f"   🕐 Last Update: {last_update}")
                print(f"   📅 Timestamp: {timestamp}")
                print()
        
        # Display UVB sensors
        if uvb_sensors:
            print("☀️ UVB SENSORS:")
            print("-" * 40)
            for sensor_id, data in uvb_sensors.items():
                value = data.get('value', 'N/A')
                unit = data.get('unit', 'N/A')
                timestamp = data.get('timestamp', 'N/A')
                last_update = self.last_update.get(sensor_id, 'N/A')
                
                location = "Outside Dome" if "outside" in sensor_id else "Inside Dome"
                status = "✅ Working" if value != 'N/A' and value != 0 else "⚠️ No reading"
                print(f"📍 {location} - {status}")
                print(f"   ☀️ UVB Level: {value} {unit}")
                print(f"   🕐 Last Update: {last_update}")
                print(f"   📅 Timestamp: {timestamp}")
                print()
        
        # Display force sensor data
        if force_sensors:
            print("🔘 FORCE SENSOR DATA:")
            print("-" * 40)
            for sensor_id, data in force_sensors.items():
                value = data.get('value', 'N/A')
                unit = data.get('unit', 'N/A')
                timestamp = data.get('timestamp', 'N/A')
                last_update = self.last_update.get(sensor_id, 'N/A')
                
                sensor_type = "TimeCycle" if "timecycle" in sensor_id else "Day/Night Mode"
                print(f"📍 {sensor_type}")
                print(f"   🔘 Value: {value} {unit}")
                print(f"   🕐 Last Update: {last_update}")
                print(f"   📅 Timestamp: {timestamp}")
                print()
        
        # Display summary
        total_sensors = len(self.sensor_data)
        working_sensors = sum(1 for data in self.sensor_data.values() 
                            if data.get('value') not in ['N/A', None])
        
        print("📈 SUMMARY:")
        print("-" * 40)
        print(f"   Total sensors detected: {total_sensors}")
        print(f"   Working sensors: {working_sensors}")
        print(f"   Light sensors: {len(light_sensors)}")
        print(f"   UVB sensors: {len(uvb_sensors)}")
        print(f"   Force sensors: {len(force_sensors)}")
        print("="*70)
    
    def connect(self):
        """Connect to MQTT broker"""
        try:
            self.client.connect(self.broker, self.port, 60)
            self.client.loop_start()
        except Exception as e:
            print(f"❌ Error connecting to MQTT broker: {e}")
    
    def disconnect(self):
        """Disconnect from MQTT broker"""
        self.client.loop_stop()
        self.client.disconnect()

def main():
    print("🚀 Starting SensorShare MQTT Receiver...")
    print(f"📡 Connecting to Rotator-PI at {MQTT_BROKER}:{MQTT_PORT}")
    print("🎯 This receiver will display data from:")
    print("   - Light sensors (inside & outside dome)")
    print("   - UVB sensors (inside & outside dome)")
    print("   - Force sensor (TimeCycle & Day/Night mode)")
    print("   - Motor status based on light levels and mode")
    print("   - Summary data from all sensors")
    
    # Initialize receiver
    receiver = SensorDataReceiver(MQTT_BROKER, MQTT_PORT)
    receiver.connect()
    
    # Wait for connection
    time.sleep(2)
    
    if not receiver.connected:
        print("❌ Could not connect to MQTT broker. Exiting.")
        return
    
    print("\n🎯 Listening for sensor data...")
    print("Press Ctrl+C to stop")
    
    try:
        while True:
            # Keep the program running
            time.sleep(1)
            
    except KeyboardInterrupt:
        print("\n🛑 Stopping SensorShare Receiver...")
    except Exception as e:
        print(f"❌ Error in main loop: {e}")
    finally:
        receiver.disconnect()
        print("👋 SensorShare Receiver stopped")

if __name__ == "__main__":
    main()