import pymysql
from contextlib import closing
import pymysql
from pymysql.cursors import DictCursor
from time import sleep
from datetime import datetime

def strSost(x):
    x = str(x.lower())

    if x == '0':
        return 'Отключен'
    elif x == '1':
        return 'Подключен'
    else:
        return 'Суперпозиция'

#Запуск
in1 = 1
in2 = 1
in3 = 1
in4 = 1
in5 = 1

print('Программа по управлению системами микро грида "smart-energy" запущенна')
print ('Время запуска: ' + datetime.today().strftime('%Y-%m-%d %H:%M:%S'))
print()

#Работа с бд
with closing(
pymysql.connect(
    host='remotemysql.com',
    user='psmg6F26LS',
    password='jDixxlgdxx',
    db='psmg6F26LS',
    cursorclass=DictCursor
)
) as connection:
    with connection.cursor() as cursor:
        while True:
            try:
                date = datetime.today().strftime('%Y-%m-%d %H:%M:%S')

                #TODO дописать получение этих значений
                voutsun = 1.0
                aoutsun = 1.0

                vouth = 0.0
                aouth = 0.0

                pAll = (voutsun * aoutsun) + (vouth * aouth)

                #Система приоритетов
                #mysql
                query="INSERT INTO `ener-data`(`id`, `date`, `v-out-sun`, `a-out-sun`, `v-out-h`, `a-out-h`, `in1`, `in2`, `in3`, `in4`, `in5`) VALUES (null,'" + date + "'," + str(voutsun) + "," + str(aoutsun) + "," +str(vouth) + "," + str(aouth) + "," + str(in1)  + "," + str(in2)  + "," + str(in3) + "," + str(in4)  + "," + str(in5) + ")"
                cursor.execute(query)
                connection.commit()

                query="SELECT * FROM `ener-data`    WHERE 1 ORDER BY id DESC LIMIT 1"
                cursor.execute(query)
                for row in cursor:
                    #Вывод показания
                    pAll = (row['v-out-sun'] * row['a-out-sun']) + (row['v-out-h'] * row['a-out-h'])
                    print('Запись #' + str(row['id']) + ' от ' + str(row['date']) + ':')
                    print('Показания источников:')
                    print(' Солнечная панель | ' + str(row['v-out-sun']) + 'В ' + str(row['a-out-sun']) + 'А ' + str(row['v-out-sun'] * row['a-out-sun']) + 'Вт')
                    print(' Водородный ТЭ    | ' + str(row['v-out-h']) + 'В ' + str(row['a-out-h']) + 'А ' + str(row['v-out-h'] * row['a-out-h']) + 'Вт')
                print(' Общая мощьность  | ' + str(pAll))
                print('Состояния потребителей:')
                print(' 1 | Потребитель | ' + strSost(row['in1']))
                print(' 2 | Потребитель | ' + strSost(row['in2']))
                print(' 3 | Потребитель | ' + strSost(row['in3']))
                print(' 4 | Потребитель | ' + strSost(row['in4']))
                print(' 5 | Потребитель | ' + strSost(row['in5']))
                print()
                sleep(5)
            except KeyboardInterrupt:
                print('Завершение работы')
                exit()