Update app.py

app.py

@@ -27,18 +27,20 @@ def load_data():
             data = json.load(file)
             if not isinstance(data, dict):
                 logging.warning("Данные не в формате dict, инициализация пустой базы")
-                return {'posts': [], 'users': {}, 'pending_organizations': []}
+                return {'posts': [], 'users': {}, 'pending_organizations': [], 'orders': {}}
             if 'posts' not in data:
                 data['posts'] = []
             if 'users' not in data:
                 data['users'] = {}
             if 'pending_organizations' not in data:
                 data['pending_organizations'] = []
+            if 'orders' not in data:
+                data['orders'] = {}
             logging.info("Данные успешно загружены")
             return data
     except Exception as e:
         logging.error(f"Ошибка загрузки данных: {e}")
-        return {'posts': [], 'users': {}, 'pending_organizations': []}
+        return {'posts': [], 'users': {}, 'pending_organizations': [], 'orders': {}}
 
 def save_data(data):
     try:
@@ -80,7 +82,7 @@ def download_db_from_hf():
         logging.error(f"Ошибка скачивания базы: {e}")
         if not os.path.exists(DATA_FILE):
             with open(DATA_FILE, 'w', encoding='utf-8') as f:
-                json.dump({'posts': [], 'users': {}, 'pending_organizations': []}, f)
+                json.dump({'posts': [], 'users': {}, 'pending_organizations': [], 'orders': {}}, f)
 
 def periodic_backup():
     while True:
@@ -101,6 +103,7 @@ BASE_STYLE = '''
         --shadow: 0 10px 40px rgba(0, 0, 0, 0.15);
         --glass-bg: rgba(255, 255, 255, 0.15);
         --transition: all 0.4s ease;
+        --cart-btn: #10b981;
     }
     * { margin: 0; padding: 0; box-sizing: border-box; }
     body {
@@ -208,6 +211,12 @@ BASE_STYLE = '''
         transform: scale(1.08);
         background: #5439cc;
     }
