prog_pyqt5.py

import sys
from PyQt5 import QtWidgets, uic,QtCore
from post import PostMachine, Tape
from PyQt5.QtWidgets import QTableWidgetItem, QHBoxLayout, QPushButton
from PyQt5.QtGui import QColor
import tkinter as tk
from tkinter import filedialog


# Загрузка интерфейса из файла PostMachine.ui
class Ui(QtWidgets.QMainWindow):
    
    curentBut=15 #Текущее местоположение каретки
    timeDistanse=1000 #Скорость анимации
    listNum=[]
    
    #Инициализация
    def __init__(self):
        
        super(Ui, self).__init__()
        uic.loadUi("PostMachine.ui", self)
        
        #Присвоение кнопкам методов
        self.pushButton.clicked.connect(self.start_machine)
        self.pushButton_addCommand.clicked.connect(self.data_save)
        self.pushButton_importFile.clicked.connect(self.import_command)
        self.pushButton_saveFile.clicked.connect(self.save_command)
        self.pushButton_Clear.clicked.connect(self.Clear_Table)
        self.pushButton_spravka.clicked.connect(self.get_spravka)
        self.pushButton_resetTape.clicked.connect(self.butCleanTape)
        self.action500.triggered.connect(self.timeDistanse500)
        self.action1000.triggered.connect(self.timeDistanse1000)      
        self.action1500.triggered.connect(self.timeDistanse1500)      
        self.action2000.triggered.connect(self.timeDistanse2000)                 
        
        #Создание ленты
        Ui.createTape(self)
        
        #Настройка таблицы для вывода и ввода команд
        self.tableWidget.setColumnCount(2)
        self.tableWidget.setHorizontalHeaderLabels(("Команда","Номер следующей команды"))
        self.tableWidget.setColumnWidth(0,100)
        self.tableWidget.setColumnWidth(1,250)
    
    #Метод для добавления команд в таблицу    
    def data_save(self):
        
        #Передача данных из ЛайнЭдитов в переменные
        com = self.lineEdit_com.text()
        nextCom = self.lineEdit_comNext.text()
        tchk="."
        row=self.tableWidget.rowCount()+1
        
        #Строка для проверки команд
        txtcom=".><10?"
        
        
        #Нахождение количества строк в таблице
        rowCount = self.tableWidget.rowCount()
        
        #Заполнение таблицы данными
        self.tableWidget.insertRow(rowCount)
        if com != "" and txtcom.find(com)!=-1:
            self.tableWidget.setItem(rowCount,0,QTableWidgetItem(com))
        else:
            self.tableWidget.setItem(rowCount,0,QTableWidgetItem(tchk))
            
        if nextCom !="" and Ui.checknum(self)!=-1:    
            self.tableWidget.setItem(rowCount,1,QTableWidgetItem(nextCom))
        else:
            self.tableWidget.setItem(rowCount,1,QTableWidgetItem(str(row)))
        
        #Переделываем номера строк для удобства    
        Ui.rowsEdit(self)
        
    #Метод для импорта программы из файла        
    def import_command(self):
        
        # Создать корневое окно
        root = tk.Tk()
        root.withdraw()  # Скрыть корневое окно

        # Открыть диалоговое окно выбора файла
        file_path = filedialog.askopenfilename()

        # После выбора файла, можно использовать `file_path` для обработки выбранного файла
        if file_path:
            with open(file_path, 'r') as file:
                lines = file.readlines()
                self.tableWidget.setRowCount(len(lines))
                for i, line in enumerate(lines):
                    items = line.strip().split()  # Предполагаем, что данные разделены табуляцией
                    
                    for j, item in enumerate(items):
                        if (len(item)>=3):
                            item = QTableWidgetItem(Ui.readingStr(item))    #Правильная запись ? как 2, 4 а не 2,4
                        else:
                            item = QTableWidgetItem(item)
                        self.tableWidget.setItem(i, j, item)
                        
        #Переделываем номера строк для удобства                    
        Ui.rowsEdit(self)

        # Закрыть корневое окно (необязательно, если вы завершили работу с диалоговым окном)
        root.destroy()
    
    #Метод для сохранения программы в текстовый файл    
    def save_command(self):
        
        # Укажите имя файла, который вы хотите создать
        file_name = self.lineEdit_FileName.text()
        
        #Проверка имени файла
        if file_name !="" and file_name.endswith(".txt"):
            # Откройте файл для записи (создаст новый файл или перезапишет существующий)
            with open(file_name, 'w', encoding='utf-8') as file:
                #Заполнение файла данными из 
                for row in range(self.tableWidget.rowCount()):
                    row_data = []
                    for column in range(self.tableWidget.columnCount()):
                        item = self.tableWidget.item(row, column)
                        if item is not None:
                            cell_data = item.text()
                            if (len(cell_data)>=4):
                                row_data.append(Ui.writingStr(cell_data))   #Правильная запись ? как 2,4 а не 2, 4
                            else:
                                row_data.append(cell_data)
                        else:
                            row_data.append("")
                    file.write("\t".join(row_data) + "\n")
        else:
            QtWidgets.QMessageBox.warning(self, "Ошибка!", "Вы ввели недопустимое название файла!\n Или же не указали расширение (.txt)")
                    
    #Метод запуска программы
    def start_machine(self):
        
        try:
            #Создаем объект класса
            machine = PostMachine()
            time =0
            
            self.lineEdit_result.setText("")
            
            check= False
            checkStop= False
            postNumStr="0"
            #Добавляем команды из таблицы на выполнение
            for row in range(self.tableWidget.rowCount()):
                item0 = self.tableWidget.item(row,0)
                item1 = self.tableWidget.item(row,1)
                
                
                itemtxt=Ui.listNum[Ui.curentBut]
                
                #Метод проверки команды ?
                com=item0.text()
                if check==False:
                    if str(com)=="?":
                        check=True
                        if itemtxt=="0":
                            text=item1.text()
                            count=text[3]
                            postNumStr=text[3]
                        else:
                            text=item1.text()
                            count=text[0]
                            postNumStr=text[0]
                            
                #Отображение анимации            
                if (check==False and checkStop==False and str(row)==postNumStr) or (check==True and str(row)==count and checkStop==False and str(row)==postNumStr):
                    time=time+ Ui.timeDistanse
                    check=False
                    if str(com)==">":
                        QtCore.QTimer.singleShot(time, self.goRight)
                    elif str(com)=="<":
                        QtCore.QTimer.singleShot(time, self.goLeft)
                    elif str(com)=="1":
                        QtCore.QTimer.singleShot(time, self.setOne)
                        Ui.listNum[Ui.curentBut]="1"
                    elif str(com)=="0":
                        QtCore.QTimer.singleShot(time, self.setZero)
                        Ui.listNum[Ui.curentBut]="0"
                    elif str(com)==".":
                        checkStop=True
                    postNumStr=item1.text()
                comText= (f"{item0.text()} {item1.text()}")
                machine.add_command(comText)
                
            #Запуск программы
            machine.run()
            result = machine.get_tape()
            
            #Выводим результат
            time=time+Ui.timeDistanse
            QtCore.QTimer.singleShot(time, lambda: self.lineEdit_result.setText(result))
        except Exception as e:
            QtWidgets.QMessageBox.warning(self, f"Ошибка!", "Возможно вы не правильно ввели команду.\nПроверьте введённые вами команды.\n"+str({e}))

    
    #Возвращает целочисленный индекс строки
    def return_row_index(self,row):
        if len(row)>1:
            return 1
        else:
            return int(row)
                   
    #Метод для очистки таблицы и ленты
    def Clear_Table(self):
        
        # Очистить содержимое ячеек
        self.tableWidget.clearContents()  
        
        # Установить количество строк в 0
        self.tableWidget.setRowCount(0) 
        
        #Очистка ленты
        Ui.butCleanTape(self)
        self.lineEdit_result.setText("")
    
    #Метод для создания ленты    
    def createTape(self):
        hbox=self.horizontalLayout
        hbox2= self.horizontalLayout_2
        hbox3= self.horizontalLayout_3
        k=-15
        for i in range(31):
            if(k==0):
                button = QPushButton("*")
                button.setStyleSheet("background-color: red; color: white;")
                hbox3.addWidget(button)
            else:
                hbox3.addWidget(QPushButton(""))
            hbox.addWidget(QPushButton("0"))
            Ui.listNum.append("0")
            hbox2.addWidget(QPushButton(str(k)))
            k=k+1
        
        self.setLayout(hbox)
        self.setLayout(hbox2)
        self.setLayout(hbox3)
        Ui.curentBut=15
    
    #Метод для ресета ленты
    def butCleanTape(self):
        hbox=self.horizontalLayout
        hbox2= self.horizontalLayout_2
        hbox3= self.horizontalLayout_3
        Ui.cleanTape(self,hbox)
        Ui.cleanTape(self,hbox2)
        Ui.cleanTape(self,hbox3)
        Ui.createTape(self)
        self.lineEdit_result.setText("")
        
    #Метод для очистки ленты
    def cleanTape(self,hbox):
        while hbox.count() > 0:
            item = hbox.takeAt(0)  # Извлекаем элемент с индексом 0
            widget = item.widget()  # Получаем виджет из элемента
            Ui.listNum.clear()
            if widget is not None:
                widget.deleteLater()  # Удаляем виджет
            else:
                hbox.removeItem(item)  # Удаляем элемент из макета
    
    #Метод перехода каретки вправо            
    def goRight(self):

        hbox3= self.horizontalLayout_3
        item = hbox3.itemAt(Ui.curentBut).widget()  
        item.setText("")
        item.setStyleSheet("background-color: white; color: black;")
        Ui.curentBut=Ui.curentBut+1
        item = hbox3.itemAt(Ui.curentBut).widget()  
        item.setStyleSheet("background-color: red; color: white;")
        item.setText("*")
    
    #Метод перехода каретки влево     
    def goLeft(self):
        hbox3= self.horizontalLayout_3
        item = hbox3.itemAt(Ui.curentBut).widget()  
        item.setText("")
        item.setStyleSheet("background-color: white; color: black;")
        Ui.curentBut=Ui.curentBut-1
        item = hbox3.itemAt(Ui.curentBut).widget()  
        item.setStyleSheet("background-color: red; color: white;")
        item.setText("*")
    
    #Метод изменения значения ячейки на ленте на 1      
    def setOne(self):
        hbox=self.horizontalLayout
        item = hbox.itemAt(Ui.curentBut).widget()  
        item.setText("1")
    
    #Метод изменения значения ячейки на ленте на 0  
    def setZero(self):
        hbox=self.horizontalLayout
        item = hbox.itemAt(Ui.curentBut).widget()  
        item.setText("0")
    
    #Метод для изменения нумерации ячеек    
    def rowsEdit(self):
        for i in range(self.tableWidget.rowCount()):
            item = QTableWidgetItem(str(i))
            self.tableWidget.setVerticalHeaderItem(i, item)
    
    #Проверка на правильность введённых данных        
    def checknum(self):
        textNum="0123456789, "
        nextCom = self.lineEdit_comNext.text()
        for i in nextCom:
            if textNum.find(i)==-1:
                return -1
        return 1
        
    
    #Изменение скорости анимации
    def timeDistanse500(self):
        Ui.timeDistanse=500    
    def timeDistanse1000(self):
        Ui.timeDistanse=1000
    def timeDistanse1500(self):
        Ui.timeDistanse=1500
    def timeDistanse2000(self):
        Ui.timeDistanse=2000
    
    #для записи ? в файл с большими числами
    def writingStr(cell_data):
        result=""
        for str in range(len(cell_data)):
            if cell_data[str]!=" ":
                result=result+cell_data[str]
        return result
    
    #для вывода ? из файла с большими числами
    def readingStr(item):
        result=""
        for str in range(len(item)):
            if item[str]==",":
                result=result+item[str]+" "
            else:
                 result=result+item[str]          
        return result
    
    #Метод при вызове справки    
    def get_spravka(self):
        str0="В приложении 'Машина Поста':\n"
        str1="-Для записи в поле 'Команда' использовать только символы: '?.<>10'\n"
        str2="-Для записи в поле 'Номер следующей команды' использовать только формат ввода чисел или 'число, число'\n"
        str3="-Для добавления команды в таблицу нажать 'Добавить команду'\n"
        str4="-Для того чтобы сохранить программу, нужно ввести в поле 'Название файла' название файла и добавить расширение (.txt), затем нажать на кнопку 'Сохранить программу'\n"
        str5="-Для того чтобы импортировать программу нужно нажать на кнопку 'Импортировать программу' и выбрать нужный файл в открывшемся окне\n"
        str6="-Для запуска программы нажать на 'Запуск'\n"
        str7="-Для очистки таблицы нажать на 'Очистить'\n"
        str8="-Для очистки ленты нажать на 'Очистить ленту'\n"
        str9="-Для выбора скрости анимации нажать на 'Скорость (мс)' и выбрать нужную скорость\n"
        strComplete=str0+str1+str2+str3+str4+str5+str6+str7+str8+str9+str0
        QtWidgets.QMessageBox.information(self, "Справка", strComplete)
        

        
#Настройка формы
if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    window = Ui()
    window.show()
    sys.exit(app.exec_())