from flask import Flask, render_template_string, request, redirect, url_for, send_file, flash, jsonify import json import os import logging import threading import time from datetime import datetime from huggingface_hub import HfApi, hf_hub_download from huggingface_hub.utils import RepositoryNotFoundError, HfHubHTTPError from werkzeug.utils import secure_filename from dotenv import load_dotenv import requests import uuid import hmac import hashlib from urllib.parse import unquote, parse_qs load_dotenv() app = Flask(__name__) app.secret_key = 'your_unique_secret_key_soola_cosmetics_67890_no_login' DATA_FILE = 'data.json' USERS_DATA_FILE = 'users.json' SYNC_FILES = [DATA_FILE, USERS_DATA_FILE] REPO_ID = "Kgshop/aiocult" HF_TOKEN_WRITE = os.getenv("HF_TOKEN") HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") BOT_TOKEN = "8337420753:AAEEAIp7k-UL49Tli5-Q6DMSN2OunB3H_CA" STORE_ADDRESS = " Алматы, «Байсат» 1 этаж, синий сектор 117-118 Бутик " CURRENCY_CODE = 'TON' CURRENCY_NAME = 'Toncoin' DOWNLOAD_RETRIES = 3 DOWNLOAD_DELAY = 5 STATUS_MAP_RU = { "new": "Новый", "paid": "Оплачен", "accepted": "Принят", "prepared": "Собран", "shipped": "Отправлен" } logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') def download_db_from_hf(specific_file=None, retries=DOWNLOAD_RETRIES, delay=DOWNLOAD_DELAY): if not HF_TOKEN_READ and not HF_TOKEN_WRITE: logging.warning("HF_TOKEN_READ/HF_TOKEN_WRITE not set. Download might fail for private repos.") token_to_use = HF_TOKEN_READ if HF_TOKEN_READ else HF_TOKEN_WRITE files_to_download = [specific_file] if specific_file else SYNC_FILES logging.info(f"Attempting download for {files_to_download} from {REPO_ID}...") all_successful = True for file_name in files_to_download: success = False for attempt in range(retries + 1): try: logging.info(f"Downloading {file_name} (Attempt {attempt + 1}/{retries + 1})...") local_path = hf_hub_download( repo_id=REPO_ID, filename=file_name, repo_type="dataset", token=token_to_use, local_dir=".", local_dir_use_symlinks=False, force_download=True, resume_download=False ) logging.info(f"Successfully downloaded {file_name} to {local_path}.") success = True break except RepositoryNotFoundError: logging.error(f"Repository {REPO_ID} not found. Download cancelled for all files.") return False except HfHubHTTPError as e: if e.response.status_code == 404: logging.warning(f"File {file_name} not found in repo {REPO_ID} (404). Skipping this file.") if attempt == 0 and not os.path.exists(file_name): try: if file_name == DATA_FILE: with open(file_name, 'w', encoding='utf-8') as f: json.dump({'products': [], 'categories': [], 'orders': {}, 'settings': {'ton_wallet_address': ''}}, f) logging.info(f"Created empty local file {file_name} because it was not found on HF.") elif file_name == USERS_DATA_FILE: with open(file_name, 'w', encoding='utf-8') as f: json.dump([], f) logging.info(f"Created empty local file {file_name} because it was not found on HF.") except Exception as create_e: logging.error(f"Failed to create empty local file {file_name}: {create_e}") success = False break else: logging.error(f"HTTP error downloading {file_name} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...") except requests.exceptions.RequestException as e: logging.error(f"Network error downloading {file_name} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...") except Exception as e: logging.error(f"Unexpected error downloading {file_name} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...", exc_info=True) if attempt < retries: time.sleep(delay) if not success: logging.error(f"Failed to download {file_name} after {retries + 1} attempts.") all_successful = False logging.info(f"Download process finished. Overall success: {all_successful}") return all_successful def upload_db_to_hf(specific_file=None): if not HF_TOKEN_WRITE: logging.warning("HF_TOKEN (for writing) not set. Skipping upload to Hugging Face.") return try: api = HfApi() files_to_upload = [specific_file] if specific_file else SYNC_FILES logging.info(f"Starting upload of {files_to_upload} to HF repo {REPO_ID}...") for file_name in files_to_upload: if os.path.exists(file_name): try: api.upload_file( path_or_fileobj=file_name, path_in_repo=file_name, repo_id=REPO_ID, repo_type="dataset", token=HF_TOKEN_WRITE, commit_message=f"Sync {file_name} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}" ) logging.info(f"File {file_name} successfully uploaded to Hugging Face.") except Exception as e: logging.error(f"Error uploading file {file_name} to Hugging Face: {e}") else: logging.warning(f"File {file_name} not found locally, skipping upload.") logging.info("Finished uploading files to HF.") except Exception as e: logging.error(f"General error during Hugging Face upload initialization or process: {e}", exc_info=True) def periodic_backup(): backup_interval = 1800 logging.info(f"Setting up periodic backup every {backup_interval} seconds.") while True: time.sleep(backup_interval) logging.info("Starting periodic backup...") upload_db_to_hf() logging.info("Periodic backup finished.") def load_data(): default_data = {'products': [], 'categories': [], 'orders': {}, 'settings': {'ton_wallet_address': ''}} try: with open(DATA_FILE, 'r', encoding='utf-8') as file: data = json.load(file) logging.info(f"Local data loaded successfully from {DATA_FILE}") if not isinstance(data, dict): logging.warning(f"Local {DATA_FILE} is not a dictionary. Attempting download.") raise FileNotFoundError if 'products' not in data: data['products'] = [] if 'categories' not in data: data['categories'] = [] if 'orders' not in data: data['orders'] = {} if 'settings' not in data: data['settings'] = {'ton_wallet_address': ''} return data except FileNotFoundError: logging.warning(f"Local file {DATA_FILE} not found. Attempting download from HF.") except json.JSONDecodeError: logging.error(f"Error decoding JSON in local {DATA_FILE}. File might be corrupt. Attempting download.") if download_db_from_hf(specific_file=DATA_FILE): try: with open(DATA_FILE, 'r', encoding='utf-8') as file: data = json.load(file) logging.info(f"Data loaded successfully from {DATA_FILE} after download.") if not isinstance(data, dict): logging.error(f"Downloaded {DATA_FILE} is not a dictionary. Using default.") return default_data if 'products' not in data: data['products'] = [] if 'categories' not in data: data['categories'] = [] if 'orders' not in data: data['orders'] = {} if 'settings' not in data: data['settings'] = {'ton_wallet_address': ''} return data except FileNotFoundError: logging.error(f"File {DATA_FILE} still not found even after download reported success. Using default.") return default_data except json.JSONDecodeError: logging.error(f"Error decoding JSON in downloaded {DATA_FILE}. Using default.") return default_data except Exception as e: logging.error(f"Unknown error loading downloaded {DATA_FILE}: {e}. Using default.", exc_info=True) return default_data else: logging.error(f"Failed to download {DATA_FILE} from HF after retries. Using empty default data structure.") if not os.path.exists(DATA_FILE): try: with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump(default_data, f) logging.info(f"Created empty local file {DATA_FILE} after failed download.") except Exception as create_e: logging.error(f"Failed to create empty local file {DATA_FILE}: {create_e}") return default_data def save_data(data): try: if not isinstance(data, dict): logging.error("Attempted to save invalid data structure (not a dict). Aborting save.") return if 'products' not in data: data['products'] = [] if 'categories' not in data: data['categories'] = [] if 'orders' not in data: data['orders'] = {} if 'settings' not in data: data['settings'] = {'ton_wallet_address': ''} with open(DATA_FILE, 'w', encoding='utf-8') as file: json.dump(data, file, ensure_ascii=False, indent=4) logging.info(f"Data successfully saved to {DATA_FILE}") upload_db_to_hf(specific_file=DATA_FILE) except Exception as e: logging.error(f"Error saving data to {DATA_FILE}: {e}", exc_info=True) def load_users(): default_users = {} try: with open(USERS_DATA_FILE, 'r', encoding='utf-8') as file: users_list = json.load(file) if not isinstance(users_list, list): raise json.JSONDecodeError("Users file is not a list", "", 0) logging.info(f"Local users loaded successfully from {USERS_DATA_FILE}") return {str(user['id']): user for user in users_list} except FileNotFoundError: logging.warning(f"Local file {USERS_DATA_FILE} not found. Attempting download from HF.") except json.JSONDecodeError: logging.error(f"Error decoding JSON in local {USERS_DATA_FILE}. File might be corrupt. Attempting download.") if download_db_from_hf(specific_file=USERS_DATA_FILE): try: with open(USERS_DATA_FILE, 'r', encoding='utf-8') as file: users_list = json.load(file) if not isinstance(users_list, list): logging.error(f"Downloaded {USERS_DATA_FILE} is not a list. Using default.") return default_users logging.info(f"Users loaded successfully from {USERS_DATA_FILE} after download.") return {str(user['id']): user for user in users_list} except FileNotFoundError: logging.error(f"File {USERS_DATA_FILE} still not found after download. Using default.") return default_users except json.JSONDecodeError: logging.error(f"Error decoding JSON in downloaded {USERS_DATA_FILE}. Using default.") return default_users except Exception as e: logging.error(f"Unknown error loading downloaded {USERS_DATA_FILE}: {e}. Using default.", exc_info=True) return default_users else: logging.error(f"Failed to download {USERS_DATA_FILE} from HF. Using empty default.") if not os.path.exists(USERS_DATA_FILE): try: with open(USERS_DATA_FILE, 'w', encoding='utf-8') as f: json.dump([], f) logging.info(f"Created empty local file {USERS_DATA_FILE} after failed download.") except Exception as create_e: logging.error(f"Failed to create empty local file {USERS_DATA_FILE}: {create_e}") return default_users def save_users(users_data): try: if not isinstance(users_data, dict): logging.error("Attempted to save invalid users data (not a dict). Aborting save.") return users_list = list(users_data.values()) with open(USERS_DATA_FILE, 'w', encoding='utf-8') as file: json.dump(users_list, file, ensure_ascii=False, indent=4) logging.info(f"Users data successfully saved to {USERS_DATA_FILE}") upload_db_to_hf(specific_file=USERS_DATA_FILE) except Exception as e: logging.error(f"Error saving users data to {USERS_DATA_FILE}: {e}", exc_info=True) def broadcast_new_product_notification(product, base_url): users = load_users() if not users: return text = ( f"🎉 Новый товар в каталоге! 👋\n\n" f"{product['name']}\n" f"Цена: {product['price']:.4f} {CURRENCY_CODE}\n\n" f"Посмотрите его в каталоге:\n{base_url}" ) url = f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage" params_base = { "text": text, "parse_mode": "HTML" } for user_id in users.keys(): params = params_base.copy() params["chat_id"] = user_id try: response = requests.get(url, params=params) result = response.json() if result.get("ok"): logging.info(f"Sent notification to {user_id}") else: logging.error(f"Error {user_id}: {result.get('description')}") except Exception as e: logging.error(f"Ex {user_id}: {e}") CATALOG_TEMPLATE = '''
{% endif %}
{{ product.get('description', '')[:50] }}{% if product.get('description', '')|length > 50 %}...{% endif %}
Категория: {{ product.get('category', 'Без категории') }}
Цена: {{ "%.4f"|format(product['price']) }} {{ currency_code }}
Описание:
{{ product.get('description', 'Описание отсутствует.')|replace('\\n', '
')|safe }}
Доступные цвета/варианты: {{ colors|select('ne', '')|join(', ') }}
{% endif %} {% set models = product.get('models', []) %} {% if models and models|select('ne', '')|list|length > 0 %}Доступные модели/объемы: {{ models|select('ne', '')|join(', ') }}
{% endif %}ИТОГО К ОПЛАТЕ: {{ "%.4f"|format(order.total_price) }} {{ currency_code }}
Для оплаты заказа подключите ваш TON кошелек и подтвердите транзакцию.
Текущий статус вашего заказа: {{ status_map_ru.get(order.status, order.status) }}
Спасибо за покупку!
В данный момент оплата недоступна. Пожалуйста, свяжитесь с администрацией магазина.
Заказ с таким ID не найден.
← Вернуться в каталог {% endif %}
Пока никто не заходил в приложение.
{% endif %}{{ variant_display }}
{% endif %}Активных заказов нет.
{% endif %}Категорий пока нет.
{% endif %}Магазин работает в режиме каталога с оплатой через TON Connect.
Адрес магазина: {{ store_address }}
Валюта: {{ currency_name }} ({{ currency_code }})
Категория: {{ product.get('category', 'Без категории') }}
Цена: {{ "%.4f"|format(product['price']) }} {{ currency_code }}
Описание: {{ product.get('description', 'N/A')[:100] }}{% if product.get('description', '')|length > 100 %}...{% endif %}
{% set colors = product.get('colors', []) %} {% set models = product.get('models', []) %}Цвета: {{ colors|select('ne', '')|join(', ') if colors|select('ne', '')|list|length > 0 else 'Нет' }}
Модели: {{ models|select('ne', '')|join(', ') if models|select('ne', '')|list|length > 0 else 'Нет' }}
Товаров пока нет.
{% endif %}