e

Paused

App Files Files Community

Eluza133 commited on Mar 8

Commit

a196b95

verified ·

1 Parent(s): 977890e

Update app.py

Browse files

Files changed (1) hide show

app.py +58 -44

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-from flask import Flask, render_template_string, request, redirect, url_for, session, flash, jsonify, send_file
 from flask_caching import Cache
 import json
 import os
@@ -8,8 +8,6 @@ import time
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
-import random
-import string
 import requests
 from io import BytesIO
@@ -19,7 +17,7 @@ DATA_FILE = 'cloud_data.json'
 REPO_ID = "Eluza133/Z1e1u"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
-ADMIN_PASSWORD = "87132morflot"
 cache = Cache(app, config={'CACHE_TYPE': 'simple'})
 logging.basicConfig(level=logging.INFO)
@@ -90,9 +88,6 @@ def periodic_backup():
         upload_db_to_hf()
         time.sleep(1800)
-def generate_token():
-    return ''.join(random.choices(string.ascii_letters + string.digits, k=13))
 def get_file_type(filename):
     video_extensions = ('.mp4', '.mov', '.avi')
     image_extensions = ('.jpg', '.jpeg', '.png', '.gif')
@@ -280,22 +275,35 @@ body.dark .file-item {
 }
 '''
-@app.route('/admhosto', methods=['GET', 'POST'])
 def register():
     if request.method == 'POST':
         password = request.form.get('password')
-        if password == ADMIN_PASSWORD:
-            token = generate_token()
-            data = load_data()
-            data['users'][token] = {
-                'created_at': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
-                'files': []
-            }
-            save_data(data)
-            flash(f'Ваш токен: {token}. Сохраните его!')
             return redirect(url_for('register'))
-        else:
-            flash('Неверный пароль!')
     html = '''
 <!DOCTYPE html>
@@ -318,9 +326,12 @@ def register():
             {% endif %}
         {% endwith %}
         <form method="POST">
             <input type="password" name="password" placeholder="Введите пароль" required>
             <button type="submit" class="btn">Зарегистрироваться</button>
         </form>
     </div>
 </body>
 </html>
@@ -330,13 +341,15 @@ def register():
 @app.route('/', methods=['GET', 'POST'])
 def login():
     if request.method == 'POST':
-        token = request.form.get('token')
         data = load_data()
-        if token in data['users'] and len(token) == 13:
-            session['token'] = token
             return redirect(url_for('dashboard'))
         else:
-            flash('Неверный токен! Токен должен быть 13 символов.')
     html = '''
 <!DOCTYPE html>
@@ -359,10 +372,11 @@ def login():
             {% endif %}
         {% endwith %}
         <form method="POST">
-            <input type="text" name="token" placeholder="Введите ваш токен" required>
             <button type="submit" class="btn">Войти</button>
         </form>
-        <p style="margin-top: 20px;">Нет токена? <a href="{{ url_for('register') }}">Зарегистрируйт��сь</a></p>
     </div>
 </body>
 </html>
@@ -371,18 +385,19 @@ def login():
 @app.route('/dashboard', methods=['GET', 'POST'])
 def dashboard():
-    if 'token' not in session:
         flash('Пожалуйста, войдите!')
         return redirect(url_for('login'))
-    token = session['token']
     data = load_data()
-    if token not in data['users']:
-        session.pop('token', None)
-        flash('Токен недействителен!')
         return redirect(url_for('login'))
-    user_files = data['users'][token]['files']
     if request.method == 'POST':
         files = request.files.getlist('files')
@@ -403,14 +418,14 @@ def dashboard():
                     temp_files.append((temp_path, filename))
             for temp_path, filename in temp_files:
-                file_path = f"cloud_files/{token}/{filename}"
                 api.upload_file(
                     path_or_fileobj=temp_path,
                     path_in_repo=file_path,
                     repo_id=REPO_ID,
                     repo_type="dataset",
                     token=HF_TOKEN_WRITE,
-                    commit_message=f"Загружен файл для {token}"
                 )
                 file_info = {
@@ -419,7 +434,7 @@ def dashboard():
                     'type': get_file_type(filename),
                     'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
                 }
-                data['users'][token]['files'].append(file_info)
                 if os.path.exists(temp_path):
                     os.remove(temp_path)
@@ -442,7 +457,7 @@ def dashboard():
 <body>
     <div class="container">
         <h1>Zues Cloud Dashboard</h1>
-        <p>Токен: {{ token }}</p>
         {% with messages = get_flashed_messages() %}
             {% if messages %}
                 {% for message in messages %}
@@ -552,23 +567,22 @@ def dashboard():
 </body>
 </html>
 '''
-    return render_template_string(html, token=token, user_files=user_files, repo_id=REPO_ID)
 @app.route('/download/<path:file_path>/<filename>')
 def download_file(file_path, filename):
-    if 'token' not in session:
         flash('Пожалуйста, войдите!')
         return redirect(url_for('login'))
-    token = session['token']
     data = load_data()
-    if token not in data['users']:
-        session.pop('token', None)
-        flash('Токен недействителен!')
         return redirect(url_for('login'))
-    # Проверяем, что файл принадлежит пользователю
-    user_files = data['users'][token]['files']
     if not any(file['path'] == file_path for file in user_files):
         flash('У вас нет доступа к этому файлу!')
         return redirect(url_for('dashboard'))
@@ -590,7 +604,7 @@ def download_file(file_path, filename):
 @app.route('/logout')
 def logout():
