'''
  _   _ ____  __  __ 
 | \ | |  _ \|  \/  |
 |  \| | |_) | |\/| |
 | |\  |  __/| |  | |
 |_| \_|_|   |_|  |_|
                                     
Script to get NPM values
PM and the sensor temp/hum
And store them inside sqlite database
Uses RTC module for timing
/usr/bin/python3 /var/www/nebuleair_pro_4g/NPM/get_data_v2.py
'''

import serial
import requests
import json
import sys
import sqlite3
import smbus2
import time
from datetime import datetime

# Connect to the SQLite database
conn = sqlite3.connect("/var/www/nebuleair_pro_4g/sqlite/sensors.db")
cursor = conn.cursor()

#RTC module
DS3231_ADDR = 0x68
REG_TIME = 0x00

def bcd_to_dec(bcd):
    return (bcd // 16 * 10) + (bcd % 16)

def read_time(bus):
    """Try to read and decode time from the RTC module (DS3231)."""
    try:
        data = bus.read_i2c_block_data(DS3231_ADDR, REG_TIME, 7)
        seconds = bcd_to_dec(data[0] & 0x7F)
        minutes = bcd_to_dec(data[1])
        hours = bcd_to_dec(data[2] & 0x3F)
        day = bcd_to_dec(data[4])
        month = bcd_to_dec(data[5])
        year = bcd_to_dec(data[6]) + 2000
        return datetime(year, month, day, hours, minutes, seconds)
    except OSError:
        return None  # RTC module not connected

def load_config(config_file):
    try:
        with open(config_file, 'r') as file:
            config_data = json.load(file)
        return config_data
    except Exception as e:
        print(f"Error loading config file: {e}")
        return {}

# Load the configuration data
config_file = '/var/www/nebuleair_pro_4g/config.json'
config = load_config(config_file)
npm_solo_port = config.get('NPM_solo_port', '')  #port du NPM solo

ser = serial.Serial(
    port=npm_solo_port,
    baudrate=115200,
    parity=serial.PARITY_EVEN,
    stopbits=serial.STOPBITS_ONE,
    bytesize=serial.EIGHTBITS,
    timeout = 0.5
)

#ser.write(b'\x81\x11\x6E')      #data10s
ser.write(b'\x81\x12\x6D')      #data60s


#print("Start get_data_v2.py script")
byte_data = ser.readline()
#print(byte_data)
stateByte = int.from_bytes(byte_data[2:3], byteorder='big')
Statebits = [int(bit) for bit in bin(stateByte)[2:].zfill(8)]
PM1 = int.from_bytes(byte_data[9:11], byteorder='big')/10
PM25 = int.from_bytes(byte_data[11:13], byteorder='big')/10
PM10 = int.from_bytes(byte_data[13:15], byteorder='big')/10

# Write command to retrieve temperature and humidity data
ser.write(b'\x81\x14\x6B')  # Temp and humidity command
byte_data_temp_hum = ser.readline()

# Decode temperature and humidity values
temperature = int.from_bytes(byte_data_temp_hum[3:5], byteorder='big') / 100.0
humidity = int.from_bytes(byte_data_temp_hum[5:7], byteorder='big') / 100.0

#print(f"State: {Statebits}")
#print(f"PM1: {PM1}")
#print(f"PM25: {PM25}")
#print(f"PM10: {PM10}")
#print(f"temp: {temperature}")
#print(f"hum: {humidity}")
#GET RTC TIME
# Read RTC time
bus = smbus2.SMBus(1)
# Try to read RTC time
rtc_time = read_time(bus)

if rtc_time:
    rtc_time_str = rtc_time.strftime('%Y-%m-%d %H:%M:%S')
    #print(rtc_time_str)
else:
    print("Error! RTC module not connected")
    rtc_time_str = "1970-01-01 00:00:00"  # Default fallback time


#save to sqlite database
try:
    cursor.execute('''
    INSERT INTO data_NPM (timestamp,PM1, PM25, PM10, temp_npm, hum_npm) VALUES (?,?,?,?,?,?)'''
    , (rtc_time_str,PM1,PM25,PM10,temperature,humidity  ))

    # Commit and close the connection
    conn.commit()

    #print("Sensor data saved successfully!")

except Exception as e:
    print(f"Database error: {e}")    


conn.close()