+    .cart-btn {
+        background: var(--cart-btn);
+    }
+    .cart-btn:hover {
+        background: #0d9f6e;
+    }
     input, textarea, select {
         width: 100%;
         padding: 14px;
@@ -296,6 +305,9 @@ NAV_HTML = '''
             {% if is_authenticated %}
                 <a href="{{ url_for('profile') }}" class="nav-link"><span>👤</span> Профиль ({{ username }})</a>
                 <a href="{{ url_for('upload') }}" class="nav-link"><span>⬆️</span> Загрузить</a>
+                {% if user_type == 'seller' and verified %}
+                    <a href="{{ url_for('orders') }}" class="nav-link"><span>🛒</span> Заказы</a>
+                {% endif %}
                 <a href="{{ url_for('logout') }}" class="nav-link logout-btn"><span>🚪</span> Выйти</a>
             {% else %}
                 <a href="{{ url_for('login') }}" class="nav-link"><span>🔑</span> Войти</a>
@@ -622,6 +634,8 @@ def feed():
     posts = sorted(data.get('posts', []), key=lambda x: datetime.strptime(x['upload_date'], '%Y-%m-%d %H:%M:%S'), reverse=True)
     is_authenticated = 'username' in session
     username = session.get('username', None)
+    user_type = data['users'].get(username, {}).get('type') if username else None
+    verified = data['users'].get(username, {}).get('verified', False) if username else False
 
     search_query = request.form.get('search', '').strip().lower() if request.method == 'POST' else request.args.get('search', '').strip().lower()
     if search_query:
@@ -745,16 +759,21 @@ def feed():
             </div>
             <div class="post-grid">
                 {% for post in posts %}
-                <a href="{{ url_for('post_page', post_id=post['id']) }}" class="post-item">
-                    <video class="post-preview" preload="metadata" muted>
-                        <source src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/videos/{{ post['filename'] }}" type="video/mp4">
-                    </video>
-                    <h2>{{ post['title'] }}</h2>
+                <div class="post-item">
+                    <a href="{{ url_for('post_page', post_id=post['id']) }}">
+                        <video class="post-preview" preload="metadata" muted>
+                            <source src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/videos/{{ post['filename'] }}" type="video/mp4">
+                        </video>
+                        <h2>{{ post['title'] }}</h2>
+                    </a>
                     <p>{{ post['description'] }}</p>
                     <p class="price">Цена: {{ post['price'] }} {{ post['currency'] }}</p>
                     <p>Загрузил: <a href="{{ url_for('user_profile', username=post['uploader']) }}" class="username-link">{{ post['uploader'] }}</a> | {{ post['upload_date'] }}</p>
                     <p class="stats">Просмотров: {{ post['views'] }} | Лайков: {{ post['likes']|length }}</p>
-                </a>
+                    {% if is_authenticated %}
+                        <button class="btn cart-btn" onclick="addToCart('{{ post['id'] }}', '{{ post['title'] }}', '{{ post['price'] }}', '{{ post['currency'] }}', '{{ post['uploader'] }}')">В корзину</button>
+                    {% endif %}
+                </div>
                 {% endfor %}
             </div>
         </div>
@@ -781,6 +800,16 @@ def feed():
                     document.getElementById('imageModal').style.display = 'none';
                 }
             }
+            function addToCart(postId, title, price, currency, uploader) {
+                let cart = JSON.parse(localStorage.getItem('cart') || '[]');
+                if (!cart.some(item => item.postId === postId)) {
+                    cart.push({ postId, title, price, currency, uploader });
+                    localStorage.setItem('cart', JSON.stringify(cart));
+                    alert('Добавлено в корзину!');
+                } else {
+                    alert('Этот товар уже в корзине!');
+                }
+            }
             window.onload = () => {
                 if (localStorage.getItem('theme') === 'dark') document.body.classList.add('dark');
                 const videos = document.querySelectorAll('.post-preview');
@@ -794,7 +823,7 @@ def feed():
     </body>
     </html>
     '''
-    return render_template_string(html, posts=posts, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID, search_query=search_query)
+    return render_template_string(html, posts=posts, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID, search_query=search_query, user_type=user_type, verified=verified)
 
 # Страница публикации
 @app.route('/post/<post_id>', methods=['GET', 'POST'])
@@ -806,6 +835,8 @@ def post_page(post_id):
     
     is_authenticated = 'username' in session
     username = session.get('username', None)
+    user_type = data['users'].get(username, {}).get('type') if username else None
+    verified = data['users'].get(username, {}).get('verified', False) if username else False
     post['views'] = post.get('views', 0) + 1
     save_data(data)
 
@@ -912,6 +943,7 @@ def post_page(post_id):
                         {% if username in post['likes'] %}Убрать лайк{% else %}Лайк{% endif %}
                     </button>
                 </form>
+                <button class="btn cart-btn" onclick="addToCart('{{ post['id'] }}', '{{ post['title'] }}', '{{ post['price'] }}', '{{ post['currency'] }}', '{{ post['uploader'] }}')">В корзину</button>
                 <form method="POST" class="comment-section">
                     <textarea name="comment" placeholder="Оставьте комментарий" rows="4"></textarea>
                     <button type="submit" class="btn">Отправить</button>
@@ -936,6 +968,16 @@ def post_page(post_id):
                 document.body.classList.toggle('dark');
                 localStorage.setItem('theme', document.body.classList.contains('dark') ? 'dark' : 'light');
             }
+            function addToCart(postId, title, price, currency, uploader) {
+                let cart = JSON.parse(localStorage.getItem('cart') || '[]');
+                if (!cart.some(item => item.postId === postId)) {
+                    cart.push({ postId, title, price, currency, uploader });
+                    localStorage.setItem('cart', JSON.stringify(cart));
+                    alert('Добавлено в корзину!');
+                } else {
+                    alert('Этот товар уже в корзине!');
+                }
+            }
             window.onload = () => {
                 if (localStorage.getItem('theme') === 'dark') document.body.classList.add('dark');
             };
