HYTOS

import os

def finder():

    print("finder start")

    targetText = ['restriction orif union<']

    path = 'W:\ID2_Project\POSCO\Temp'

    print("target : " + str(len(targetText)))

    print("path : " + path)

    start = input("start enter 1 : ")

    if start != '1':

        return

    xmlList = os.listdir(path)

    nonXmlList = []

    for index in range(len(xmlList)):

        if xmlList[index][-3:] != 'xml':

            nonXmlList.append(index)

    nonXmlList.sort(reverse=True)

    for index in nonXmlList:

        xmlList.pop(index)

    for xml in xmlList:

        xmlPath = os.path.join(path, xml)

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

        xmlCon = fw.read()

        fw.close()

        number = 0

        for target in targetText:

            while xmlCon.find(target) != -1:

                xmlCon = xmlCon.replace(target, '', 1)

                number += 1

        if number > 0:

            print("drawing : " + xml)

            print("count : " + str(number))

            print("")

def changer():

    print("changer start")

    originText = 'kg/cm²g'

    newText = ''

    path = 'W:\ID2_Project\POSCO\Temp'

    print("origin : " + originText)

    print("new : " + newText)

    print("path : " + path)

    start = input("start enter 1 : ")

    if start != '1':

        return

    xmlList = os.listdir(path)

    nonXmlList = []

    for index in range(len(xmlList)):

        if xmlList[index][-3:] != 'xml':

            nonXmlList.append(index)

    nonXmlList.sort(reverse=True)

    for index in nonXmlList:

        xmlList.pop(index)

    for xml in xmlList:

        xmlPath = os.path.join(path, xml)

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

        xmlCon = fw.read()

        fw.close()

        xmlCon = xmlCon.replace(originText, newText)

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

        fw.write(xmlCon)

        fw.close()

def adder():

    import re

    print("changer_2 start")

    findingText = 'Freeze="True" Length="" UID="81400489-ef7f-40f3-855f-256c2ffc7329">'

    distance = 0

    addingText = 'SP-'

    aboid = 'SP'

    path = 'W:\ID2_Project\REB\Temp'

    print("finding : " + findingText)

    print("distance : " + str(distance))

    print("adding : " + addingText)

    print("path : " + path)

    start = input("start enter 1 : ")

    if start != '1':

        return

    xmlList = os.listdir(path)

    nonXmlList = []

    for index in range(len(xmlList)):

        if xmlList[index][-3:] != 'xml':

            nonXmlList.append(index)

    nonXmlList.sort(reverse=True)

    for index in nonXmlList:

        xmlList.pop(index)

    for xml in xmlList:

        xmlPath = os.path.join(path, xml)

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

        xmlCon = fw.read()

        fw.close()

        index = []

        for position in re.finditer(findingText, xmlCon):

            pos = position.end()

            index.append(pos)

        index.reverse()

        for position in index:

            if aboid not in xmlCon[position - 1 + distance: position + 10 + distance]:

                xmlCon = xmlCon[:position + distance] + addingText + xmlCon[position + distance:]        

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

        fw.write(xmlCon)

        fw.close()

if __name__ == '__main__':

    selection = input("select fuction(changer = 1, finder = 2, adder = 3) : ")

    if selection == '1':

        changer()

    elif selection == '2':

        finder()

    elif selection == '3':

        adder()