spiceditor 0.0.4__py3-none-any.whl

spiceditor/spice_console.py

@@ -0,0 +1,282 @@
+import sys
+
+from PyQt5.QtCore import QTimer, pyqtSignal
+from PyQt5.QtGui import QFont
+from PyQt5.QtWidgets import QWidget, QVBoxLayout, QApplication
+from easyconfig2.easyconfig import EasyConfig2
+from qtconsole.manager import QtKernelManager
+from qtconsole.rich_jupyter_widget import RichJupyterWidget
+from termqt import Terminal
+
+
+# from qtpyTerminal import qtpyTerminal
+
+class SpiceConsole(QWidget):
+    done = pyqtSignal()
+
+    def __init__(self, config):
+        super().__init__()
+        self.keep_code = False
+        self.config = config
+
+    def execute(self, code, clear=True):
+        pass
+
+    def clear(self):
+        pass
+
+    def set_dark_mode(self, value):
+        pass
+
+    def set_font_size(self, size):
+        pass
+
+    def set_config(self, config):
+        self.config = config
+
+    def config_read(self):
+        pass
+
+    def set_keep_code(self, value):
+        self.keep_code = value
+
+    def get_file_extension(self):
+        return ".py"
+
+    def update_config(self, **kwargs):
+        pass
+
+
+class JupyterConsole(SpiceConsole):
+
+    def __init__(self, config):
+        super().__init__(config)
+
+        kernel_manager = QtKernelManager(kernel_name='python3')
+        kernel_manager.start_kernel()
+        kernel_client = kernel_manager.client()
+        kernel_client.start_channels()
+
+        self.jupyter_widget = RichJupyterWidget()
+        font = QFont("Monospace")
+        font.setStyleHint(QFont.TypeWriter)
+        font.setPixelSize(18)
+        self.jupyter_widget.font = font
+
+        # self.jupyter_widget._set_font()
+        self.jupyter_widget.kernel_manager = kernel_manager
+        self.jupyter_widget.kernel_client = kernel_client
+
+        # Customize the prompt
+        self.jupyter_widget.include_other_output = False
+        self.jupyter_widget.banner = ""  # Remove banner
+        self.jupyter_widget.input_prompt = ""  # Remove input prompt
+        self.jupyter_widget.output_prompt = ""  # Remove output prompt
+        self.jupyter_widget.set_default_style(colors='linux')
+
+        self.editor = self.jupyter_widget._control
+
+        layout = QVBoxLayout()
+        layout.addWidget(self.jupyter_widget)
+        self.setLayout(layout)
+        self.timer = QTimer()
+        self.timer.setSingleShot(True)
+        self.timer.timeout.connect(self.done.emit)
+
+    def config_read(self):
+        pass
+
+    def update_config(self, **kwargs):
+        path = self.config.progs_path.get()
+        if path:
+            self.jupyter_widget.execute("cd " + path, hidden=True)
+
+        size = self.config.font_size.get()
+
+        if size>=0:
+            self.set_font_size(size + 10)
+
+    def set_dark_mode(self, value):
+        if value:
+            self.jupyter_widget.set_default_style(colors='linux')
+        else:
+            self.jupyter_widget.set_default_style(colors='lightbg')
+
+    def execute(self, code, clear=True):
+
+
+        # self.jupyter_widget._control.setText("")
+        # def filtering():
+        #     text = self.editor.toPlainText()
+        #     if text.endswith("   ...: "):
+        #         if clear:
+        #             self.editor.clear()
+        #
+        #     # pattern = r"In \[\d+\]:"
+        #     # if re.search(pattern, text[-10:]):
+        #     #     self.timer.stop()
+        #     #     self.timer.start(250)
+        #
+        # self.editor.textChanged.connect(filtering)
+        # # self.jupyter_widget._control.setFocus()
+        # if "input" in code:
+        #     self.jupyter_widget._control.setFocus()
+        # QApplication.processEvents()
+
+        def run():
+            if code.strip():
+                code_reset = code
+                self.jupyter_widget.execute(code_reset, interactive=True)
+                if clear:
+                    self.jupyter_widget._control.clear()
+
+        clearer = '''
+        import sys
+        import os
+        import importlib
+
+        cwd = os.getcwd()
+
+        # Identify modules that are imported from the current directory
+        user_modules = [m for m in sys.modules if hasattr(
+            sys.modules[m], '__file__') and sys.modules[m].__file__ and sys.modules[m].__file__.startswith(cwd)]
+
+        for m in user_modules:
+            print("Reloading ", m, type(m))
+            importlib.reload(sys.modules[m])
+
+
+        '''
+        self.jupyter_widget.execute(clearer, hidden=True)
+        QTimer.singleShot(100, run)
+
+    def clear(self):
+        pass  # self.jupyter_widget.execute("%clear")
+
+    def set_font_size(self, font_size):
+        font = QFont("Monospace")
+        font.setStyleHint(QFont.TypeWriter)
+        font.setPixelSize(font_size)
+        self.jupyter_widget._control.setFont(font)
+
+
+import platform
+
+
+class TermQtConsole(SpiceConsole):
+    def __init__(self, config):
+        super().__init__(config)
+        self.terminal = Terminal(400, 600, font_size=18)
+        layout = QVBoxLayout()
+        layout.addWidget(self.terminal)
+        self.setLayout(layout)
+
+        my_platform = platform.system()
+
+        if my_platform in ["Linux", "Darwin"]:
+            bin = "/bin/bash"
+
+            from termqt import TerminalPOSIXExecIO
+            terminal_io = TerminalPOSIXExecIO(
+                self.terminal.row_len,
+                self.terminal.col_len,
+                bin
+            )
+        elif my_platform == "Windows":
+            bin = "cmd"
+
+            from termqt import TerminalWinptyIO
+            terminal_io = TerminalWinptyIO(
+                self.terminal.row_len,
+                self.terminal.col_len,
+                bin
+            )
+
+            # it turned out that cmd prefers to handle resize by itself
+            # see https://github.com/TerryGeng/termqt/issues/7
+            auto_wrap_enabled = False
+        else:
+            sys.exit(-1)
+
+        self.terminal.enable_auto_wrap(True)
+
+        terminal_io.stdout_callback = self.terminal.stdout
+        self.terminal.stdin_callback = terminal_io.write
+        self.terminal.resize_callback = terminal_io.resize
+        terminal_io.spawn()
+        # self.terminal.input("python")
+
+    def set_config(self, config: EasyConfig2):
+        super().set_config(config)
+        terminal = config.root().addSubSection("Terminal")
+        self.init = terminal.addString("init", pretty="Init command (e.g. python)")
+        self.temp_file = terminal.addString("temp_file", pretty="Temp file name")
+        self.command = terminal.addString("command", pretty="Command")
+        self.file_extension = terminal.addCombobox("file_extension", pretty="File extension", items=[".py", ".pas"])
+
+    def config_read(self):
+        super().config_read()
+        init = self.init.get_value()
+        if init is not None:
+            init += "\n"
+            self.terminal.input(init.encode("utf-8"))
+
+    def execute(self, code, clear=True):
+        temp_file = self.temp_file.get_value()
+        if temp_file is not None and temp_file.strip():
+            with open(temp_file, "w") as f:
+                f.write(code)
+        else:
+            self.terminal.input(code.encode("utf-8"))
+
+        command = self.command.get_value()
+        if command is not None and command.strip():
+            command += "\r\n"
+            self.terminal.input(command.encode("utf-8"))
+
+    def clear(self):
+        self.terminal.input("clear\r\n".encode("utf-8"))
+
+    def get_file_extension(self):
+        return self.file_extension.get_item(self.file_extension.get(0))
+
+    def set_font_size(self, size):
+        try:
+            self.terminal.font_size = int(size * 0.75)
+            font = QFont("Monospace")
+            font.setStyleHint(QFont.Monospace)
+            font.setPointSize(size)
+            self.terminal.set_font(font)
+            # self.terminal.input("clear\r\n".encode("utf-8"))
+        except:
+            pass
+
+    def update_config(self):
+        font_size = self.config.root().get_child("font_size").get_value()
+        if font_size is not None:
+            self.set_font_size(font_size + 10)
+
+    def resizeEvent(self, a0):
+        # TODO: self.terminal.set_canvas_size(self.width(), self.height())
+        pass
+
+# class Console2(SpiceTerminal):
+#     def __init__(self):
+#         super().__init__()
+#         self.terminal = qtpyTerminal()
+#         self.terminal.term.setFont(QFont("Monospace", 14))
+#         self.terminal.term.setMinimumWidth(200)
+#         self.terminal.setMinimumWidth(200)
+#
+#         layout = QVBoxLayout()
+#         layout.addWidget(self.terminal)
+#         self.setLayout(layout)
+#         self.terminal.start()
+#
+#     def execute(self, code, clear=True):
+#         with open("output.pas", "w") as f:
+#             f.write(code)
+#         self.terminal.push("fpc output.pas && ./output\n")
+#
+#     def clear(self):
+#         self.terminal.push("clear\n")