@@ -943,7 +985,7 @@ def post_page(post_id):
     </body>
     </html>
     '''
-    return render_template_string(html, post=post, repo_id=REPO_ID, is_authenticated=is_authenticated, username=username)
+    return render_template_string(html, post=post, repo_id=REPO_ID, is_authenticated=is_authenticated, username=username, user_type=user_type, verified=verified)
 
 # Профиль пользователя
 @app.route('/profile', methods=['GET', 'POST'])
@@ -999,6 +1041,27 @@ def profile():
             data['users'][username]['link'] = link
             save_data(data)
             return redirect(url_for('profile'))
+        elif 'checkout' in request.form:
+            cart = json.loads(request.form.get('cart_data', '[]'))
+            if cart:
+                for item in cart:
+                    seller = item['uploader']
+                    if seller not in data['orders']:
+                        data['orders'][seller] = []
+                    order_id = f"order_{len(data['orders'].get(seller, [])) + 1}_{int(time.time())}"
+                    data['orders'][seller].append({
+                        'order_id': order_id,
+                        'buyer': username,
+                        'post_id': item['postId'],
+                        'title': item['title'],
+                        'price': item['price'],
+                        'currency': item['currency'],
+                        'date': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+                        'status': 'pending'
+                    })
+                save_data(data)
+                flash('Заказ успешно оформлен!', 'success')
+            return redirect(url_for('profile'))
 
     html = '''
     <!DOCTYPE html>
@@ -1046,12 +1109,12 @@ def profile():
                 font-size: 1.2em;
                 margin-bottom: 15px;
             }
