Spaces:
Running
Running
from flask import Flask, request, jsonify, redirect, url_for, render_template, send_from_directory, send_file, render_template_string | |
import os | |
import sqlite3 | |
from datetime import datetime | |
import pytz | |
import io | |
import base64 | |
from dotenv import load_dotenv | |
import requests | |
import json | |
import logging | |
import uuid | |
from io import BytesIO | |
from tabulate import tabulate | |
import numpy as np | |
from scipy.optimize import linprog | |
from typing import Dict, Any, Union, List | |
import itertools | |
import traceback | |
import logging | |
from werkzeug.utils import secure_filename | |
import globs | |
from api_logic import api | |
from urllib.parse import urlencode # Добавлен правильный импорт | |
load_dotenv() | |
# Инициализация базы данных | |
def init_db(db_name): | |
conn = sqlite3.connect(db_name) | |
cursor = conn.cursor() | |
# Таблица с системными данными (твоя старая таблица) | |
cursor.execute(''' | |
CREATE TABLE IF NOT EXISTS system_data ( | |
id INTEGER PRIMARY KEY AUTOINCREMENT, | |
date_time TEXT, | |
dey TEXT, | |
wek TEXT, | |
v_hid TEXT, | |
v_min TEXT, | |
ph TEXT, | |
ec TEXT, | |
tS TEXT, | |
tA TEXT, | |
hDm TEXT, | |
sVen TEXT, | |
onA TEXT, | |
onB TEXT, | |
onC TEXT, | |
nPh TEXT, | |
nEC TEXT, | |
nTa TEXT, | |
nLon TEXT, | |
nLoff TEXT | |
) | |
''') | |
# **Новая таблица для пользователей бота** | |
cursor.execute(''' | |
CREATE TABLE IF NOT EXISTS bot_users ( | |
id INTEGER PRIMARY KEY AUTOINCREMENT, -- Уникальный ID | |
chat_id INTEGER UNIQUE, -- Telegram ID пользователя | |
created_at TEXT -- Время добавления (ISO формат) | |
) | |
''') | |
conn.commit() | |
conn.close() | |
# Глобальные переменные | |
api_key_sys = os.getenv('api_key') # Берём значение API-ключа | |
btg_key = os.getenv('btg_key') # Берём значение | |
btg_id = os.getenv('btg_id') # Берём значение | |
btg_on = os.getenv('btg_on') # Берём значение | |
globs.dey = 0 | |
globs.wek = 0 | |
globs.v_hid = 0 | |
globs.v_min = 0 | |
globs.ph = 0 | |
globs.ec = 0 | |
globs.tS = 0 | |
globs.tA = 0 | |
globs.hDm = 0 | |
globs.sVen = 0 | |
globs.onA = 0 | |
globs.onB = 0 | |
globs.onC = 0 | |
globs.ph_eep = 0 | |
globs.ph_on_eep = 0 | |
globs.ec_eep = 0 | |
globs.ec_A_eep = 0 | |
globs.ec_B_eep = 0 | |
globs.ec_C_eep = 0 | |
globs.l_ON_h_eep = 0 | |
globs.l_ON_m_eep = 0 | |
globs.l_OFF_h_eep = 0 | |
globs.l_OFF_m_eep = 0 | |
globs.t_Voz_eep = 0 | |
# Создаем экземпляр Flask-приложения | |
app = Flask(__name__, template_folder="./") | |
app.config['DEBUG'] = True | |
UPLOAD_FOLDER = 'uploads' | |
ALLOWED_EXTENSIONS = {'jpg', 'jpeg', 'png', 'gif'} | |
# Создаем папку для загрузок если ее нет | |
os.makedirs(UPLOAD_FOLDER, exist_ok=True) | |
# Глобальная переменная для хранения последнего изображения в памяти | |
latest_image = {"data": None, "filename": None} | |
# Настроим логирование | |
logging.basicConfig(level=logging.DEBUG) | |
# Функция сохранения в базу данных системы автоматизации гидропоники | |
def save_data_to_db(db_name, data): | |
try: | |
conn = sqlite3.connect(db_name) | |
cursor = conn.cursor() | |
# ✅ Устанавливаем московское время (UTC+3) | |
moscow_tz = pytz.timezone("Europe/Moscow") | |
current_time = datetime.now(moscow_tz).strftime('%Y-%m-%d %H:%M:%S') | |
# Вставляем данные в таблицу | |
cursor.execute(''' | |
INSERT INTO system_data ( | |
date_time, dey, wek, v_hid, v_min, ph, ec, tS, tA, hDm, sVen, onA, onB, onC, nPh, nEC, nTa, nLon, nLoff | |
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) | |
''', ( | |
current_time, # ✅ Дата и время по Москве | |
data['dey'], data['wek'], data['v_hid'], data['v_min'], data['ph'], data['ec'], | |
data['tS'], data['tA'], data['hDm'], data['sVen'], data['onA'], data['onB'], | |
data['onC'], data['nPh'], data['nEC'], data['nTa'], data['nLon'], data['nLoff'] | |
)) | |
conn.commit() | |
conn.close() | |
except Exception as e: | |
return jsonify({'status': 'error', 'message': str(e)}), 500 | |
# Маршрут сохранения в базу | |
def sav_db_api(): | |
# Инициализируем базу данных | |
init_db('system_data.db') | |
# Получаем данные из запроса | |
data = { | |
'dey': request.args.get('dey'), | |
'wek': request.args.get('wek'), | |
'v_hid': request.args.get('v_hid'), | |
'v_min': request.args.get('v_min'), | |
'ph': request.args.get('ph'), | |
'ec': request.args.get('ec'), | |
'tS': request.args.get('tS'), | |
'tA': request.args.get('tA'), | |
'hDm': request.args.get('hDm'), | |
'sVen': request.args.get('sVen'), | |
'onA': request.args.get('onA'), | |
'onB': request.args.get('onB'), | |
'onC': request.args.get('onC'), | |
'nPh': request.args.get('nPh'), | |
'nEC': request.args.get('nEC'), | |
'nTa': request.args.get('nTa'), | |
'nLon': request.args.get('nLon'), | |
'nLoff': request.args.get('nLoff') | |
} | |
# Проверяем, что все необходимые параметры переданы | |
required_params = ['dey', 'wek', 'v_hid', 'v_min', 'ph', 'ec', 'tS', 'tA', 'hDm', 'sVen', 'onA', 'onB', 'onC', 'nPh', 'nEC', 'nTa', 'nLon', 'nLoff'] | |
for param in required_params: | |
if data[param] is None: | |
return jsonify({'status': 'error', 'message': f'Отсутствует параметр: {param}'}), 400 | |
# Сохраняем данные в базу | |
save_data_to_db('system_data.db', data) | |
# Возвращаем ответ | |
return jsonify({'status': 'success', 'message': 'Save OK'}) | |
# Проверка входа на страницы | |
def check_api_key(): | |
api_sys_param = request.args.get('api_sys') # Получаем параметр из запроса | |
if api_sys_param == api_key_sys: | |
return jsonify({"status": "ok"}), 200 # ✅ Совпадает — отправляем "ok" | |
else: | |
return jsonify({"status": "error", "message": "Invalid API key"}), 403 # ❌ Ошибка 403 | |
# Тестовый запрос с установки | |
def test_server(): | |
api_key_param = request.args.get('api_sys') # Получаем параметр из запроса | |
err_ser = 1 if api_key_param == api_key_sys else 0 # Проверяем совпадение ключей | |
return jsonify(err_ser=err_ser) | |
def test_server_str(): | |
api_key_param = request.args.get('api_sys') | |
err_ser = "1" if api_key_param == api_key_sys else "0" | |
return err_ser # Возвращаем строку "1" или "0" | |
# Тестовый запрос с установки | |
def btg_teleg(): | |
api_key_param = request.args.get('api_sys') # Получаем параметр из запроса | |
return jsonify(btg_key_ser=btg_key,btg_id_ser=btg_id,btg_on_ser=btg_on) | |
# Маршрут для вывода всех данных из таблицы | |
def get_all_data(): | |
try: | |
conn = sqlite3.connect('system_data.db') | |
cursor = conn.cursor() | |
# Выполняем запрос для получения всех данных из таблицы | |
cursor.execute('SELECT * FROM system_data') | |
rows = cursor.fetchall() | |
# Получаем названия столбцов | |
column_names = [description[0] for description in cursor.description] | |
# Преобразуем данные в формат JSON | |
data = [] | |
for row in rows: | |
data.append(dict(zip(column_names, row))) | |
conn.close() | |
# Возвращаем данные в формате JSON | |
return jsonify(data) | |
except Exception as e: | |
return jsonify({'status': 'error', 'message': str(e)}), 500 | |
# Удаление базы | |
def delete_db(): | |
try: | |
conn = sqlite3.connect("system_data.db") # Используем вашу БД | |
cursor = conn.cursor() | |
# ✅ Удаляем все записи из таблицы | |
cursor.execute("DELETE FROM system_data") | |
# ✅ Сбрасываем автоинкрементный счётчик ID (для SQLite) | |
cursor.execute("DELETE FROM sqlite_sequence WHERE name='system_data'") | |
conn.commit() | |
conn.close() | |
return jsonify({'status': 'ok', 'message': 'База данных успешно очищена'}) | |
except Exception as e: | |
return jsonify({'status': 'error', 'message': str(e)}), 500 | |
def plot_week(): | |
try: | |
# Получаем номер недели из параметров запроса | |
week_number = request.args.get('week', default=1, type=int) | |
week_number = max(1, min(30, week_number)) # Ограничиваем диапазон 1-30 | |
# Подключаемся к базе данных | |
conn = sqlite3.connect('system_data.db') | |
cursor = conn.cursor() | |
# Проверяем существование таблицы | |
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='system_data'") | |
table_exists = cursor.fetchone() | |
if not table_exists: | |
conn.close() | |
return render_template('plot_week.html', data=None, week_number=week_number, table_exists=False) | |
# Запрашиваем данные за выбранную неделю | |
cursor.execute(''' | |
SELECT date_time, dey, ph, ec, tS, tA, hDm, sVen, onA, onB, onC, v_hid, v_min | |
FROM system_data | |
WHERE wek = ? | |
ORDER BY date_time | |
''', (str(week_number),)) # Приводим week_number к строке, так как wek имеет тип TEXT | |
rows = cursor.fetchall() | |
conn.close() | |
# Если данных нет | |
if not rows: | |
return render_template('plot_week.html', data=None, week_number=week_number, table_exists=True) | |
# Формируем данные для JSON | |
data = { | |
'week': week_number, | |
'dates': [row[0] for row in rows], | |
'days_of_week': [int(row[1]) if row[1] else 0 for row in rows], # Преобразуем dey в int | |
'ph': [float(row[2]) if row[2] else 0.0 for row in rows], # pH | |
'ec': [float(row[3]) if row[3] else 0.0 for row in rows], # EC | |
'tS': [float(row[4]) if row[4] else 0.0 for row in rows], # Температура раствора | |
'tA': [float(row[5]) if row[5] else 0.0 for row in rows], # Температура воздуха | |
'hDm': [float(row[6]) if row[6] else 0.0 for row in rows], # Влажность воздуха | |
'sVen': [float(row[7]) if row[7] else 0.0 for row in rows], # Обороты вентилятора | |
'onA': [float(row[8]) if row[8] else 0.0 for row in rows], # Насос A | |
'onB': [float(row[9]) if row[9] else 0.0 for row in rows], # Насос B | |
'onC': [float(row[10]) if row[10] else 0.0 for row in rows], # Насос C | |
'sus': [f"{row[11]}:{row[12]}" if row[11] and row[12] else "0:0" for row in rows] # Объединяем v_hid и v_min | |
} | |
# Отправляем данные в HTML-шаблон | |
return render_template('plot_week.html', data=data, week_number=week_number, table_exists=True) | |
except Exception as e: | |
# Логируем ошибку в консоль для отладки | |
print(f"Ошибка: {str(e)}") | |
return render_template('plot_week.html', data=None, week_number=week_number, table_exists=True, message=f"Ошибка: {str(e)}") | |
def index(): | |
return flask.render_template('index.html') | |
def online(): | |
return render_template('online.html') | |
def table(): | |
return render_template('table.html') | |
def online_api(): | |
# Устанавливаем московское время (UTC+3) | |
moscow_tz = pytz.timezone("Europe/Moscow") | |
current_time = datetime.now(moscow_tz) | |
# Форматируем дату и время отдельно | |
date = current_time.strftime('%Y-%m-%d') # Например, "2025-03-23" | |
time = current_time.strftime('%H:%M:%S') # Например, "14:35:42" | |
return jsonify( | |
dey=globs.dey, | |
wek=globs.wek, | |
v_hid=globs.v_hid, | |
v_min=globs.v_min, | |
ph=globs.ph, | |
ec=globs.ec, | |
tS=globs.tS, | |
tA=globs.tA, | |
hDm=globs.hDm, | |
sVen=globs.sVen, | |
rFul=globs.rFul, | |
rLi=globs.rLi, | |
rWat=globs.rWat, | |
rRas=globs.rRas, | |
rPH=globs.rPH, | |
rEC=globs.rEC, | |
rSl=globs.rSl, | |
rLe=globs.rLe, | |
alW=globs.alW, | |
ec_A_eep=globs.ec_A_eep, | |
ec_B_eep=globs.ec_B_eep, | |
ec_C_eep=globs.ec_C_eep, | |
date=date, # Добавляем дату | |
time=time # Добавляем время | |
) | |
def settings(): | |
return render_template('settings.html') | |
def settings_api(): | |
return jsonify(ph_eep=globs.ph_eep, | |
ph_on_eep=globs.ph_on_eep, | |
ec_eep=globs.ec_eep, | |
ec_A_eep=globs.ec_A_eep, | |
ec_B_eep=globs.ec_B_eep, | |
ec_C_eep=globs.ec_C_eep, | |
l_ON_h_eep=globs.l_ON_h_eep, | |
l_ON_m_eep=globs.l_ON_m_eep, | |
l_OFF_h_eep=globs.l_OFF_h_eep, | |
l_OFF_m_eep=globs.l_OFF_m_eep, | |
t_Voz_eep=globs.t_Voz_eep, | |
set_st=globs.set_status | |
) | |
def set_pH_value(): | |
ph_value = request.args.get('value') | |
globs.ph_set = ph_value | |
globs.eep_set = 1 | |
return "pH value set successfully" | |
def ph_on_value(): | |
ph_on_value = request.args.get('value') | |
globs.ph_on_set = ph_on_value | |
globs.eep_set = 2 | |
return "EC value set successfully" | |
def set_EC_value(): | |
ec_value = request.args.get('value') | |
globs.ec_set = ec_value | |
globs.eep_set = 3 | |
return "EC value set successfully" | |
def ec_A_setValue(): | |
ec_A_setValue = request.args.get('value') | |
globs.ec_A_set = ec_A_setValue | |
globs.eep_set = 4 | |
return "EC value set successfully" | |
def ec_B_setValue(): | |
ec_B_setValue = request.args.get('value') | |
globs.ec_B_set = ec_B_setValue | |
globs.eep_set = 5 | |
return "EC value set successfully" | |
def ec_C_setValue(): | |
ec_C_setValue = request.args.get('value') | |
globs.ec_C_set = ec_C_setValue | |
globs.eep_set = 6 | |
return "EC value set successfully" | |
def l_ON_set(): | |
globs.l_ON_h_set = request.args.get('l_ON_h_set') | |
globs.l_ON_m_set = request.args.get('l_ON_m_set') | |
globs.eep_set = 7 | |
return "EC value set successfully" | |
def l_OFF_set(): | |
globs.l_OFF_h_set = request.args.get('l_OFF_h_set') | |
globs.l_OFF_m_set = request.args.get('l_OFF_m_set') | |
globs.eep_set = 8 | |
return "EC value set successfully" | |
def t_Voz_eep_set(): | |
t_Voz_eep_set = request.args.get('value') | |
globs.t_Voz_set = t_Voz_eep_set | |
globs.eep_set = 9 | |
return "EC value set successfully" | |
def but_start(): | |
globs.eep_set = 10 | |
return jsonify(value_set="start") | |
def but_stop(): | |
globs.eep_set = 11 | |
return jsonify(value_set="stop") | |
def but_res(): | |
globs.eep_set = 12 | |
return jsonify(value_set="res") | |
def but_sliv(): | |
globs.eep_set = 13 | |
return jsonify(value_set="sliv") | |
def handle_api(): | |
response = api() | |
return response | |
def handle_save_db(): | |
response = save_db() | |
return response | |
def set_res(): | |
globs.eep_set = 0 | |
return jsonify(value_set="reset") | |
def allowed_file(filename): | |
return '.' in filename and \ | |
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS | |
def upload_file(): | |
if 'file' not in request.files: | |
return jsonify({"error": "No file part"}), 400 | |
file = request.files['file'] | |
if file.filename == '': | |
return jsonify({"error": "No selected file"}), 400 | |
if not allowed_file(file.filename): | |
return jsonify({"error": "Invalid file type"}), 400 | |
# Генерируем имя для файла с использованием времени | |
timestamp = datetime.now().strftime('%Y.%m.%d_%H:%M:%S_') | |
filename = timestamp + file.filename | |
save_path = os.path.join(UPLOAD_FOLDER, filename) | |
# Открываем файл для записи и собираем его части | |
try: | |
with open(save_path, 'wb') as f: | |
while chunk := file.read(1024): # Чтение и запись данных частями | |
f.write(chunk) | |
return jsonify({ | |
"message": "File uploaded successfully", | |
"filename": filename, | |
"path": save_path | |
}), 200 | |
except Exception as e: | |
return jsonify({"error": str(e)}), 500 | |
def uploaded_file(filename): | |
return send_from_directory(UPLOAD_FOLDER, filename) | |
# 🧠 2. Маршрут: сохраняет файл только в память (BytesIO) | |
# Маршрут для загрузки файла в память | |
# Загрузка изображения в память (в виде байтов) | |
def upload_file_to_memory(): | |
if 'file' not in request.files: | |
return jsonify({"error": "No file part"}), 400 | |
file = request.files['file'] | |
if file.filename == '': | |
return jsonify({"error": "No selected file"}), 400 | |
if not file.filename.lower().endswith(('.jpg', '.jpeg', '.png')): | |
return jsonify({"error": "Invalid file type"}), 400 | |
timestamp = datetime.now().strftime('%Y.%m.%d_%H:%M:%S_') | |
filename = timestamp + file.filename | |
try: | |
# Читаем весь файл в байты | |
file_bytes = file.read() | |
# Сохраняем в переменную | |
latest_image["data"] = file_bytes | |
latest_image["filename"] = filename | |
return jsonify({ | |
"message": "Image uploaded successfully", | |
"filename": filename | |
}), 200 | |
except Exception as e: | |
return jsonify({"error": str(e)}), 500 | |
# Получение последнего изображения | |
def get_last_image(): | |
if not latest_image["data"]: | |
return jsonify({"error": "No image available"}), 404 | |
# Возвращаем через новый BytesIO каждый раз | |
return send_file( | |
BytesIO(latest_image["data"]), | |
mimetype='image/jpeg', | |
download_name=latest_image["filename"] | |
) | |
def view_image(): | |
return render_template('show_image.html') | |
def nutri_call(): | |
return render_template('nutri_call.html') | |
from tabulate import tabulate | |
# Входные данные (пример) | |
INPUT_DATA = { | |
"fertilizerConstants": { | |
"Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca": 0.16972}, | |
"Калий азотнокислый": {"N (NO3-)": 0.13854, "K": 0.36672}, | |
"Аммоний азотнокислый": {"N (NO3-)": 0.17499, "N (NH4+)": 0.17499}, | |
"Сульфат магния": {"Mg": 0.10220, "S": 0.13483}, | |
"Монофосфат калия": {"P": 0.22761, "K": 0.28731}, | |
"Калий сернокислый": {"K": 0.44874, "S": 0.18401}, | |
"Кальций хлорид": {"Ca": 0.272, "Cl": 0.483} | |
}, | |
"profileSettings": { | |
"P": 31, "K": 210, "Mg": 24, "Ca": 82, "S": 57.5, "CaCl": 18.5, | |
"NO3_RAT": 8.25, "TOTAL_NITROG": 125, "liters": 100, | |
"activation_cacl": 5, # Активация CaCl | |
"enhancement_cacl": 0.1 | |
} | |
} | |
from tabulate import tabulate | |
# Входные данные (пример) | |
INPUT_DATA = { | |
"fertilizerConstants": { | |
"Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca": 0.16972}, | |
"Калий азотнокислый": {"N (NO3-)": 0.13854, "K": 0.36672}, | |
"Аммоний азотнокислый": {"N (NO3-)": 0.17499, "N (NH4+)": 0.17499}, | |
"Сульфат магния": {"Mg": 0.10220, "S": 0.13483}, | |
"Монофосфат калия": {"P": 0.22761, "K": 0.28731}, | |
"Калий сернокислый": {"K": 0.44874, "S": 0.18401} | |
}, | |
"profileSettings": { | |
"P": 31, "K": 210, "Mg": 24, "Ca": 82, "S": 57.5, | |
"NO3_RAT": 8.25, "TOTAL_NITROG": 125, "liters": 100 | |
} | |
} | |
class NutrientCalculator: | |
def __init__(self, fertilizer_constants, profile_settings, liters, rounding_precision): | |
self.fertilizers = fertilizer_constants | |
self.profile = profile_settings | |
self.volume = liters | |
self.rounding_precision = rounding_precision | |
total_parts = self.profile["NO3_RAT"] + 1 | |
self.target = { | |
'P': self.profile["P"], | |
'K': self.profile["K"], | |
'Mg': self.profile["Mg"], | |
'Ca': self.profile["Ca"], # Общий кальций | |
'S': self.profile["S"], | |
'N (NO3-)': self.profile["TOTAL_NITROG"] * (self.profile["NO3_RAT"] / total_parts), | |
'N (NH4+)': self.profile["TOTAL_NITROG"] * (1 / total_parts) | |
} | |
self.actual = {k: 0.0 for k in self.target} # CaCl не учитываем в фактическом балансе | |
self.results = {fert: {'граммы': 0.0} for fert in self.fertilizers} | |
def calculate(self): | |
# Добавляем аммонийный азот | |
self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target["N (NH4+)"]) | |
# Добавляем фосфор | |
self._apply_fertilizer("Монофосфат калия", "P", self.target["P"]) | |
# Добавляем магний | |
self._apply_fertilizer("Сульфат магния", "Mg", self.target["Mg"]) | |
# Балансируем калий и серу | |
self._balance_k_s() | |
# Распределяем кальций | |
self._distribute_calcium() | |
# Балансируем нитратный азот | |
no3_needed = self.target["N (NO3-)"] - self.actual["N (NO3-)"] | |
if no3_needed > 0: | |
self._apply_fertilizer("Калий азотнокислый", "N (NO3-)", no3_needed) | |
return self._verify_results() | |
def _apply_fertilizer(self, name, element, target_ppm): | |
if name not in self.fertilizers: | |
raise KeyError(f"Удобрение '{name}' не найдено!") | |
content = self.fertilizers[name].get(element, 0) | |
if content == 0: | |
print(f"ПРЕДУПРЕЖДЕНИЕ: Удобрение '{name}' не содержит элемент '{element}'") | |
return | |
grams = (target_ppm * self.volume) / (content * 1000) | |
self.results[name]['граммы'] += round(grams, self.rounding_precision) | |
for el, val in self.fertilizers[name].items(): | |
added_ppm = (grams * val * 1000) / self.volume | |
if el in self.actual: | |
self.actual[el] += round(added_ppm, self.rounding_precision) | |
def _balance_k_s(self): | |
k_needed = self.target["K"] - self.actual["K"] | |
s_needed = self.target["S"] - self.actual["S"] | |
# Используем "Калий сернокислый" для баланса K и S | |
if k_needed > 0 and s_needed > 0: | |
k_fraction = self.fertilizers["Калий сернокислый"].get("K", 0) | |
s_fraction = self.fertilizers["Калий сернокислый"].get("S", 0) | |
if k_fraction == 0 or s_fraction == 0: | |
print("ПРЕДУПРЕЖДЕНИЕ: Удобрение 'Калий сернокислый' содержит нулевые значения!") | |
return | |
k_from_k2so4 = min(k_needed, s_needed * k_fraction / s_fraction) | |
self._apply_fertilizer("Калий сернокислый", "K", k_from_k2so4) | |
# Оставшийся калий добавляем через "Калий азотнокислый" | |
remaining_k = self.target["K"] - self.actual["K"] | |
if remaining_k > 0: | |
self._apply_fertilizer("Калий азотнокислый", "K", remaining_k) | |
def _distribute_calcium(self): | |
ca_target = self.target["Ca"] # Общий кальций | |
# Весь кальций добавляется через "Кальциевую селитру" | |
if ca_target > 0: | |
self._apply_fertilizer("Кальциевая селитра", "Ca", ca_target) | |
def _verify_results(self): | |
deficits = {} | |
for el in self.target: | |
diff = self.target[el] - self.actual[el] | |
if abs(diff) > 0.1: | |
deficits[el] = round(diff, self.rounding_precision) | |
return { | |
'fertilizers': {k: round(v['граммы'], self.rounding_precision) for k, v in self.results.items()}, | |
'actual_profile': {k: round(v, self.rounding_precision) for k, v in self.actual.items()}, | |
'deficits': deficits, | |
'total_ppm': round(sum(self.actual.values()), self.rounding_precision) | |
} | |
def generate_report(self, results): | |
fert_table = [] | |
for name, data in results['fertilizers'].items(): | |
fert_table.append([name, f"{data} г"]) | |
element_table = [] | |
for el in sorted(self.target.keys()): | |
if el == "CaCl": | |
continue | |
element_table.append([ | |
el, | |
f"{self.target[el]} ppm", | |
f"{results['actual_profile'][el]} ppm", | |
f"{results['actual_profile'][el] - self.target[el]:+.2f} ppm" | |
]) | |
report = "РЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:\n" | |
report += tabulate(fert_table, headers=["Удобрение", "Количество"], tablefmt="grid") | |
report += "\n\nБАЛАНС ЭЛЕМЕНТОВ:\n" | |
report += tabulate(element_table, | |
headers=["Элемент", "Цель", "Факт", "Отклонение"], | |
tablefmt="grid") | |
report += f"\n\nОбщая концентрация: {results['total_ppm']} ppm" | |
if results['deficits']: | |
report += "\n\nВНИМАНИЕ: Обнаружены небольшие отклонения:" | |
for el, diff in results['deficits'].items(): | |
report += f"\n- {el}: не хватает {abs(diff)} ppm" | |
return report | |
# Извлекаем данные из INPUT_DATA | |
fertilizer_constants = INPUT_DATA["fertilizerConstants"] | |
profile_settings = INPUT_DATA["profileSettings"] | |
liters = profile_settings["liters"] | |
rounding_precision = 3 | |
# Создаем калькулятор | |
calculator = NutrientCalculator( | |
fertilizer_constants=fertilizer_constants, | |
profile_settings=profile_settings, | |
liters=liters, | |
rounding_precision=rounding_precision | |
) | |
# Запуск расчета | |
results = calculator.calculate() | |
print(calculator.generate_report(results)) | |
def handle_calculation(): | |
try: | |
# 1. Получаем и парсим данные | |
data = request.get_json() | |
if not data: | |
return jsonify({"error": "No JSON data received"}), 400 | |
# Логируем входные данные для отладки | |
print("\n=== ВХОДНЫЕ ДАННЫЕ ===") | |
print(json.dumps(data, indent=2, ensure_ascii=False)) | |
# 2. Извлекаем основные параметры | |
fertilizer_constants = data.get("fertilizerConstants", {}) | |
profile_settings = data.get("profileSettings", {}) | |
# Проверяем наличие необходимых полей | |
if not fertilizer_constants or not profile_settings: | |
return jsonify({"error": "Missing fertilizerConstants or profileSettings"}), 400 | |
# 3. Извлекаем дополнительные параметры | |
liters = profile_settings.get("liters", 100) # Объем раствора (литры) | |
rounding_precision = profile_settings.get("rounding_precision", 3) # Точность округления | |
# 4. Создаем и запускаем калькулятор | |
calculator = NutrientCalculator( | |
fertilizer_constants=fertilizer_constants, | |
profile_settings=profile_settings, | |
liters=liters, | |
rounding_precision=rounding_precision | |
) | |
results = calculator.calculate() | |
# 5. Формируем дополнительные данные | |
element_contributions = {} | |
for fert_name in calculator.fertilizers.keys(): | |
grams = calculator.results[fert_name]['граммы'] | |
element_contributions[fert_name] = {} | |
for element, percent in calculator.fertilizers[fert_name].items(): | |
added_ppm = (grams * percent * 1000) / calculator.volume | |
element_contributions[fert_name][element] = round(added_ppm, rounding_precision) | |
# 6. Формируем полный ответ | |
response = { | |
"fertilizers": results['fertilizers'], | |
"actual_profile": results['actual_profile'], | |
"deficits": results['deficits'], | |
"total_ppm": results['total_ppm'], | |
"element_contributions": element_contributions, | |
"nitrogen_ratios": { | |
"NH4_RATIO": 1, | |
"NO3_RATIO": profile_settings.get("NO3_RAT", 0), | |
"TOTAL_NITROGEN": profile_settings.get("TOTAL_NITROG", 0) | |
} | |
} | |
return jsonify(response), 200 | |
except Exception as e: | |
error_msg = f"Server error: {str(e)}" | |
print(error_msg) | |
return jsonify({ | |
"error": error_msg, | |
"fertilizers": {}, | |
"actual_profile": {}, | |
"deficits": {}, | |
"element_contributions": {}, | |
"total_ppm": 0 | |
}), 500 | |
if __name__ == '__main__': | |
app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860))) | |