spiceditor/spice_magic_editor.py

@@ -0,0 +1,389 @@
+import random
+import re
+
+import autopep8
+from PyQt5 import QtGui
+from PyQt5.QtCore import pyqtSignal, Qt, QTimer
+from PyQt5.QtGui import QFont, QFontMetrics, QColor, QPainter, QTextCursor
+from PyQt5.QtWidgets import QTextEdit, QHBoxLayout, QScrollBar, QApplication
+
+from spiceditor.line_number_text_edit import LineNumberTextEdit
+from spiceditor.magic_scrollbar import MagicScrollBar
+
+
+class SpiceMagicEditor(QTextEdit):
+    ctrl_enter = pyqtSignal()
+    info = pyqtSignal(str, int, int)
+
+    def __init__(self, highlighter=None, font_size=18):
+        super().__init__()
+        self.highlighter = highlighter
+        self.suggestion = None
+        self.candidates = []
+        self.count = 0
+        self.mode = 0
+        self.code = ""
+        self.delay = 0.01
+        self.autocomplete_words = []
+
+        self.line_number_area = LineNumberTextEdit(self)
+        self.line_number_area.verticalScrollBar().valueChanged.connect(self.verticalScrollBar().setValue)
+        self.line_number_area.document().setDocumentMargin(0)
+
+        self.setContentsMargins(0, 0, 0, 0)
+        self.document().setDocumentMargin(0)
+        self.setViewportMargins(60, 0, 0, 0)
+        self.setLineWrapMode(QTextEdit.NoWrap)
+        self.setPlaceholderText("Write Python code here...")
+
+        self.setHorizontalScrollBar(MagicScrollBar())
+        self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
+
+        self.textChanged.connect(self.text_changed)
+        self.verticalScrollBar().valueChanged.connect(self.line_number_area.verticalScrollBar().setValue)
+        self.horizontalScrollBar().rangeChanged.connect(self.text_changed)
+
+        self.cursorPositionChanged.connect(
+            lambda: self.line_number_area.highlight_line(self.textCursor().blockNumber()))
+
+        if self.highlighter:
+            self.highlighter.setDocument(self.document())
+
+        self.set_font_size(font_size)
+
+    def set_font_size(self, font_size):
+        font = QFont("Courier New")
+        # font.setStyleHint(QFont.TypeWriter)
+        font.setPixelSize(font_size)
+        self.setFont(font)
+        self.line_number_area.setFont(font)
+        three_numbers_width = QFontMetrics(font).width("000")
+        self.line_number_area.setFixedWidth(int(three_numbers_width + 12))
+        self.setViewportMargins(three_numbers_width + 15, 0, 0, 0)
+
+    def show_code(self):
+        self.show_all_code()
+
+    def set_dark_mode(self, dark):
+        self.line_number_area.set_dark_mode(dark)
+        self.highlighter.set_dark_mode(dark)
+        self.highlighter.setDocument(self.document())
+
+    def set_text(self, text):
+        self.setPlainText(text)
+        self.text_changed()
+
+    def text_changed(self):
+        text = self.toPlainText()
+        lines = text.split("\n")
+        v1 = self.line_number_area.verticalScrollBar().value()
+        self.line_number_area.blockSignals(True)
+        self.line_number_area.clear()
+        text = str()
+        for i in range(len(lines)):
+            text += "{:3d}\n".format(i + 1)
+
+        self.line_number_area.setPlainText(text)
+        self.line_number_area.setFixedHeight(self.height())
+        self.line_number_area.verticalScrollBar().setMaximum(self.verticalScrollBar().maximum())
+        self.line_number_area.verticalScrollBar().setValue(v1)
+        self.line_number_area.blockSignals(False)
+
+    def resizeEvent(self, a0) -> None:
+        self.text_changed()
+        self.blockSignals(True)
+        super().resizeEvent(a0)
+        self.blockSignals(False)
+
+    def format_code(self):
+        self.format_code()
+
+    def get_text(self):
+        return self.toPlainText()
+
+    def clear(self):
+        self.set_code("")
+        self.setPlainText("")
+        self.count = 0
+        self.set_mode(0)
+
+    def get_code(self):
+        return self.code
+
+    def get_remaining_chars(self):
+        diff = len(self.get_code()) - self.count
+        return diff
+
+    def set_delay(self, delay):
+        self.delay = delay
+
+    def append_autocomplete(self, words, clear=False):
+        if clear:
+            self.autocomplete_words.clear()
+        self.autocomplete_words += words if words else ""
+
+    def set_code(self, code):
+        self.setText("")
+        self.count = 0
+        self.code = code
+        self.set_mode(1)
+        self.setFocus()
+
+    def set_mode(self, mode):
+        self.mode = mode
+        self.setCursorWidth(3 if self.mode == 1 else 1)
+        # self.setReadOnly(self.mode == 1)
+        self.update()
+
+    def on_return_key(self, e):
+        return False
+
+    def complete_line(self, sleep=True):
+        self.info.emit(self.get_next_line(), self.get_remaining_chars(), 20)
+        if self.count < len(self.code):
+            # self.setText(self.toPlainText() + self.code[self.count])
+            self.insertPlainText(self.code[self.count])
+            self.moveCursor(QtGui.QTextCursor.End)
+            self.count += 1
+
+            if self.code[self.count - 1] == "\n":
+                # if next line is empty continue
+                # and show that line too
+                if len(self.get_rest_of_line()) > 0:
+                    return True
+
+            QApplication.processEvents()
+            delay = int(self.delay) + random.randint(0, int(self.delay))
+            QTimer.singleShot(delay, self.complete_line)
+
+    def get_rest_of_line(self):
+        count = self.count
+        text = ""
+        while count < len(self.code):
+            text += self.code[count]
+            count += 1
+            if self.code[count - 1] == "\n":
+                return text[1:]
+        return ""
+
+    def get_spaces(self, line):
+        spaces = 0
+        for c in line:
+            if c == " ":
+                spaces += 1
+            else:
+                break
+        return spaces
+
+    def get_current_line(self):
+        cursor = self.textCursor()
+        cursor.movePosition(QtGui.QTextCursor.StartOfLine)
+        cursor.movePosition(QtGui.QTextCursor.EndOfLine, QtGui.QTextCursor.KeepAnchor)
+        current_line = cursor.selectedText()
+        return current_line
+
+    def append_next_char(self):
+        self.count += 1
+        self.setText(self.code[:self.count])
+        self.moveCursor(QtGui.QTextCursor.End)
+
+    def show_all_code(self):
+        self.setText(self.code)
+        self.set_mode(0)
+        self.moveCursor(QtGui.QTextCursor.End)
+
+    def get_current_line_text(self):
+        # Get the QTextCursor
+        cursor = self.textCursor()
+
+        # Move the cursor to the start and end of the current line
+        cursor.select(cursor.LineUnderCursor)
+
+        # Get the selected text
+        return cursor.selectedText()
+
+    def keyPressEvent(self, e: QtGui.QKeyEvent) -> None:
+        self.setFocusPolicy(Qt.StrongFocus)
+
+        if e.key() == Qt.Key_Escape:
+            self.set_mode(0 if self.mode == 1 else 0)
+
+        if self.mode == 1:
+
+            if e.key() == Qt.Key_Down:
+                self.show_all_code()
+            elif e.key() == Qt.Key_Control:
+                return
+            elif e.key() == Qt.Key_End:
+                while self.complete_line(False):
+                    pass
+                self.set_mode(0)
+            elif e.key() == Qt.Key_Tab:
+                if e.modifiers() == Qt.ControlModifier:
+                    while self.complete_line():
+                        pass
+                    self.set_mode(0)
+                else:
+                    self.complete_line()
+            elif self.count < len(self.code):
+                self.info.emit(self.get_rest_of_line(), self.get_remaining_chars(), 1000)
+                self.append_next_char()
+            elif e.key() == Qt.Key_Return:
+                self.set_mode(0)
+                super().keyPressEvent(e)
+            elif e.key() == Qt.Key_Backspace:
+                self.set_mode(0)
+            else:
+                self.setText(self.toPlainText() + "\n")
+                self.moveCursor(QtGui.QTextCursor.End)
+
+        elif self.mode == 0:
+
+            if e.key() == Qt.Key_Tab:
+                self.tab_pressed()
+            elif e.key() == Qt.Key_Backspace:
+                self.suggestion = None
+                if self.get_current_line_text().endswith("    "):
+                    for i in range(4):
+                        self.textCursor().deletePreviousChar()
+                else:
+                    super().keyPressEvent(e)
+            elif e.key() == Qt.Key_Return:
+                self.suggestion = None
+                if e.modifiers() == Qt.ControlModifier:
+                    self.ctrl_enter.emit()
+                elif self.on_return_key(e):
+                    pass
+                else:
+                    super().keyPressEvent(e)
+            else:
+                self.suggestion = None
+                super().keyPressEvent(e)
+        self.cursorPositionChanged.emit()
+
+    def indent_selected(self):
+        cursor = self.textCursor()
+
+        if not cursor.hasSelection():
+            return  # No selection, do nothing
+
+        # Get selected text (preserve newlines)
+        selected_text = cursor.selection().toPlainText()
+
+        # Add 4 spaces to each line
+        indented_text = "\n".join("    " + line for line in selected_text.splitlines())
+
+        # Replace selected text with indented version
+        cursor.beginEditBlock()  # Start undo-able action
+        cursor.insertText(indented_text)
+        cursor.endEditBlock()  # End undo-able action
+
+    def get_text_before_cursor(self):
+        cursor = self.textCursor()
+        cursor.movePosition(QTextCursor.StartOfBlock, QTextCursor.KeepAnchor)  # Selecciona desde el inicio de la línea
+        return cursor.selectedText()
+
+    def tab_pressed(self):
+
+        if self.textCursor().hasSelection():
+            self.indent_selected()
+            return
+
+        # Let see if we have some autocomplete candidates
+        if self.suggestion is None:
+            current_words = re.split(r'\W+', self.toPlainText())
+            text_before_cursor = self.get_text_before_cursor()
+            words_before_cursos = re.split(r"[+\-*/= ]", text_before_cursor)
+            self.candidates = []
+            if words_before_cursos[-1] != "":
+                word_set = list(set(current_words + self.highlighter.get_keywords() + self.autocomplete_words))
+                self.candidates = [word for word in word_set if word.startswith(words_before_cursos[-1])]
+                if words_before_cursos[-1] in self.candidates:
+                    self.candidates.remove(words_before_cursos[-1])
+                    self.candidates.append(words_before_cursos[-1])
+                self.suggestion = words_before_cursos[-1]
+
+        if len(self.candidates) > 1:
+            # Remove the current suggestion
+            for _ in range(len(self.suggestion)):
+                self.textCursor().deletePreviousChar()
+            self.candidates.append(self.suggestion)
+            self.suggestion = self.candidates.pop(0)
+            self.insertPlainText(self.suggestion)
+        else:
+            self.insertPlainText("    ")
+
+
+
+        # if len(self.candidates) > 0:
+
+        # current_line = self.get_current_line_text()
+        #
+        # # it was just a tab
+        # if len(current_line) == 0 or current_line.endswith("    "):
+        #     self.insertPlainText("    ")
+        #     self.moveCursor(QtGui.QTextCursor.End)
+        #     self.suggestion = None
+        #     return
+        #
+        # # it was just a tab
+        # if len(current_line) > 0 and current_line[-1] in " (:)":
+        #     self.suggestion = None
+        #     return
+        #
+        # if self.suggestion is None:
+        #
+        #     if len(self.candidates) == 0:
+        #         self.insertPlainText("    ")
+        #         return
+        #     elif len(self.candidates) == 1:
+        #         self.insertPlainText(self.candidates[0][len(unfinished_word):])
+        #         self.moveCursor(QtGui.QTextCursor.End)
+        #         return
+        #     else:
+        #         self.suggestion = unfinished_word
+        #
+        # for _ in range(len(self.suggestion)):
+        #     self.textCursor().deletePreviousChar()
+        # self.candidates.append(self.suggestion)
+        # self.suggestion = self.candidates.pop(0)
+        # self.insertPlainText(self.suggestion)
+
+    def get_next_line(self):
+        count = self.count
+        remaining = self.code[count:]
+        remaining = remaining.split("\n")
+        remaining = remaining[1:]
+        remaining = [x for x in remaining if x.strip()]
+        if remaining:
+            return remaining[0]
+
+
+class PascalEditor(SpiceMagicEditor):
+    pass
+
+
+class PythonEditor(SpiceMagicEditor):
+    def format_code(self):
+        code = self.toPlainText()
+        if not code.endswith("\n"):
+            code += "\n"
+        self.setPlainText(autopep8.fix_code(code))
+        self.moveCursor(QtGui.QTextCursor.End)
+
+    def on_return_key(self, e):
+        current_line = self.get_current_line()
+        spaces = self.get_spaces(current_line)
+        if current_line.endswith(":"):
+            if self.textCursor().positionInBlock() == len(current_line):
+                self.insertPlainText("\n" + " " * (spaces + 4))
+                return True
+            return False
+
+        elif current_line.startswith(" "):
+            if current_line.strip():
+                self.insertPlainText("\n" + " " * spaces)
+            else:
+                self.insertPlainText("\n" + " " * (max(spaces - 4, 0)))
+            return True
+        return False