Create app.py

app.py

@@ -0,0 +1,300 @@
+import os
+import time
+import threading
+import requests
+import logging
+from datetime import datetime
+from flask import Flask, render_template_string, request, jsonify
+from flask_sqlalchemy import SQLAlchemy
+
+# ================= 配置区域 =================
+# 优先从环境变量获取数据库连接字符串 (适配 Docker)
+# 如果没有环境变量，则使用默认值 (请修改下方的默认值)
+DEFAULT_DB_URI = 'postgresql://postgres:password@192.168.1.10:5432/alist_sync'
+DB_URI = os.environ.get('DB_URI', DEFAULT_DB_URI)
+
+PORT = int(os.environ.get('PORT', 5000))
+# ===========================================
+
+# 初始化
+app = Flask(__name__)
+app.config['SQLALCHEMY_DATABASE_URI'] = DB_URI
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+app.config['SQLALCHEMY_ENGINE_OPTIONS'] = {'pool_pre_ping': True} # 自动检测断开的连接
+
+db = SQLAlchemy(app)
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# --- 数据库模型 ---
+class AppConfig(db.Model):
+    __tablename__ = 'config'
+    id = db.Column(db.Integer, primary_key=True)
+    url = db.Column(db.String(255), default="http://localhost:5244")
+    username = db.Column(db.String(255), default="admin")
+    password = db.Column(db.String(255), default="")
+    path1 = db.Column(db.String(500), default="") # 源路径
+    path2 = db.Column(db.String(500), default="") # 目标路径
+    interval = db.Column(db.Integer, default=60)  # 分钟
+    auto_sync = db.Column(db.Boolean, default=False)
+
+class AppLog(db.Model):
+    __tablename__ = 'logs'
+    id = db.Column(db.Integer, primary_key=True)
+    timestamp = db.Column(db.DateTime, default=datetime.now)
+    message = db.Column(db.Text)
+
+# --- 工具函数 ---
+def add_log(msg):
+    """写入日志到数据库 (线程安全)"""
+    print(f"[系统日志] {msg}")
+    try:
+        with app.app_context():
+            log = AppLog(message=str(msg), timestamp=datetime.now())
+            db.session.add(log)
+            db.session.commit()
+            # 清理旧日志 (保留最近500条)
+            last_id = db.session.query(db.func.max(AppLog.id)).scalar()
+            if last_id and last_id > 1000:
+                AppLog.query.filter(AppLog.id < (last_id - 500)).delete()
+                db.session.commit()
+    except Exception as e:
+        print(f"日志写入失败: {e}")
+
+def get_token(base_url, username, password):
+    try:
+        res = requests.post(f"{base_url.rstrip('/')}/api/auth/login", 
+                          json={'username': username, 'password': password}, timeout=10)
+        if res.status_code == 200:
+            return res.json().get('data', {}).get('token')
+    except Exception as e:
+        add_log(f"登录失败: {e}")
+    return None
+
+def list_files(base_url, token, path):
+    files = []
+    try:
+        res = requests.post(f"{base_url.rstrip('/')}/api/fs/list", 
+                           headers={'Authorization': token},
+                           json={"path": path, "password": "", "page": 1, "per_page": 0, "refresh": True}, timeout=30)
+        if res.status_code == 200:
+            data = res.json().get('data', {}).get('content', [])
+            if data: files = data
+    except Exception as e:
+        add_log(f"获取列表失败: {path} - {e}")
+    return files
+
+def copy_files_api(base_url, token, src_dir, dst_dir, file_names):
+    try:
+        res = requests.post(f"{base_url.rstrip('/')}/api/fs/copy",
+                           headers={'Authorization': token},
+                           json={"src_dir": src_dir, "dst_dir": dst_dir, "names": file_names}, timeout=30)
+        return res.json()
+    except Exception as e:
+        return {"code": 500, "message": str(e)}
+
+# --- 后台守护线程 ---
+def background_worker():
+    """后台执行自动同步任务"""
+    add_log("✅ 后台守护线程已启动")
+    while True:
+        try:
+            with app.app_context():
+                cfg = AppConfig.query.first()
+                if not cfg: # 初始化配置
+                    cfg = AppConfig()
+                    db.session.add(cfg)
+                    db.session.commit()
+                
+                if not cfg.auto_sync:
+                    time.sleep(10)
+                    continue
+                
+                # 开始任务
+                interval = cfg.interval if cfg.interval > 0 else 60
+                add_log(f"⏳ 开始执行定时同步任务 (间隔: {interval}分钟)...")
+                
+                token = get_token(cfg.url, cfg.username, cfg.password)
+                if token:
+                    src_list = list_files(cfg.url, token, cfg.path1)
+                    dst_list = list_files(cfg.url, token, cfg.path2)
+                    
+                    if src_list:
+                        dst_names = {f['name'] for f in dst_list}
+                        missing_files = [item['name'] for item in src_list if item['name'] not in dst_names]
+                        
+                        if missing_files:
+                            add_log(f"发现 {len(missing_files)} 个缺失文件，开始推送任务...")
+                            # 分批处理，每批20个
+                            batch_size = 20
+                            for i in range(0, len(missing_files), batch_size):
+                                batch = missing_files[i:i+batch_size]
+                                res = copy_files_api(cfg.url, token, cfg.path1, cfg.path2, batch)
+                                if res.get('code') == 200:
+                                    add_log(f"✅ 指令发送成功: {batch}")
+                                else:
+                                    add_log(f"❌ 指令发送失败: {res.get('message')}")
+                        else:
+                            add_log("无差异文件，无需同步")
+                
+                add_log(f"💤 任务结束，休眠 {interval} 分钟")
+                time.sleep(interval * 60)
+
+        except Exception as e:
+            add_log(f"❌ 后台线程异常: {e}")
+            time.sleep(60)
+
+# --- Web 路由 ---
+HTML_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Alist 自动同步 (PostgreSQL版)</title>
+    <link href="https://cdn.bootcdn.net/ajax/libs/twitter-bootstrap/5.1.3/css/bootstrap.min.css" rel="stylesheet">
+    <style>
+        body { background: #f4f7f6; padding-top: 30px; }
+        .log-box { height: 400px; overflow-y: scroll; background: #1e1e1e; color: #00ff00; padding: 10px; font-family: 'Consolas', monospace; font-size: 13px; border-radius: 5px; }
+    </style>
+</head>
+<body>
+<div class="container">
+    <div class="row">
+        <div class="col-lg-8 offset-lg-2">
+            <div class="card shadow">
+                <div class="card-header bg-dark text-white d-flex justify-content-between">
+                    <h5 class="mb-0">Alist 增量同步控制台</h5>
+                    <span class="badge bg-info">DB: PostgreSQL/MySQL</span>
+                </div>
+                <div class="card-body">
+                    <form id="cfgForm">
+                        <div class="row g-3">
+                            <div class="col-md-4"><label class="form-label">Alist URL</label><input type="text" class="form-control" name="url"></div>
+                            <div class="col-md-4"><label class="form-label">用户名</label><input type="text" class="form-control" name="username"></div>
+                            <div class="col-md-4"><label class="form-label">密码</label><input type="password" class="form-control" name="password"></div>
+                            <div class="col-12"><hr></div>
+                            <div class="col-md-6"><label class="form-label">源路径 (Source)</label><input type="text" class="form-control" name="path1" placeholder="/阿里云盘/电影"></div>
+                            <div class="col-md-6"><label class="form-label">目标路径 (Target)</label><input type="text" class="form-control" name="path2" placeholder="/本地存储/电影"></div>
+                            <div class="col-md-6"><label class="form-label">同步间隔 (分钟)</label><input type="number" class="form-control" name="interval"></div>
+                            <div class="col-md-6 d-flex align-items-end">
+                                <div class="form-check form-switch mb-2">
+                                    <input class="form-check-input" type="checkbox" id="auto_sync" name="auto_sync">
+                                    <label class="form-check-label fw-bold" for="auto_sync">开启后台自动同步</label>
+                                </div>
+                            </div>
+                        </div>
+                        <div class="mt-4 text-end">
+                            <button type="button" class="btn btn-success" onclick="saveConfig()">💾 保存配置并应用</button>
+                        </div>
+                    </form>
+                </div>
+            </div>
+
+            <div class="card shadow mt-4">
+                <div class="card-header bg-secondary text-white">
+                    运行日志 <button class="btn btn-sm btn-light float-end" onclick="loadLogs()">刷新</button>
+                </div>
+                <div class="card-body p-0">
+                    <div class="log-box" id="logContainer">加载中...</div>
+                </div>
+            </div>
+        </div>
+    </div>
+</div>
+<script src="https://cdn.bootcdn.net/ajax/libs/jquery/3.6.0/jquery.min.js"></script>
+<script>
+    function loadData() {
+        $.get('/api/config', function(res) {
+            $('input[name="url"]').val(res.url);
+            $('input[name="username"]').val(res.username);
+            $('input[name="password"]').val(res.password);
+            $('input[name="path1"]').val(res.path1);
+            $('input[name="path2"]').val(res.path2);
+            $('input[name="interval"]').val(res.interval);
+            $('#auto_sync').prop('checked', res.auto_sync);
+        });
+        loadLogs();
+    }
+    function saveConfig() {
+        const data = {
+            url: $('input[name="url"]').val(),
+            username: $('input[name="username"]').val(),
+            password: $('input[name="password"]').val(),
+            path1: $('input[name="path1"]').val(),
+            path2: $('input[name="path2"]').val(),
+            interval: parseInt($('input[name="interval"]').val()),
+            auto_sync: $('#auto_sync').is(':checked')
+        };
+        $.ajax({ url: '/api/config', type: 'POST', contentType: 'application/json', data: JSON.stringify(data), 
+            success: function() { alert('保存成功！后台将在下一周期应用新配置。'); loadLogs(); }});
+    }
+    function loadLogs() {
+        $.get('/api/logs', function(res) {
+            let html = '';
+            res.forEach(l => { html += `<div>[${l.timestamp}] ${l.message}</div>`; });
+            $('#logContainer').html(html);
+        });
+    }
+    loadData();
+    setInterval(loadLogs, 10000);
+</script>
+</body>
+</html>
+"""
+
+@app.route('/')
+def index():
+    return render_template_string(HTML_TEMPLATE)
+
+@app.route('/api/config', methods=['GET', 'POST'])
+def api_config():
+    cfg = AppConfig.query.first()
+    if not cfg:
+        cfg = AppConfig()
+        db.session.add(cfg)
+        db.session.commit()
+
+    if request.method == 'POST':
+        data = request.json
+        cfg.url = data.get('url')
+        cfg.username = data.get('username')
+        cfg.password = data.get('password')
+        cfg.path1 = data.get('path1')
+        cfg.path2 = data.get('path2')
+        cfg.interval = data.get('interval')
+        status_change = data.get('auto_sync') != cfg.auto_sync
+        cfg.auto_sync = data.get('auto_sync')
+        db.session.commit()
+        if status_change:
+            add_log(f"用户手动{'开启' if cfg.auto_sync else '关闭'}了自动同步")
+        return jsonify({'success': True})
+    
+    return jsonify({
+        'url': cfg.url, 'username': cfg.username, 'password': cfg.password,
+        'path1': cfg.path1, 'path2': cfg.path2, 
+        'interval': cfg.interval, 'auto_sync': cfg.auto_sync
+    })
+
+@app.route('/api/logs')
+def api_logs():
+    logs = AppLog.query.order_by(AppLog.timestamp.desc()).limit(50).all()
+    return jsonify([{'timestamp': l.timestamp.strftime("%m-%d %H:%M:%S"), 'message': l.message} for l in logs])
+
+# --- 启动逻辑 ---
+if __name__ == '__main__':
+    # 等待数据库准备好 (针对 Docker compose 启动慢的情况)
+    with app.app_context():
+        try:
+            db.create_all()
+            print("✅ 数据库连接成功，表结构已就绪。")
+        except Exception as e:
+            print(f"❌ 数据库连接失败，请检查配置: {e}")
+            # 不退出，让用户看报错信息
+    
+    # 启动后台线程
+    t = threading.Thread(target=background_worker, daemon=True)
+    t.start()
+    
+    # 启动 Flask
+    app.run(host='0.0.0.0', port=PORT, debug=False, use_reloader=False)