-            .post-grid {
+            .post-grid, .cart-grid {
                 display: grid;
                 grid-template-columns: repeat(auto-fill, minmax(340px, 1fr));
                 gap: 30px;
             }
-            .post-item {
+            .post-item, .cart-item {
                 background: var(--card-bg-light);
                 backdrop-filter: blur(20px);
                 padding: 25px;
@@ -1059,10 +1122,10 @@ def profile():
                 box-shadow: var(--shadow);
                 transition: var(--transition);
             }
-            body.dark .post-item {
+            body.dark .post-item, body.dark .cart-item {
                 background: var(--card-bg-dark);
             }
-            .post-item:hover {
+            .post-item:hover, .cart-item:hover {
                 transform: translateY(-15px);
             }
             .post-preview {
@@ -1072,7 +1135,7 @@ def profile():
                 border-radius: 15px;
                 margin-bottom: 20px;
             }
-            .post-item h3 {
+            .post-item h3, .cart-item h3 {
                 font-size: 1.5em;
                 margin-bottom: 15px;
             }
@@ -1099,6 +1162,12 @@ def profile():
                 font-weight: 600;
                 color: var(--primary);
             }
+            .remove-cart-btn {
+                background: #ef4444;
+            }
+            .remove-cart-btn:hover {
+                background: #dc2626;
+            }
         </style>
     </head>
     <body>
@@ -1131,6 +1200,9 @@ def profile():
             {% if user_type == 'seller' and verified %}
             <a href="{{ url_for('upload') }}" class="btn">Добавить видео</a>
             {% endif %}
+            <h2>Корзина</h2>
+            <div class="cart-grid" id="cartGrid"></div>
+            <button class="btn" onclick="checkout()" style="margin-top: 20px;">Оформить заказ</button>
             <h2>Ваши видео</h2>
             <div class="post-grid">
                 {% if user_posts %}
@@ -1169,8 +1241,54 @@ def profile():
                     alert('Ссылка скопирована!');
                 });
             }
+            function renderCart() {
+                const cart = JSON.parse(localStorage.getItem('cart') || '[]');
+                const cartGrid = document.getElementById('cartGrid');
+                cartGrid.innerHTML = '';
+                cart.forEach(item => {
+                    const div = document.createElement('div');
+                    div.className = 'cart-item';
+                    div.innerHTML = `
+                        <h3>${item.title}</h3>
+                        <p class="price">Цена: ${item.price} ${item.currency}</p>
+                        <p>Продавец: <a href="/profile/${item.uploader}" class="username-link">${item.uploader}</a></p>
+                        <button class="btn remove-cart-btn" onclick="removeFromCart('${item.postId}')">Удалить</button>
+                    `;
+                    cartGrid.appendChild(div);
+                });
+            }
+            function removeFromCart(postId) {
+                let cart = JSON.parse(localStorage.getItem('cart') || '[]');
+                cart = cart.filter(item => item.postId !== postId);
+                localStorage.setItem('cart', JSON.stringify(cart));
+                renderCart();
+            }
+            function checkout() {
+                const cart = JSON.parse(localStorage.getItem('cart') || '[]');
+                if (cart.length === 0) {
+                    alert('Корзина пуста!');
+                    return;
+                }
+                const form = document.createElement('form');
+                form.method = 'POST';
+                form.action = '/profile';
+                const input = document.createElement('input');
+                input.type = 'hidden';
+                input.name = 'checkout';
+                input.value = 'true';
+                const cartData = document.createElement('input');
+                cartData.type = 'hidden';
+                cartData.name = 'cart_data';
+                cartData.value = JSON.stringify(cart);
+                form.appendChild(input);
+                form.appendChild(cartData);
+                document.body.appendChild(form);
+                form.submit();
+                localStorage.removeItem('cart');
+            }
             window.onload = () => {
                 if (localStorage.getItem('theme') === 'dark') document.body.classList.add('dark');
+                renderCart();
                 const videos = document.querySelectorAll('.post-preview');
                 videos.forEach(video => {
                     video.addEventListener('loadedmetadata', () => {
@@ -1193,6 +1311,7 @@ def user_profile(username):
     
     user_posts = sorted([p for p in data['posts'] if p['uploader'] == username], key=lambda x: datetime.strptime(x['upload_date'], '%Y-%m-%d %H:%M:%S'), reverse=True)
     is_authenticated = 'username' in session
+    current_user = session.get('username')
     total_views = sum(post.get('views', 0) for post in user_posts)
     total_likes = sum(len(post.get('likes', [])) for post in user_posts)
     user_data = data['users'].get(username, {})
@@ -1437,6 +1556,8 @@ def upload():
 
     is_authenticated = 'username' in session
     username = session.get('username', None)
+    user_type = data['users'].get(username, {}).get('type')
+    verified = data['users'].get(username, {}).get('verified', False)
     html = '''
     <!DOCTYPE html>
     <html lang="ru">
@@ -1566,7 +1687,150 @@ def upload():
     </body>
     </html>
     '''
-    return render_template_string(html, username=username, is_authenticated=is_authenticated)
+    return render_template_string(html, username=username, is_authenticated=is_authenticated, user_type=user_type, verified=verified)
+
+# Заказы продавца
+@app.route('/orders', methods=['GET', 'POST'])
+def orders():
+    if 'username' not in session:
+        flash('Войдите, чтобы просмотреть заказы!', 'error')
+        return redirect(url_for('login'))
+    
+    data = load_data()
+    username = session['username']
+    user_data = data['users'].get(username, {})
+    if user_data.get('type') != 'seller' or not user_data.get('verified'):
+        flash('Только проверенные продавцы могут просматривать заказы!', 'error')
+        return redirect(url_for('profile'))
+
+    orders = data['orders'].get(username, [])
+    is_authenticated = 'username' in session
+    user_type = user_data.get('type')
+    verified = user_data.get('verified', False)
+
+    if request.method == 'POST':
+        if 'update_status' in request.form:
+            order_id = request.form.get('order_id')
+            new_status = request.form.get('status')
+            for order in orders:
+                if order['order_id'] == order_id:
+                    order['status'] = new_status
+                    break
+            save_data(data)
+            return redirect(url_for('orders'))
+
+    html = '''
+    <!DOCTYPE html>
+    <html lang="ru">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>Заказы - {{ username }}</title>
+        <link href="https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600&display=swap" rel="stylesheet">
+        <style>
+            ''' + BASE_STYLE + '''
+            .container {
+                max-width: 950px;
+            }
+            h1 {
+                font-size: 2.5em;
+                font-weight: 700;
+                margin-bottom: 40px;
+                background: linear-gradient(45deg, var(--primary), var(--secondary));
+                -webkit-background-clip: text;
+                color: transparent;
+            }
+            .order-grid {
+                display: grid;
+                grid-template-columns: repeat(auto-fill, minmax(340px, 1fr));
+                gap: 30px;
+            }
+            .order-item {
+                background: var(--card-bg-light);
+                backdrop-filter: blur(20px);
+                padding: 25px;
+                border-radius: 20px;
+                box-shadow: var(--shadow);
+                transition: var(--transition);
+            }
+            body.dark .order-item {
+                background: var(--card-bg-dark);
+            }
+            .order-item:hover {
+                transform: translateY(-15px);
+            }
+            .order-item h3 {
+                font-size: 1.5em;
+                margin-bottom: 15px;
+            }
+            .order-item p {
+                margin-bottom: 15px;
+                font-size: 1.1em;
+            }
+            .price {
+                font-weight: 600;
+                color: var(--primary);
+            }
+            .username-link {
+                color: var(--primary);
+                font-weight: 600;
+                text-decoration: none;
+            }
+            .username-link:hover {
+                text-decoration: underline;
+            }
+            select {
+                width: auto;
+                display: inline-block;
+            }
+        </style>
+    </head>
+    <body>
+        <button class="menu-btn" onclick="toggleSidebar()">☰</button>
+        ''' + NAV_HTML + '''
+        <button class="theme-toggle" onclick="toggleTheme()">🌙</button>
+        <div class="container">
+            <h1>Ваши заказы</h1>
+            <div class="order-grid">
+                {% if orders %}
+                    {% for order in orders %}
+                    <div class="order-item">
+                        <h3>{{ order['title'] }}</h3>
+                        <p class="price">Цена: {{ order['price'] }} {{ order['currency'] }}</p>
+                        <p>Покупатель: <a href="{{ url_for('user_profile', username=order['buyer']) }}" class="username-link">{{ order['buyer'] }}</a></p>
+                        <p>Дата: {{ order['date'] }}</p>
+                        <form method="POST">
+                            <input type="hidden" name="order_id" value="{{ order['order_id'] }}">
+                            <select name="status" onchange="this.form.submit()">
+                                <option value="pending" {% if order['status'] == 'pending' %}selected{% endif %}>В ожидании</option>
+                                <option value="processing" {% if order['status'] == 'processing' %}selected{% endif %}>В обработке</option>
+                                <option value="completed" {% if order['status'] == 'completed' %}selected{% endif %}>Завершено</option>
+                            </select>
+                            <input type="hidden" name="update_status" value="true">
+                        </form>
+                    </div>
+                    {% endfor %}
+                {% else %}
+                    <p style="font-size: 1.2em;">У вас пока нет заказов.</p>
+                {% endif %}
+            </div>
+        </div>
+        <script>
+            function toggleSidebar() {
+                document.getElementById('sidebar').classList.toggle('active');
+            }
+            function toggleTheme() {
+                document.body.classList.toggle('dark');
+                localStorage.setItem('theme', document.body.classList.contains('dark') ? 'dark' : 'light');
+            }
+            window.onload = () => {
+                if (localStorage.getItem('theme') === 'dark') document.body.classList.add('dark');
+            };
+        </script>
+    </body>
+    </html>
+    '''
+    return render_template_string(html, username=username, orders=orders, is_authenticated=is_authenticated, user_type=user_type, verified=verified)
 
 # Админ-панель
 @app.route('/admhosto', methods=['GET', 'POST'])
@@ -1576,6 +1840,8 @@ def admin_panel():
     pending_orgs = data.get('pending_organizations', [])
     is_authenticated = 'username' in session
     username = session.get('username', None)
+    user_type = data['users'].get(username, {}).get('type') if username else None
+    verified = data['users'].get(username, {}).get('verified', False) if username else False
 
     if request.method == 'POST':
         if 'delete' in request.form:
@@ -1603,7 +1869,7 @@ def admin_panel():
         posts = [post for post in posts if search_query in post['title'].lower() or search_query in post['description'].lower()]
 
     html = '''
-    <!DOCTYPE html>
+    <!DOCTYPE html
     <html lang="ru">
     <head>
         <meta charset="UTF-8">
@@ -1797,7 +2063,7 @@ def admin_panel():
     </body>
     </html>
     '''
-    return render_template_string(html, posts=posts, pending_orgs=pending_orgs, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID, search_query=search_query)
+    return render_template_string(html, posts=posts, pending_orgs=pending_orgs, is_authenticated=is_authenticated, username=username, repo_id=REPO_ID, search_query=search_query, user_type=user_type, verified=verified)
 
 if __name__ == '__main__':
     backup_thread = threading.Thread(target=periodic_backup, daemon=True)