absolut import paths / isort

Author: PythonFZ

misc/conftest.py

@@ -3,12 +3,12 @@
 import subprocess
 import time
 from pathlib import Path
-import molify
+
 import ase
-from ase.calculators.singlepoint import SinglePointCalculator
+import molify
 import numpy as np
-
 import pytest
+from ase.calculators.singlepoint import SinglePointCalculator
 from playwright.sync_api import Page, sync_playwright
 
 MISC_DIR = Path(__file__).parent

misc/test_screenshots.py

@@ -3,7 +3,7 @@
 import ase
 
 from zndraw import ZnDraw
-from zndraw.geometries import Floor, Box, Curve, Camera, Sphere, Arrow
+from zndraw.geometries import Arrow, Box, Camera, Curve, Floor, Sphere
 
 
 def test_overview(server, page, capture, bmim_bf4, request):
@@ -509,7 +509,8 @@ def test_chat(server, page, capture, bmim_bf4, request):
 def test_custom_modifier(server, page, capture, bmim_bf4, request):
     """Capture custom modifier registered with vis.register_extension()."""
     from pydantic import Field
-    from zndraw.extensions import Extension, Category
+
+    from zndraw.extensions import Category, Extension
 
     class ScaleAtoms(Extension):
         """Scale atom positions by a factor."""

src/zndraw-mcp/__init__.py

@@ -1,17 +1,17 @@
 import contextlib
 import io
+import subprocess
 import traceback
 
-import plotly.graph_objects as go
-import plotly.express as px
-import pandas as pd
-import numpy as np
 import ase
+import numpy as np
+import pandas as pd
+import plotly.express as px
+import plotly.graph_objects as go
 from fastmcp import FastMCP
-from zndraw.server_manager import get_server_status, ServerInfo
-from zndraw import ZnDraw
-import subprocess
 
+from zndraw import ZnDraw
+from zndraw.server_manager import ServerInfo, get_server_status
 
 # Initialize the MCP Server
 mcp = FastMCP("ZnDraw Bridge")

src/zndraw/app/frame_routes.py

@@ -1011,7 +1011,8 @@ def decode_numpy_recursively(obj):
 
         # Merge updates into existing frame
         # First, decode existing frame values to apply updates
-        from asebytes import decode as asebytes_decode, encode as asebytes_encode
+        from asebytes import decode as asebytes_decode
+        from asebytes import encode as asebytes_encode
 
         # Decode existing frame to ASE Atoms object for easier manipulation
         atoms = asebytes_decode(existing_frame)

src/zndraw/app/geometry_routes.py

@@ -8,9 +8,9 @@
 
 from flask import Blueprint, current_app, request
 
+from zndraw.auth import require_auth
 from zndraw.geometries import geometries as geometry_classes
 from zndraw.server import socketio
-from zndraw.auth import require_auth
 
 from .constants import SocketEvents
 from .redis_keys import RoomKeys

src/zndraw/app/job_routes.py

@@ -7,8 +7,8 @@
 
 from flask import Blueprint, current_app, jsonify, request
 
-from zndraw.server import socketio
 from zndraw.auth import require_auth
+from zndraw.server import socketio
 
 from .job_manager import JobManager, JobStatus
 from .redis_keys import ExtensionKeys

src/zndraw/app/room_routes.py

@@ -11,9 +11,12 @@
 
 from flask import Blueprint, current_app, request
 
-from zndraw.server import socketio
 from zndraw.app.constants import SocketEvents
-from zndraw.auth import require_auth, require_admin
+from zndraw.auth import require_admin, require_auth
+from zndraw.extensions.analysis import analysis
+from zndraw.extensions.modifiers import modifiers
+from zndraw.extensions.selections import selections
+from zndraw.server import socketio
 
 from .frame_index_manager import FrameIndexManager
 from .redis_keys import GlobalIndexKeys, RoomKeys, SessionKeys
@@ -23,9 +26,6 @@
     get_lock_key,
     requires_lock,
 )
-from zndraw.extensions.analysis import analysis
-from zndraw.extensions.modifiers import modifiers
-from zndraw.extensions.selections import selections
 
 log = logging.getLogger(__name__)
 
@@ -54,7 +54,7 @@ def list_rooms():
         }]
     """
     from zndraw.app.room_data_fetcher import BatchedRoomDataFetcher
-    from zndraw.auth import extract_token_from_request, decode_jwt_token, AuthError
+    from zndraw.auth import AuthError, decode_jwt_token, extract_token_from_request
 
     redis_client = current_app.extensions["redis"]
     room_service = current_app.extensions["room_service"]

src/zndraw/app/worker_routes.py

@@ -9,16 +9,14 @@
 
 from flask import Blueprint, current_app, request
 
+from zndraw.app.constants import SocketEvents
+from zndraw.app.redis_keys import ExtensionKeys, FilesystemKeys, SessionKeys
 from zndraw.extensions.analysis import analysis
 from zndraw.extensions.modifiers import modifiers
 from zndraw.extensions.selections import selections
 from zndraw.server import socketio
-
 from zndraw.settings import RoomConfig
 
-from .constants import SocketEvents
-from .redis_keys import ExtensionKeys, FilesystemKeys, SessionKeys
-
 log = logging.getLogger(__name__)
 
 workers = Blueprint("workers", __name__)

src/zndraw/extensions/__init__.py

@@ -1,3 +1,3 @@
-from zndraw.extensions.abc import Extension, Category
+from zndraw.extensions.abc import Category, Extension
 
 __all__ = ["Category", "Extension"]

src/zndraw/extensions/analysis.py

@@ -5,7 +5,7 @@
 import plotly.express as px
 from pydantic import Field
 
-from zndraw.extensions.abc import Extension, Category
+from zndraw.extensions.abc import Category, Extension
 
 if t.TYPE_CHECKING:
     from zndraw import ZnDraw