feat: no-server

Author: dfface

epub_browser/library.py

@@ -13,7 +13,7 @@ def __init__(self, output_dir=None):
         self.books = {}  # 存储所有书籍信息，使用哈希作为键
         self.output_dir = output_dir
 
-        # 创建基础目录用于服务器
+        # 创建基础目录
         if output_dir is not None:
             if os.path.exists(output_dir):
                 try:
@@ -811,3 +811,20 @@ def create_library_home(self):
         library_html = minify_html.minify(library_html, minify_css=True, minify_js=True)
         with open(os.path.join(self.base_directory, 'index.html'), 'w', encoding='utf-8') as f:
             f.write(library_html)
+    
+    def cleanup(self):
+        """清理所有文件"""
+        if self.output_dir is not None:
+            # 用户自己的目录，不要一个全删
+            for book_hash, book_info in self.books.items():
+                temp_dir = book_info['temp_dir']
+                if os.path.exists(temp_dir):
+                    shutil.rmtree(temp_dir)
+                    print(f"Cleaned up book: {book_info['title']}, path: {temp_dir}")
+            os.remove(os.path.join(self.output_dir, "index.html"))
+            return
+        else:
+            # 清理基础目录
+            if os.path.exists(self.base_directory):
+                shutil.rmtree(self.base_directory)
+                print(f"Cleaned up library base directory: {self.base_directory}")

epub_browser/main.py

@@ -21,6 +21,7 @@ def main():
     parser.add_argument('--output-dir', '-o', help='Output directory for converted books')
     parser.add_argument('--keep-files', action='store_true', help='Keep converted files after server stops')
     parser.add_argument('--log', action='store_true', help='Enable log messages')
+    parser.add_argument('--no-server', action='store_true', help='Do not start a server, just generate files')
 
     args = parser.parse_args()
 
@@ -32,15 +33,14 @@ def main():
 
     # 创建图书馆
     library = EPUBLibrary(args.output_dir)
-    server_instance = EPUBServer(library, args.log)
 
     # 添加所有书籍
     success_count = 0
-    # 收集真实的 epub file
+    # 收集所有的 epub file，可能传递了路径需要下钻
     real_epub_files = []
     for filename in args.filename:
-        files = library.epub_file_discover(filename)
-        real_epub_files.extend(files)
+        cur_files = library.epub_file_discover(filename)
+        real_epub_files.extend(cur_files)
     # 构建图书馆
     for filename in tqdm(real_epub_files):
         if library.add_book(filename):
@@ -50,7 +50,12 @@ def main():
         sys.exit(1)
 
     library.create_library_home()
-    
+
+    # 创建服务器
+    if args.no_server:
+        print(f"Files generated in: {library.base_directory}")
+        return
+    server_instance = EPUBServer(library, args.log)
     try:
         server_instance.start_server(
             port=args.port, 
@@ -63,7 +68,7 @@ def main():
     finally:
         server_instance.stop_server()
         if not args.keep_files:
-            server_instance.cleanup()
+            library.cleanup()
 
 
 if __name__ == '__main__':

epub_browser/server.py

@@ -314,24 +314,4 @@ def stop_server(self):
 
     def is_running(self):
         """检查服务器是否正在运行"""
-        return self._is_running
-    
-    def cleanup(self):
-        """清理所有临时文件"""
-        # 先停止服务器
-        if self.is_running():
-            self.stop_server()
-        if self.library.output_dir is not None:
-            # 用户自己的目录，不要一个全删
-            for book_hash, book_info in self.library.books.items():
-                temp_dir = book_info['temp_dir']
-                if os.path.exists(temp_dir):
-                    shutil.rmtree(temp_dir)
-                    print(f"Cleaned up book: {book_info['title']}, path: {temp_dir}")
-            os.remove(os.path.join(self.library.output_dir, "index.html"))
-            return
-        else:
-            # 清理基础目录
-            if os.path.exists(self.library.base_directory):
-                shutil.rmtree(self.library.base_directory)
-                print(f"Cleaned up library base directory: {self.library.base_directory}")
+        return self._is_running