From 036f7c43e54fafe1f138c5af2a4973fbfd2bc811 Mon Sep 17 00:00:00 2001 From: Banyapon Poolsawas Date: Wed, 7 Aug 2024 08:53:58 +0700 Subject: [PATCH] Local Server The try-except block in ServerThread catches potential OSError exceptions, which can occur if there are issues with the socket or port. --- __init__.py | 8 ++-- aframe_exporter.py | 100 ++++++++++++++++++++++++++++++++++++++++++--- 2 files changed, 98 insertions(+), 10 deletions(-) diff --git a/__init__.py b/__init__.py index d7c0f08..8113525 100644 --- a/__init__.py +++ b/__init__.py @@ -4,10 +4,10 @@ bl_info = { "name": "A-Frame Exporter", "author": "Banyapon Poolsawas", - "version": (1, 0, 2), - "blender": (4, 0, 1), - "location": "File > Export", - "description": "Export Blender scene to A-Frame HTML and GLB for WebXR", + "version": (1, 0, 5), + "blender": (4, 0, 0), + "location": "File > Export, Scene Properties > A-Frame Export Settings", + "description": "Export Blender scene to A-Frame HTML and GLB with local server", "warning": "", "category": "Export", } diff --git a/aframe_exporter.py b/aframe_exporter.py index 8e6f19e..191403d 100644 --- a/aframe_exporter.py +++ b/aframe_exporter.py @@ -1,13 +1,58 @@ import bpy import os import shutil +import http.server +import socketserver +import threading +import webbrowser +from bpy.props import BoolProperty, IntProperty from bpy_extras.io_utils import ExportHelper +class ServerThread(threading.Thread): + def __init__(self, directory, port): + super().__init__() + self.directory = directory + self.port = port + self.httpd = None + self.error = None + + def run(self): + os.chdir(self.directory) + + # Error handling for the server + try: + handler = http.server.SimpleHTTPRequestHandler + with socketserver.TCPServer(("", self.port), handler) as httpd: + self.httpd = httpd + print(f"Serving at port {self.port}") + httpd.serve_forever() + except OSError as e: + self.error = e + + def stop(self): + if self.httpd: + self.httpd.shutdown() + class ExportAFrame(bpy.types.Operator, ExportHelper): bl_idname = "export_scene.aframe" bl_label = "Export A-Frame" filename_ext = ".html" + start_server: BoolProperty( + name="Start Local Server", + description="Automatically start a local server after export", + default=False + ) + + port: IntProperty( + name="Port", + description="Port number for the local server", + default=8200, + min=1, + max=65535 + ) + + def execute(self, context): filepath = self.filepath glb_filepath = os.path.splitext(filepath)[0] + ".glb" @@ -15,19 +60,18 @@ def execute(self, context): # Export GLB bpy.ops.export_scene.gltf(filepath=glb_filepath, export_format='GLB') - # Copy favicon.ico if it exists - favicon_path = os.path.join(os.path.dirname(__file__), "icons", "favicon.ico") + # Copy favicon + favicon_path = os.path.join(os.path.dirname(__file__), "icons", "favicon.ico") if os.path.exists(favicon_path): - shutil.copy(favicon_path, os.path.dirname(filepath)) + shutil.copy(favicon_path, os.path.dirname(filepath)) # Create HTML (A-Frame) with open(filepath, "w") as f: f.write(f""" - - + Aframe Exporter @@ -47,8 +91,52 @@ def execute(self, context): - """) + if self.start_server: + self.server_thread = ServerThread(os.path.dirname(filepath), self.port) + self.server_thread.start() + + if self.server_thread.error: # Check for errors + self.report({'ERROR'}, f"Failed to start server: {self.server_thread.error}") + return {'CANCELLED'} + + url = f"http://localhost:{self.port}/{os.path.basename(filepath)}" + else: + url = "file://" + filepath + + webbrowser.open_new_tab(url) return {'FINISHED'} + + def start_local_server(self, directory, port): + os.chdir(directory) + handler = http.server.SimpleHTTPRequestHandler + with socketserver.TCPServer(("", port), handler) as httpd: + print(f"Serving at port {port}") + threading.Thread(target=httpd.serve_forever).start() + + def invoke(self, context, event): + context.window_manager.fileselect_add(self) + return {'RUNNING_MODAL'} + +class ExportAFramePanel(bpy.types.Panel): + bl_label = "A-Frame Export Settings" # Label for the panel + bl_idname = "OBJECT_PT_aframe_export" # Unique identifier for the panel + bl_space_type = 'PROPERTIES' # Panel location (Properties Editor) + bl_region_type = 'WINDOW' # Panel type (window) + bl_context = "scene" # Context for the panel (Scene) + bl_options = {'DEFAULT_CLOSED'} # Panel is closed by default + + def draw(self, context): + layout = self.layout + + # Export Button + layout.operator("export_scene.aframe", text="Export A-Frame") # Add the export operator button + + # Local Server Options + row = layout.row() # Create a new row for the checkbox and button + row.prop(context.scene, "aframe_local_server", text="Start Local Server") # Checkbox to toggle local server + if context.scene.aframe_local_server: # Show port input if the checkbox is checked + row = layout.row() # Create a new row for the port input + row.prop(context.scene, "aframe_server_port", text="Port") # Input field for the port number \ No newline at end of file