HYTOS

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

import os

from AppDocData import AppDocData

import CopyDrawing_UI

class QCopyDrawingDialog(QDialog):

    def __init__(self, parent, drawing):

        QDialog.__init__(self, parent)

        self.selectedProject = None

        self.ui = CopyDrawing_UI.Ui_CopydrawingDialog()

        self.ui.setupUi(self)

        self.drawing = drawing

        self.isAccepted = False

        self.ui.lineEdit_2.setText(os.path.splitext(self.drawing.name)[0])

    def showDialog(self):

        self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)

        self.exec_()

        return self.isAccepted

    def accept(self):

        if not self.validationCheck():

            QMessageBox.warning(self, self.tr('Notice'), self.tr('Please check drawing files(png, xml) and name'))

            return

        if self.copy_drawing():

            self.isAccepted = True

            QDialog.accept(self)

    def copy_drawing(self):

        from shutil import copyfile

        import uuid

        try:

            app_doc_data = AppDocData.instance()

            project = app_doc_data.getCurrentProject()

            drawing_path = project.getDrawingFilePath()

            temp_path = project.getTempPath()

            original_drawing = self.ui.lineEdit_2.text() + '.png'

            origianl_xml = self.ui.lineEdit_2.text() + '.xml'

            new_drawing = self.ui.lineEdit.text() + '.png'

            new_xml = self.ui.lineEdit.text() + '.xml'

            original_drawing = os.path.join(drawing_path, original_drawing)

            origianl_xml = os.path.join(temp_path, origianl_xml)

            new_drawing = os.path.join(drawing_path, new_drawing)

            new_xml = os.path.join(temp_path, new_xml)

            if os.path.exists(original_drawing) and os.path.exists(origianl_xml):

                copyfile(original_drawing, new_drawing)

                copyfile(origianl_xml, new_xml)

                fw = open(new_xml, 'r', encoding='utf8')

                xmlCon = fw.read()

                fw.close()

                uidList = []

                for index in range(len(xmlCon)):

                    if len(xmlCon) - index < 36:

                        break

                    try:

                        text = xmlCon[index:index + 36]

                        uid = uuid.UUID(text)

                        uidList.append(str(uid))

                    except:

                        pass

                    finally:

                        pass

                uidList = list(set(uidList))

                for uid in uidList:

                    xmlCon = xmlCon.replace(uid, str(uuid.uuid4()))

                fw = open(new_xml, 'w', encoding='utf8')

                fw.write(xmlCon)

                fw.close()

                QMessageBox.about(self, self.tr("Information"), self.tr('Successfully copied.'))

                return True

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = f"error occurred({repr(ex)}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:" \

                      f"{sys.exc_info()[-1].tb_lineno}"

            App.mainWnd().addMessage.emit(MessageType.Error, message)

        return False

    def validationCheck(self):

        app_doc_data = AppDocData.instance()

        project = app_doc_data.getCurrentProject()

        drawing_path = project.getDrawingFilePath()

        temp_path = project.getTempPath()

        drawing_files = [os.path.splitext(f)[0] for f in os.listdir(drawing_path) if os.path.isfile(os.path.join(drawing_path, f)) and (os.path.splitext(f)[1].upper() == '.PNG')]

        xml_files = [os.path.splitext(f)[0] for f in os.listdir(temp_path) if os.path.isfile(os.path.join(temp_path, f)) and (os.path.splitext(f)[1].upper() == '.XML')]

        original_drawing = self.ui.lineEdit_2.text()

        new_drawing = self.ui.lineEdit.text()

        if original_drawing not in drawing_files or original_drawing not in xml_files:

            return False

        if new_drawing in drawing_files or new_drawing in xml_files:

            return False

        return True

    def reject(self):

        QDialog.reject(self)