#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import csv
from datetime import datetime
import signal
import time
import board
import adafruit_ina260
import ambient

# ambientサイトへデータを送信するためのキーを登録
ambi1 = ambient.Ambient(xxxx1, "xxxxxxxxxxxxxxx1") 
ambi2 = ambient.Ambient(xxxx2, "xxxxxxxxxxxxxxx2")

REC_START_HOUR = 5 
REC_END_HOUR = 17
 
# CSVファイルからWの値の合計を計算する関数
def calculate_w_total(csv_filename):
    w1_total = 0.0  # Wの合計を初期化
    w2_total = 0.0

    try:
        with open(csv_filename, mode='r', newline='') as file:
            reader = csv.reader(file)
            #header = next(reader)  # ヘッダー行をスキップ

            for row in reader:
                if len(row) >= 5:
                    w1_value = float(row[4])  # Wの値を浮動小数点数に変換
                    w1_total += w1_value  # Wの値を合計に追加
                    w2_value = float(row[7])  # Wの値を浮動小数点数に変換
                    w2_total += w2_value  # Wの値を合計に追加

		# 1秒間隔のデータを1時間データにして Wh 単位としてリターンする
        return w1_total/3600000, w2_total/3600000

    except FileNotFoundError:
        print(f"ファイル '{csv_filename}' が見つかりませんでした。")
        return None
    except Exception as e:
        print(f"エラーが発生しました: {e}")
        return None
 
# メインタスク関数
def task(arg1, arg2):
 
    # 電流・電圧・電力の取得
    cur1 = ina260a.current
    vol1 = ina260a.voltage
    po1  = ina260a.power
    cur2 = ina260b.current
    vol2 = ina260b.voltage
    po2  = ina260b.power

    global po1d
    global po2d

    po1d.append(po1)
    po2d.append(po2)

    # 日時の取得
    _now = datetime.now()
    today = _now.strftime("%Y-%m-%d")
    nowtime = _now.strftime("%H:%M:%S")
    file_end = _now.strftime("%Y%m%d")
    
    print ('日付 %s 時刻 %s 電流 %.2f/%.2f mA 電圧 %.2f/%.2f V 電力 %.2f/%.2f mW' % (today,nowtime,cur1,cur2,vol1,vol2,po1,po2))
    
    if REC_START_HOUR <= _now.hour <= REC_END_HOUR:
        with open('/home/admin/work/power/power2_%s.csv' % (file_end),'a') as f:
            writer = csv.writer(f)
            writer.writerow([today,nowtime,cur1,vol1,po1,cur2,vol2,po2])
            
            if _now.second == 0:
                input_csv_filename = '/home/admin/work/power/power2_%s.csv' % file_end
                total_w1, total_w2 = calculate_w_total(input_csv_filename)
                total_all = total_w1 + total_w2

                if len(po1d) > 0 and len(po2d) > 0 and _now.minute == 59:
                    po1a = sum(po1d) / len(po1d)
                    po2a = sum(po2d) / len(po2d)
                    po_all = po1a + po2a
                    po1d = []
                    po2d = []
                    print(total_w1, total_w2, total_all, po1a, po2a, po_all)
                    r = ambi1.send({"d1": total_w1, "d2": total_w2, "d3": total_all, "d4": po1a, "d5": po2a, "d6": po_all})
                
                print(total_w1, total_w2, total_all)
                r = ambi2.send({"d1": cur1, "d2": vol1, "d3": po1, "d4": total_w1, "d5": total_w2, "d6": cur2, "d7": vol2, "d8": po2})
    else:
        print('記録時間外')
 
i2c = board.I2C()
ina260a = adafruit_ina260.INA260(i2c)
ina260b = adafruit_ina260.INA260(i2c,address=0x43)

po1d = []
po2d = []


signal.signal(signal.SIGALRM, task)
signal.setitimer(signal.ITIMER_REAL, 0.1, 1) # コマンドラインでCtrl+Cで止められるように
 
while True:
    time.sleep(1)