import xml.dom.minidom
import xml.etree.ElementTree
##################################################
# 전역 변수
##################################################
loopFlag = 1
fileDescriptor = -1
document = None
##################################################
# 공통
##################################################
def PrintMenu():
print()
print("==================================================")
print("도서 관리 프로그램 메뉴")
print("==================================================")
print("[l] XML 파일 로드하기" )
print("[p] XML 출력하기" )
print("[b] 도서명 목록 출력하기")
print("[a] 새 도서 추가하기" )
print("[e] 도서명 검색하기" )
print("[m] HTML 출력하기" )
print("[q] 프로그램 종료하기" )
print("==================================================")
def ExecuteFunction(menuKey):
global document
if menuKey == "l":
document = LoadXMLFile()
elif menuKey == "p":
PrintXML()
elif menuKey == "b":
PrintBookTitleList()
elif menuKey == "a":
print()
newISBN = str(input("IBSN 코드를 입력해 주시기 바랍니다 : "))
newTitle = str(input("도서명을 입력해 주시기 바랍니다 : " ))
AddBook({"ISBN" : newISBN, "title" : newTitle})
elif menuKey == 'e':
print()
keyword = str(input("검색할 도서명 키워드를 입력해 주시기 바랍니다 : "))
print()
PrintBookList(GetBookTupleList(keyword))
elif menuKey == 'm':
print()
keyword = str(input ("HTML 출력을 위한 도서명 키워드를 입력해 주시기 바라니다 : "))
print()
html = GetHTML(GetBookTupleList(keyword))
print("========================================")
print(html)
print("========================================")
elif menuKey == 'q':
ExitApplication()
else:
print ("에러 : 알 수 없는 메뉴 키를 입력하셨습니다!")
def CheckDocument():
global document
if document == None:
print("에러 : 도큐먼트가 설정되지 않았습니다!")
return False
return True
##################################################
# XML 파일 로드하기
##################################################
def LoadXMLFile():
print()
filePath = str(input("XML 파일 경로를 입력해 주시기 바랍니다 : "))
global fileDescriptor
print()
try:
fileDescriptor = open(filePath)
except IOError:
print("파일명 또는 경로가 적절하지 않습니다.")
return None
else:
try:
document = xml.dom.minidom.parse(fileDescriptor)
except Exception:
print("파일 로딩시 에러가 발생했습니다.")
else:
print("XML 파일을 로드했습니다.")
return document
return None
##################################################
# XML 출력하기
##################################################
def PrintXML():
if CheckDocument():
print()
print("========================================")
print(document.toxml())
print("========================================")
##################################################
# 도서명 목록 출력하기
##################################################
def PrintBookTitleList():
global document
if not CheckDocument():
return None
print()
booklistElement = document.childNodes[0]
for node in booklistElement.childNodes:
if node.nodeName == "book":
for childNode in node.childNodes:
if childNode.nodeName == "title":
print("도서명 :", childNode.firstChild.nodeValue)
##################################################
# 새 도서 추가하기
##################################################
def AddBook(bookDictionary):
global document
if not CheckDocument() :
return None
bookElement = document.createElement("book")
bookElement.setAttribute("ISBN", bookDictionary["ISBN"])
titleText = document.createTextNode(bookDictionary["title"])
titleElement = document.createElement("title")
titleElement.appendChild(titleText)
bookElement.appendChild(titleElement)
booklistElement = document.firstChild
booklistElement.appendChild(bookElement)
##################################################
# 도서명 검색하기
##################################################
def GetBookTupleList(keyword):
global document
if not CheckDocument():
return None
try:
elementTree = xml.etree.ElementTree.fromstring(str(document.toxml()))
except Exception:
print ("엘리먼트 트리 파싱시 에러 : XML 문서가 정확하지 않습니다!")
return None
bookTupleList = []
bookElementIterator = elementTree.iter("book")
for bookElement in bookElementIterator:
titleElement = bookElement.find("title")
if titleElement.text.find(keyword) >= 0:
bookTupleList.append((bookElement.attrib["ISBN"], titleElement.text))
return bookTupleList
def PrintBookList(bookTupleList):
for bookTuple in bookTupleList:
print(bookTuple)
##################################################
# HTML 출력하기
##################################################
def GetHTML(bookTupleList):
domImplementation = xml.dom.minidom.getDOMImplementation()
newDocument = domImplementation.createDocument(None, "html", None)
headerElement = newDocument.createElement("header")
rootElement = newDocument.documentElement
rootElement.appendChild(headerElement)
bodyElement = newDocument.createElement("body")
for bookTuple in bookTupleList:
bElement = newDocument.createElement("b")
ibsnText = newDocument.createTextNode("ISBN : " + bookTuple[0])
bElement.appendChild(ibsnText)
bodyElement.appendChild(bElement)
brElement = newDocument.createElement("br")
bodyElement.appendChild(brElement)
pElement = newDocument.createElement("p")
titleText = newDocument.createTextNode("도서명 : " + bookTuple[1])
pElement.appendChild(titleText)
bodyElement.appendChild(pElement)
bodyElement.appendChild(brElement)
rootElement.appendChild(bodyElement)
return newDocument.toxml()
##################################################
# 프로그램 종료하기
##################################################
def UnlinkDocument():
global document
if document != None:
document.unlink()
def ExitApplication():
global loopFlag
loopFlag = 0
UnlinkDocument()
##################################################
# 프로그램 종료하기
##################################################
if __name__ == "__main__":
while(loopFlag > 0):
PrintMenu()
print()
menuKey = str(input("메뉴를 선택해 주시기 바랍니다 : "))
ExecuteFunction(menuKey)
else:
print()
print("프로그램을 종료합니다.")
print()