-    session.pop('token', None)
     return redirect(url_for('login'))
 if __name__ == '__main__':

+from flask import Flask, render_template_string, request, redirect, url_for, session, flash, send_file
 from flask_caching import Cache
 import json
 import os
 from datetime import datetime
 from huggingface_hub import HfApi, hf_hub_download
 from werkzeug.utils import secure_filename
 import requests
 from io import BytesIO
 REPO_ID = "Eluza133/Z1e1u"
 HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
 HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") or HF_TOKEN_WRITE
+REGISTRATION_CODE = "morshenalphacl"
 cache = Cache(app, config={'CACHE_TYPE': 'simple'})
 logging.basicConfig(level=logging.INFO)
         upload_db_to_hf()
         time.sleep(1800)
 def get_file_type(filename):
     video_extensions = ('.mp4', '.mov', '.avi')
     image_extensions = ('.jpg', '.jpeg', '.png', '.gif')
 }
 '''
+@app.route('/register', methods=['GET', 'POST'])
 def register():
     if request.method == 'POST':
+        username = request.form.get('username')
         password = request.form.get('password')
+        code = request.form.get('code')
+        data = load_data()
+        if username in data['users']:
+            flash('Пользователь с таким логином уже существует!')
             return redirect(url_for('register'))
+        if code != REGISTRATION_CODE:
+            flash('Неверный регистрационный код!')
+            return redirect(url_for('register'))
+        if not username or not password:
+            flash('Логин и пароль обязательны!')
+            return redirect(url_for('register'))
+        data['users'][username] = {
+            'password': password,
+            'created_at': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
+            'files': []
+        }
+        save_data(data)
+        flash('Регистрация успешна! Войдите в систему.')
+        return redirect(url_for('login'))
     html = '''
 <!DOCTYPE html>
             {% endif %}
         {% endwith %}
         <form method="POST">
+            <input type="text" name="username" placeholder="Введите логин" required>
             <input type="password" name="password" placeholder="Введите пароль" required>
+            <input type="text" name="code" placeholder="Введите регистрационный код" required>
             <button type="submit" class="btn">Зарегистрироваться</button>
         </form>
+        <p style="margin-top: 20px;">Уже есть аккаунт? <a href="{{ url_for('login') }}">Войдите</a></p>
     </div>
 </body>
 </html>
 @app.route('/', methods=['GET', 'POST'])
 def login():
     if request.method == 'POST':
+        username = request.form.get('username')
+        password = request.form.get('password')
         data = load_data()
+        if username in data['users'] and data['users'][username]['password'] == password:
+            session['username'] = username
             return redirect(url_for('dashboard'))
         else:
+            flash('Неверный логин или пароль!')
     html = '''
 <!DOCTYPE html>
             {% endif %}
         {% endwith %}
         <form method="POST">
+            <input type="text" name="username" placeholder="Введите логин" required>
+            <input type="password" name="password" placeholder="Введите пароль" required>
             <button type="submit" class="btn">Войти</button>
         </form>
+        <p style="margin-top: 20px;">Нет аккаунта? <a href="{{ url_for('register') }}">Зарегистрируйтесь</a></p>
     </div>
 </body>
 </html>
 @app.route('/dashboard', methods=['GET', 'POST'])
 def dashboard():
+    if 'username' not in session:
         flash('Пожалуйста, войдите!')
         return redirect(url_for('login'))
+    username = session['username']
     data = load_data()
+    if username not in data['users']:
+        session.pop('username', None)
+        flash('Пользователь не найден!')
         return redirect(url_for('login'))
+    # Сортировка файлов по убыванию даты загрузки
+    user_files = sorted(data['users'][username]['files'], key=lambda x: x['upload_date'], reverse=True)
     if request.method == 'POST':
         files = request.files.getlist('files')
                     temp_files.append((temp_path, filename))
             for temp_path, filename in temp_files:
+                file_path = f"cloud_files/{username}/{filename}"
                 api.upload_file(
                     path_or_fileobj=temp_path,
                     path_in_repo=file_path,
                     repo_id=REPO_ID,
                     repo_type="dataset",
                     token=HF_TOKEN_WRITE,
+                    commit_message=f"Загружен файл для {username}"
                 )
                 file_info = {
                     'type': get_file_type(filename),
                     'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S')
                 }
+                data['users'][username]['files'].append(file_info)
                 if os.path.exists(temp_path):
                     os.remove(temp_path)
 <body>
     <div class="container">
         <h1>Zues Cloud Dashboard</h1>
+        <p>Пользователь: {{ username }}</p>
         {% with messages = get_flashed_messages() %}
             {% if messages %}
                 {% for message in messages %}
 </body>
 </html>
 '''
+    return render_template_string(html, username=username, user_files=user_files, repo_id=REPO_ID)
 @app.route('/download/<path:file_path>/<filename>')
 def download_file(file_path, filename):
+    if 'username' not in session:
         flash('Пожалуйста, войдите!')
         return redirect(url_for('login'))
+    username = session['username']
     data = load_data()
+    if username not in data['users']:
+        session.pop('username', None)
+        flash('Пользователь не найден!')
         return redirect(url_for('login'))
+    user_files = data['users'][username]['files']
     if not any(file['path'] == file_path for file in user_files):
         flash('У вас нет доступа к этому файлу!')
         return redirect(url_for('dashboard'))
 @app.route('/logout')
 def logout():
+    session.pop('username', None)
     return redirect(url_for('login'))
 if __name__ == '__main__':