1
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2026-03-05 00:12:49 +01:00

feat: added isort to bw-generate-cli

This commit is contained in:
2024-06-08 16:24:52 +02:00
parent 006a0894b8
commit f0391f59c9
3 changed files with 16 additions and 3 deletions

View File

@@ -9,6 +9,7 @@ import sys
from typing import Literal
import black
import isort
from qtpy.QtWidgets import QGraphicsWidget, QWidget
from bec_widgets.utils import BECConnector
@@ -32,7 +33,7 @@ class ClientGenerator:
import enum
from typing import Literal, Optional, overload
from bec_widgets.cli.client_utils import BECGuiClientMixin, RPCBase, rpc_call"""
from bec_widgets.cli.client_utils import RPCBase, rpc_call, BECGuiClientMixin"""
self.content = ""
@@ -126,6 +127,15 @@ class {class_name}(RPCBase):"""
except black.NothingChanged:
formatted_content = full_content
isort.Config(
profile="black",
line_length=100,
multi_line_output=3,
include_trailing_comma=True,
known_first_party=["bec_widgets"],
)
formatted_content = isort.code(formatted_content)
with open(file_name, "w", encoding="utf-8") as file:
file.write(formatted_content)

View File

@@ -23,7 +23,8 @@ dependencies = [
"zmq",
"h5py",
"pyqtdarktheme",
"black",
"black", # needed for bw-generate-cli
"isort", # needed for bw-generate-cli
]
@@ -35,7 +36,6 @@ dev = [
"pytest-xvfb",
"coverage",
"pytest-qt",
"isort",
"fakeredis",
]
pyqt5 = ["PyQt5>=5.9", "PyQtWebEngine>=5.9"]

View File

@@ -1,6 +1,7 @@
from textwrap import dedent
import black
import isort
from bec_widgets.cli.generate_cli import ClientGenerator
@@ -92,6 +93,8 @@ def test_client_generator_with_black_formatting():
generator.header + "\n" + generator.content, mode=black.FileMode(line_length=100)
)
generated_output_formatted = isort.code(generated_output_formatted)
assert expected_output_formatted == generated_output_formatted