weflow-cli 1.5.0

package/scripts/fav_server.py

@@ -0,0 +1,392 @@
+#!/usr/bin/env python3
+"""
+本地收藏服务器 — 浏览器点击收藏实时同步到磁盘 收藏/ 文件夹。
+
+用法:
+  python scripts/fav_server.py --date 2026-05-19
+  python scripts/fav_server.py --date 2026-05-19 --port 8765
+
+打开 http://localhost:8765 即可使用，点击 ☆ 收藏按钮实时写入磁盘。
+"""
+
+import sys, os, json
+from pathlib import Path
+from http.server import HTTPServer, SimpleHTTPRequestHandler
+import urllib.request
+
+SCRIPTS_DIR = os.path.dirname(os.path.abspath(__file__))
+PROJECT_ROOT = os.path.dirname(SCRIPTS_DIR)
+SOURCE_ROOT = os.path.join(PROJECT_ROOT, 'output', 'biz-daily')
+
+
+class FavHandler(SimpleHTTPRequestHandler):
+    """处理静态文件 + /api/fav/* 端点。"""
+    date_str = None
+    date_dir = None
+
+    # 确保文本文件以 UTF-8 编码发送，避免浏览器乱码
+    extensions_map = {**SimpleHTTPRequestHandler.extensions_map,
+        '.md': 'text/markdown; charset=utf-8',
+        '.html': 'text/html; charset=utf-8',
+        '.css': 'text/css; charset=utf-8',
+        '.js': 'text/javascript; charset=utf-8',
+        '.json': 'application/json; charset=utf-8',
+        '.txt': 'text/plain; charset=utf-8',
+    }
+
+    def end_headers(self):
+        self.send_header('Cache-Control', 'no-cache, no-store, must-revalidate')
+        self.send_header('Pragma', 'no-cache')
+        self.send_header('Expires', '0')
+        super().end_headers()
+
+    def do_GET(self):
+        try:
+            if self.path.startswith('/api/fav/list'):
+                self._handle_fav_list()
+            elif self.path == '/api/read/list':
+                self._handle_read_list()
+            elif self.path.startswith('/api/notes'):
+                self._handle_notes_get()
+            elif self.path.startswith('/proxy?url='):
+                self._handle_proxy()
+            else:
+                super().do_GET()
+        except (ConnectionResetError, BrokenPipeError, ConnectionAbortedError):
+            pass
+        except Exception:
+            try: self.send_error(500)
+            except: pass
+
+    def do_POST(self):
+        try:
+            if self.path == '/api/fav/toggle':
+                self._handle_fav_toggle()
+            elif self.path == '/api/read/toggle':
+                self._handle_read_toggle()
+            elif self.path == '/api/explain':
+                self._handle_explain()
+            elif self.path.startswith('/api/notes'):
+                self._handle_notes_post()
+            else:
+                self.send_error(404)
+        except (ConnectionResetError, BrokenPipeError, ConnectionAbortedError):
+            pass
+        except Exception:
+            try: self.send_error(500)
+            except: pass
+
+    def handle_one_request(self):
+        """Override to catch any unhandled exceptions at the lowest level."""
+        try:
+            super().handle_one_request()
+        except (ConnectionResetError, BrokenPipeError, ConnectionAbortedError):
+            pass
+        except Exception:
+            pass
+
+    def _handle_proxy(self):
+        """代理微信CDN图片，绕过防盗链Referer检查。"""
+        from urllib.parse import unquote
+        qs = self.path.split('?', 1)[1] if '?' in self.path else ''
+        url = ''
+        for p in qs.split('&'):
+            if p.startswith('url='):
+                url = unquote(p[4:])
+                break
+        if not url or not url.startswith(('http://', 'https://')):
+            self.send_error(400, 'Missing url')
+            return
+        try:
+            req = urllib.request.Request(url, headers={
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+                'Referer': 'https://mp.weixin.qq.com/',
+                'Accept': 'image/webp,image/apng,image/*,*/*;q=0.8',
+            })
+            with urllib.request.urlopen(req, timeout=15) as resp:
+                data = resp.read(10 * 1024 * 1024)  # max 10MB
+                content_type = resp.headers.get('Content-Type', 'image/jpeg')
+            self.send_response(200)
+            self.send_header('Content-Type', content_type)
+            self.send_header('Content-Length', len(data))
+            self.send_header('Cache-Control', 'public, max-age=86400')
+            self.send_header('Access-Control-Allow-Origin', '*')
+            self.end_headers()
+            self.wfile.write(data)
+        except Exception as e:
+            self.send_error(502, f'Proxy failed: {e}')
+
+    def _handle_fav_list(self):
+        favs = self._read_favs()
+        self._send_json(favs)
+
+    def _handle_fav_toggle(self):
+        content_len = int(self.headers.get('Content-Length', 0))
+        body = self.rfile.read(content_len)
+        try:
+            data = json.loads(body.decode('utf-8'))
+        except (json.JSONDecodeError, UnicodeDecodeError):
+            self.send_error(400, 'Invalid JSON')
+            return
+        article_id = data.get('id', '')
+        if not article_id:
+            self.send_error(400, 'Missing id')
+            return
+
+        favs = self._read_favs()
+        if article_id in favs:
+            favs.remove(article_id)
+            action = 'removed'
+        else:
+            favs.append(article_id)
+            action = 'added'
+
+        self._write_favs(favs)
+        self._sync_folder(favs)
+        self._send_json({'ok': True, 'action': action, 'id': article_id, 'count': len(favs)})
+
+    def _read_favs(self):
+        fav_file = Path(self.date_dir) / '.fav_state.json'
+        if fav_file.exists():
+            try:
+                with open(fav_file, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            except Exception:
+                pass
+        return []
+
+    def _write_favs(self, favs):
+        fav_file = Path(self.date_dir) / '.fav_state.json'
+        with open(fav_file, 'w', encoding='utf-8') as f:
+            json.dump(favs, f, ensure_ascii=False, indent=2)
+
+    def _sync_folder(self, favs):
+        """实时同步 收藏/ 文件夹 — 添加/移除文章文件。"""
+        fav_dir = Path(self.date_dir) / '收藏'
+        fav_dir.mkdir(parents=True, exist_ok=True)
+
+        desired_names = set()
+        for rel_path in favs:
+            src = Path(self.date_dir) / rel_path
+            if src.exists():
+                desired_names.add(src.name)
+                link = fav_dir / src.name
+                if not link.exists():
+                    try:
+                        link.symlink_to(os.path.relpath(src, fav_dir))
+                    except OSError:
+                        import shutil
+                        shutil.copy2(src, link)
+
+        # 移除不在收藏列表中的文件
+        for item in fav_dir.iterdir():
+            if item.name not in desired_names:
+                if item.is_symlink() or item.is_file():
+                    item.unlink()
+
+    # ---- 已读状态 ----
+    def _read_run_state(self):
+        f = Path(self.date_dir) / '.read_state.json'
+        if f.exists():
+            try: return json.loads(f.read_text(encoding='utf-8'))
+            except: pass
+        return {}
+    def _write_run_state(self, state):
+        (Path(self.date_dir) / '.read_state.json').write_text(json.dumps(state, ensure_ascii=False, indent=2), encoding='utf-8')
+
+    def _handle_read_list(self):
+        self._send_json(self._read_run_state())
+
+    def _handle_read_toggle(self):
+        content_len = int(self.headers.get('Content-Length', 0))
+        body = self.rfile.read(content_len)
+        try: data = json.loads(body.decode('utf-8'))
+        except: self.send_error(400); return
+        aid = data.get('id', '')
+        if not aid: self.send_error(400); return
+        state = self._read_run_state()
+        if aid in state: del state[aid]; action = 'unread'
+        else: state[aid] = True; action = 'read'
+        self._write_run_state(state)
+        self._send_json({'ok': True, 'action': action, 'id': aid})
+
+    def _handle_explain(self):
+        content_len = int(self.headers.get('Content-Length', 0))
+        body = self.rfile.read(content_len)
+        try:
+            data = json.loads(body.decode('utf-8'))
+        except (json.JSONDecodeError, UnicodeDecodeError):
+            self.send_error(400, 'Invalid JSON')
+            return
+        text = data.get('text', '').strip()
+        context = data.get('context', '').strip()
+        if not text:
+            self.send_error(400, 'Missing text')
+            return
+
+        # 读 DeepSeek API key
+        api_key = os.environ.get('ANTHROPIC_AUTH_TOKEN', '') or os.environ.get('DEEPSEEK_API_KEY', '')
+
+        prompt = f'''你是一个阅读助手。用户正在读一篇文章，有问题要问你。
+
+文章全文（供参考）：
+{context[:8000]}
+
+用户提问："{text}"
+
+回答规则：
+1. 优先用文章内容回答；如果问题与文章相关但文章信息不足，先用文章已有的部分，再用你的知识补充
+2. 如果问题与文章完全无关，直接用自己的知识回答，不要拒绝
+3. 用大白话解释，假设用户是非技术背景，避免术语堆砌
+4. 简短直接，不要啰嗦'''
+
+        payload = json.dumps({
+            'model': 'deepseek-chat',
+            'messages': [
+                {'role': 'system', 'content': '你是一个简洁准确的知识助手，用中文回答。'},
+                {'role': 'user', 'content': prompt}
+            ],
+            'max_tokens': 1000,
+            'temperature': 0.3
+        }).encode('utf-8')
+
+        try:
+            req = urllib.request.Request('https://api.deepseek.com/v1/chat/completions', data=payload)
+            req.add_header('Content-Type', 'application/json; charset=utf-8')
+            req.add_header('Authorization', f'Bearer {api_key}')
+            resp = urllib.request.urlopen(req, timeout=15)
+            result = json.loads(resp.read().decode('utf-8'))
+            explanation = result['choices'][0]['message']['content']
+            self._send_json({'ok': True, 'text': text, 'explanation': explanation})
+        except Exception as e:
+            self._send_json({'ok': False, 'error': f'AI 调用失败: {str(e)}'})
+
+    # ---- 笔记功能 ----
+    def _notes_file(self):
+        return Path(self.date_dir) / 'notes.json'
+
+    def _read_notes(self):
+        f = self._notes_file()
+        if f.exists():
+            try: return json.loads(f.read_text(encoding='utf-8'))
+            except: pass
+        return {}
+
+    def _write_notes(self, notes):
+        self._notes_file().write_text(json.dumps(notes, ensure_ascii=False, indent=2), encoding='utf-8')
+
+    def _handle_notes_get(self):
+        """GET /api/notes?article=xxx → 返回该文章的所有笔记"""
+        qs = self.path.split('?', 1)[1] if '?' in self.path else ''
+        params = {}
+        for p in qs.split('&'):
+            if '=' in p:
+                k, v = p.split('=', 1)
+                params[k] = v
+        article_id = params.get('article', '')
+        notes = self._read_notes()
+        if article_id:
+            self._send_json(notes.get(article_id, []))
+        else:
+            self._send_json(notes)
+
+    def _handle_notes_post(self):
+        """POST /api/notes → 添加笔记, DELETE /api/notes → 删除笔记"""
+        content_len = int(self.headers.get('Content-Length', 0))
+        body = self.rfile.read(content_len)
+        try:
+            data = json.loads(body.decode('utf-8'))
+        except:
+            self.send_error(400); return
+
+        if self.path == '/api/notes/delete':
+            article_id = data.get('article', '')
+            highlight_id = data.get('id', '')
+            if not article_id or not highlight_id:
+                self.send_error(400); return
+            notes = self._read_notes()
+            if article_id in notes:
+                notes[article_id] = [h for h in notes[article_id] if h.get('id') != highlight_id]
+                if not notes[article_id]:
+                    del notes[article_id]
+                self._write_notes(notes)
+            self._send_json({'ok': True})
+        else:
+            # POST /api/notes → add note
+            article_id = data.get('article', '')
+            text = data.get('text', '').strip()
+            note = data.get('note', '').strip()
+            if not article_id or not text:
+                self.send_error(400); return
+            import uuid, time
+            hid = uuid.uuid4().hex[:8]
+            notes = self._read_notes()
+            if article_id not in notes:
+                notes[article_id] = []
+            notes[article_id].append({
+                'id': hid,
+                'text': text,
+                'note': note,
+                'created': time.strftime('%Y-%m-%d %H:%M', time.localtime())
+            })
+            self._write_notes(notes)
+            self._send_json({'ok': True, 'id': hid})
+
+    def _send_json(self, data):
+        body = json.dumps(data, ensure_ascii=False).encode('utf-8')
+        self.send_response(200)
+        self.send_header('Content-Type', 'application/json; charset=utf-8')
+        self.send_header('Content-Length', len(body))
+        self.send_header('Access-Control-Allow-Origin', '*')
+        self.end_headers()
+        self.wfile.write(body)
+
+    def do_OPTIONS(self):
+        self.send_response(200)
+        self.send_header('Access-Control-Allow-Origin', '*')
+        self.send_header('Access-Control-Allow-Methods', 'GET, POST, OPTIONS')
+        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
+        self.end_headers()
+
+    def log_message(self, format, *args):
+        # 简洁日志
+        if '/api/' in str(args[0]):
+            print(f'  {args[0]}')
+
+
+def main():
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+    import argparse
+    parser = argparse.ArgumentParser(description='本地收藏服务器')
+    parser.add_argument('--date', required=True, help='日期 YYYY-MM-DD')
+    parser.add_argument('--port', type=int, default=8765, help='端口 (默认 8765)')
+    args = parser.parse_args()
+
+    date_dir = os.path.join(SOURCE_ROOT, args.date)
+    if not os.path.isdir(date_dir):
+        print(f'[ERROR] 目录不存在: {date_dir}')
+        sys.exit(1)
+
+    # 配置 Handler
+    FavHandler.date_str = args.date
+    FavHandler.date_dir = date_dir
+
+    # 切换到日报目录以提供静态文件服务
+    os.chdir(date_dir)
+
+    server = HTTPServer(('0.0.0.0', args.port), FavHandler)
+    print(f'⭐ 收藏服务器已启动')
+    print(f'   打开: http://localhost:{args.port}')
+    print(f'   日期: {args.date}')
+    print(f'   收藏文件夹: {date_dir}/收藏/')
+    print(f'   Ctrl+C 停止')
+    print()
+    try:
+        server.serve_forever()
+    except KeyboardInterrupt:
+        print('\n已停止')
+        server.server_close()
+
+
+if __name__ == '__main__':
+    main()

package/scripts/fix_topics.py

@@ -0,0 +1,111 @@
+#!/usr/bin/env python3
+"""修复已生成文章的主题分类（关键词 fallback）。用法: python scripts/fix_topics.py 2026-05-19"""
+import sys, os, re
+from pathlib import Path
+
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+from biz_daily import _guess_topic
+
+TOPICS = ['AI', '学术', '新闻', '文学', '投资']
+
+def fix_date(date_str: str):
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+    base = Path('output/biz-daily') / date_str
+    if not base.is_dir():
+        print(f'[ERROR] 目录不存在: {base}')
+        sys.exit(1)
+
+    fixed = 0
+    for topic in TOPICS:
+        topic_dir = base / topic
+        if not topic_dir.is_dir():
+            continue
+        for md_file in topic_dir.glob('*.md'):
+            try:
+                content = md_file.read_text(encoding='utf-8')
+            except:
+                continue
+
+            # Extract current topic from frontmatter
+            m = re.search(r'^topic:\s*(.+)', content, re.MULTILINE)
+            old_topic = m.group(1).strip() if m else ''
+
+            # Extract title
+            title = md_file.stem
+            m2 = re.search(r'^title:\s*"?(.+?)"?\s*$', content, re.MULTILINE)
+            if m2:
+                title = m2.group(1).strip().strip('"').strip("'")
+
+            # Guess topic
+            article = {'title': title, 'account_name': ''}
+            m3 = re.search(r'^source:\s*"?(.+?)"?\s*$', content, re.MULTILINE)
+            if m3:
+                article['account_name'] = m3.group(1).strip().strip('"').strip("'")
+
+            guessed = _guess_topic(article)
+
+            if guessed != old_topic:
+                # Update frontmatter
+                new_content = re.sub(
+                    r'^topic:\s*.+', f'topic: {guessed}',
+                    content, flags=re.MULTILINE
+                )
+                # Update topic folder if needed - move file
+                if guessed != topic:
+                    new_dir = base / guessed
+                    new_dir.mkdir(exist_ok=True)
+                    new_path = new_dir / md_file.name
+                    md_file.rename(new_path)
+                    # Also update the file at new location
+                    new_path.write_text(new_content, encoding='utf-8')
+                    print(f'  [FIX] {title[:50]} -> {guessed} (was {topic})')
+                else:
+                    md_file.write_text(new_content, encoding='utf-8')
+                fixed += 1
+
+    print(f'\n✓ 修复 {fixed} 篇，未变动 {sum(1 for _ in base.rglob("*.md") if _.name != "README.md") - fixed} 篇')
+
+    # Rebuild README
+    from biz_daily import sanitize_filename
+    all_articles = []
+    for topic in TOPICS:
+        topic_dir = base / topic
+        if not topic_dir.is_dir():
+            continue
+        for md_file in sorted(topic_dir.glob('*.md')):
+            content = md_file.read_text(encoding='utf-8')
+            title = md_file.stem
+            source = ''
+            article_time = ''
+            for line in content.split('\n'):
+                if line.startswith('title:'):
+                    title = line.split(':', 1)[1].strip().strip('"').strip("'")
+                elif line.startswith('source:'):
+                    source = line.split(':', 1)[1].strip().strip('"').strip("'")
+            mm = re.search(r'时间：\d{4}-\d{2}-\d{2} (\d{2}:\d{2})', content)
+            if mm: article_time = mm.group(1)
+            all_articles.append({'title': title, 'source': source, 'time': article_time, 'topic': topic, 'file': md_file.name})
+
+    index_lines = [f'# 公众号日报 — {date_str}', '', f'共 {len(all_articles)} 篇推送，按主题分类', '']
+    for topic in TOPICS:
+        group = [a for a in all_articles if a['topic'] == topic]
+        if not group: continue
+        index_lines.append(f'## {topic} ({len(group)}篇)')
+        index_lines.append('')
+        index_lines.append('| # | 时间 | 公众号 | 标题 |')
+        index_lines.append('|---|------|--------|------|')
+        for j, a in enumerate(group):
+            index_lines.append(f'| {j+1} | {a["time"]} | {a["source"]} | [{a["title"]}](./{topic}/{a["file"]}) |')
+        index_lines.append('')
+
+    with open(base / 'README.md', 'w', encoding='utf-8') as f:
+        f.write('\n'.join(index_lines))
+
+    print(f'  README 已重建')
+
+
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print('用法: python scripts/fix_topics.py 2026-05-19')
+        sys.exit(1)
+    fix_date(sys.argv[1])