Update app.py

app.py

@@ -14,11 +14,11 @@ import string
 app = Flask(__name__)
 app.secret_key = os.getenv("FLASK_SECRET_KEY", "supersecretkey")
 DATA_FILE = 'cloud_data.json'
-REPO_ID = "Eluza133/Z1e1u"  # Обновленный репозиторий
+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"
-MAX_STORAGE_GB = 500  # 500 ГБ для каждого пользователя
+MAX_STORAGE_GB = 500
 
 cache = Cache(app, config={'CACHE_TYPE': 'simple'})
 logging.basicConfig(level=logging.INFO)
@@ -35,7 +35,6 @@ def load_data():
                 return {'users': {}, 'files': {}}
             data.setdefault('users', {})
             data.setdefault('files', {})
-            # Обновляем структуру данных для существующих пользователей
             for token, user_data in data['users'].items():
                 if 'folders' not in user_data:
                     files = user_data.get('files', [])
@@ -48,9 +47,8 @@ def load_data():
                     }
                     if 'files' in user_data:
                         del user_data['files']
-                # Добавляем storage_used, если его нет
                 if 'storage_used' not in user_data:
-                    user_data['storage_used'] = 0  # В байтах
+                    user_data['storage_used'] = 0
             logging.info("Данные успешно загружены")
             return data
     except Exception as e:
@@ -354,7 +352,7 @@ def register():
             data['users'][token] = {
                 'created_at': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
                 'folders': {'root': {'name': 'root', 'files': [], 'subfolders': {}}},
-                'storage_used': 0  # В байтах
+                'storage_used': 0
             }
             save_data(data)
             flash(f'Ваш токен: {token}. Сохраните его!')
@@ -400,6 +398,7 @@ def login():
         data = load_data()
         if token in data['users'] and len(token) == 13:
             session['token'] = token
+            localStorage.setItem('token', token);  # Сохраняем токен в localStorage
             return redirect(url_for('dashboard'))
         else:
             flash('Неверный токен! Токен должен быть 13 символов.')
@@ -431,13 +430,10 @@ def login():
         <p style="margin-top: 20px;">Нет токена? <a href="{{ url_for('register') }}">Зарегистрируйтесь</a></p>
     </div>
     <script>
-        document.getElementById('login-form').onsubmit = (e) => {
+        document.getElementById('login-form').onsubmit = function(e) {
             const token = e.target.querySelector('input[name="token"]').value;
             localStorage.setItem('token', token);
         };
-        // Автоматический переход, если токен есть
-        const token = localStorage.getItem('token');
-        if (token) window.location = '/dashboard';
     </script>
 </body>
 </html>
@@ -458,8 +454,12 @@ def get_folder(data, token, folder_path):
 @app.route('/dashboard/<path:folder_path>', methods=['GET', 'POST'])
 def dashboard(folder_path='root'):
     if 'token' not in session:
-        flash('Пожалуйста, войдите!')
-        return redirect(url_for('login'))
+        token = request.cookies.get('token') or localStorage.getItem('token')
+        if token and token in load_data()['users']:
+            session['token'] = token
+        else:
+            flash('Пожалуйста, войдите!')
+            return redirect(url_for('login'))
     
     token = session['token']
     data = load_data()
@@ -469,8 +469,8 @@ def dashboard(folder_path='root'):
         return redirect(url_for('login'))
     
     current_folder = get_folder(data, token, folder_path)
-    storage_used = data['users'][token]['storage_used']  # В байтах
-    max_storage_bytes = MAX_STORAGE_GB * 1024 * 1024 * 1024  # 500 ГБ в байтах
+    storage_used = data['users'][token]['storage_used']
+    max_storage_bytes = MAX_STORAGE_GB * 1024 * 1024 * 1024
     storage_percent = (storage_used / max_storage_bytes) * 100 if max_storage_bytes > 0 else 0
     storage_used_gb = storage_used / (1024 * 1024 * 1024)
     storage_remaining_gb = MAX_STORAGE_GB - storage_used_gb
@@ -506,7 +506,7 @@ def dashboard(folder_path='root'):
         
         elif 'upload_files' in request.files:
             files = request.files.getlist('files')
-            total_size = sum(f.content_length for f in files if f)  # Размер в байтах
+            total_size = sum(f.content_length for f in files if f)
             if storage_used + total_size > max_storage_bytes:
                 flash('Недостаточно места для загрузки файлов!')
             else:
@@ -516,7 +516,7 @@ def dashboard(folder_path='root'):
                         temp_path = os.path.join('uploads', filename)
                         os.makedirs('uploads', exist_ok=True)
                         file.save(temp_path)
-                        file_size = os.path.getsize(temp_path)  # Размер в байтах
+                        file_size = os.path.getsize(temp_path)
                         
                         api = HfApi()
                         file_path = f"cloud_files/{token}/{folder_path}/{filename}"
@@ -534,7 +534,7 @@ def dashboard(folder_path='root'):
                             'path': file_path,
                             'type': get_file_type(filename),
                             'upload_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
-                            'size': file_size  # Сохраняем размер файла
+                            'size': file_size
                         }
                         current_folder['files'].append(file_info)
                         data['users'][token]['storage_used'] += file_size
@@ -692,6 +692,10 @@ def dashboard(folder_path='root'):
             };
             xhr.send(formData);
         };
+        window.onload = () => {
+            const token = localStorage.getItem('token');
+            if (!token) return; // Не перенаправляем автоматически
+        };
     </script>
 </body>
 </html>