Update app.py

app.py

@@ -1017,47 +1017,84 @@ def get_order_monitop():
 
 # Функция для валидации подписи ВК приложения
 def is_valid(*, query: dict, secret: str) -> bool:
-    """Check VK Apps signature"""
+    """
+    Проверяет подпись VK приложения.
+    
+    :param query: словарь параметров запроса, включая параметр "sign".
+    :param secret: секретный ключ для подписи.
+    :return: True, если подпись валидна, иначе False.
+    """
     logging.debug(f"Checking signature for query: {query}")
+
+    # Отбираем только параметры, начинающиеся с "vk_" и сортируем их по ключу
     vk_subset = OrderedDict(sorted(x for x in query.items() if x[0][:3] == "vk_"))
     logging.debug(f"Filtered VK params: {vk_subset}")
+    
+    # Формируем строку запроса и вычисляем HMAC с использованием sha256
     hash_code = b64encode(HMAC(secret.encode(), urlencode(vk_subset, doseq=True).encode(), sha256).digest())
+    
+    # Декодируем результат и заменяем символы для соответствия формату VK
     decoded_hash_code = hash_code.decode('utf-8')[:-1].replace('+', '-').replace('/', '_')
     logging.debug(f"Calculated signature: {decoded_hash_code}")
+    
+    # Сравниваем вычисленную подпись с переданной в запросе
     return query.get("sign") == decoded_hash_code
 
-# Чтение ордера по ключу и ВК ИД для приложения
-# Функция для для Чтение ордера для приложения 
+
+# Маршрут для получения информации о заказе по VK ID
 @app.route('/get_order', methods=['POST'])
 def get_order():
+    """
+    Обрабатывает POST-запрос для получения информации о заказе на основе VK ID.
+    
+    :return: JSON ответ с информацией о заказе или ошибке.
+    """
     try:
         logging.debug("Starting get_order")
 
-        # Читаем параметры из POST-запроса
+        # Чтение данных из POST-запроса
         vkid = request.form.get('vk_id', '')
         order = request.form.get('order', '')
-        apps_id = request.form.get('apps_id', '') # Сюда придёт ИД ВК приложения, значение будет ключем для поиска Защищённого ключа в переменной 'api_key_apps_vk'
-        sign = request.form.get('sign', '') # Значение sign полученное при запуске приложения
+        apps_id = request.form.get('apps_id', '')  # Получаем ID VK приложения
+        sign = request.form.get('sign', '')  # Подпись для проверки
 
-        # надо проверку подленности добавить, находим в переменной 'api_key_apps_vk' Защищённый ключ по значению apps_id и проверяем подпись  sign
-        # Если подленная продолжаем действие, если нет сразу возвращаем в ответе not
+        # Проверка наличия обязательных параметров
         if not vkid or not order:
             logging.error("VK ID and order are required")
             return json.dumps({"error": "VK ID and order are required"}), 400
 
+        # Получаем секретный ключ для приложения по его ID
+        secret_key = api_key_apps_vk.get(apps_id)  # api_key_apps_vk - это словарь с ключами для приложений
+        if not secret_key:
+            logging.error(f"App ID {apps_id} not found")
+            return json.dumps({"error": "App ID not found"}), 400
+
+        # Проверка подписи
+        query_params = {
+            'vk_id': vkid,
+            'order': order,
+            'apps_id': apps_id,
+            'sign': sign
+        }
+        if not is_valid(query=query_params, secret=secret_key):
+            logging.error("Invalid signature")
+            return json.dumps({"error": "Invalid signature"}), 400
+
+        # Подключаемся к базе данных
         conn = sqlite3.connect(DATABASE_NEW)
         cursor = conn.cursor()
 
-        # Ищем запись по vk_id
+        # Ищем запись по vk_id в базе данных
         cursor.execute("SELECT orders FROM contacts WHERE vk_id = ?", (vkid,))
         result = cursor.fetchone()
 
-        # Если запись по vk_id не найдена, возвращаем значение "not" для ордера
+        # Если запись не найдена, возвращаем ошибку
         if not result:
             logging.error(f"VK ID {vkid} not found")
             response = {order: 'not'}
             return jsonify(response), 200
 
+        # Загружаем данные заказа из найденной записи
         shop_st = result[0] if result[0] else '{}'
         shop_st_data = json.loads(shop_st)
         logging.debug(f"Existing record found. Loaded JSON: {shop_st_data}")
@@ -1065,7 +1102,7 @@ def get_order():
         # Ищем значение по ключу order
         value = shop_st_data.get(order, 'not')
 
-        # Возвращаем данные из столбца
+        # Возвращаем информацию о заказе
         response = {order: value}
         return jsonify(response), 200