L - система

import sys
import math
from PyQt5.QtWidgets import QWidget, QApplication, QInputDialog, QPushButton, QSlider, QFileDialog
from PyQt5.QtGui import QPainter, QPen, QPixmap, QImage, QColor
from PyQt5.QtCore import Qt
 
 
class LSystem(QWidget):
    def __init__(self):
        super().__init__()
        self.color = QColor(0, 0, 0)
        self.do_paint = True
        self.dic = {}
        self.plus_alpha, self.alpha = 0, 0
        self.start, self.stack = [], []
        self.initUI()
 
    def initUI(self):
        self.setGeometry(0, 0, 1000, 1000)
        self.fname = QFileDialog.getOpenFileName(
            self, 'Выберите файл', '',
            'Файл (*.txt)')[0]
        self.file_open()
        self.setWindowTitle('Знакомство с L-системами')
        self.slider = QSlider(self)
        self.slider.setOrientation(Qt.Horizontal)
        self.slider.setMinimum(1)
        self.slider.setMaximum(6)
        self.slider.setGeometry(25, 980, 800, 20)
        self.slider.valueChanged.connect(self.paint)
 
    def paintEvent(self, event):
        if self.do_paint:
            qp = QPainter()
            qp.begin(self)
            self.draw_flag(qp)
            self.do_paint = False
            qp.end()
 
    def paint(self):
        self.do_paint = True
        self.repaint()
        self.do_paint = False
 
    def draw_flag(self, qp):
        qp.setPen(self.color)
        result = ''
        run = self.start
        for i in range(self.slider.value()):
            for i in run:
                result += self.dic.get(i, i)
            run = result
            result = ''
        result = run
        alpha = 0
        x0, y0 = 250, 400
        dlina = 20
        for i in result:
            if i == '+':
                alpha += self.plus_alpha
 
            elif i == '-':
                alpha -= self.plus_alpha
            elif i == '|':
                alpha += 180
            elif i == '[':
                self.stack.append((x0, y0, alpha))
            elif i == ']':
                x0, y0, alpha = self.stack.pop()
            else:
                x = x0 + dlina * math.cos(math.radians(alpha))
                y = y0 + dlina * math.sin(math.radians(alpha))
                qp.drawLine(int(x0), int(y0), int(x), int(y))
                x0, y0 = x, y
 
    def file_open(self):
        x = open(self.fname, encoding='utf8')
        files = x.readlines()
        files = [i.strip() for i in files]
        self.plus_alpha = int(360 / int(files[1]))
        self.start = files[2].split(',')
        for i in files[-1].split(','):
            self.dic[i[0]] = i[2:]
 
 
def except_hook(cls, exception, traceback):
    sys.__excepthook__(cls, exception, traceback)
 
 
if __name__ == "__main__":
    sys.excepthook = except_hook
    app = QApplication(sys.argv)
    ex = LSystem()
    ex.show()
    sys.exit(app.exec())