main.py

import tkinter
from tkinter import * # tkinter의 모든 함수 가져오기
from tkinter import messagebox, filedialog
import os
from pathlib import Path
import openpyxl
from openpyxl import load_workbook
import os.path
from openpyxl.worksheet.table import Table, TableStyleInfo
import tkinter.ttk
import tkinter as tk


#왼쪽 목록 관련 함수
def del_t(): #오른쪽 트리 삭제용
    tree.delete(*tree.get_children())
def del_t2(): #오른쪽 트리 삭제용
    tree2.delete(*tree2.get_children())

def btn1():
    insert_tree(og_sheets[0])
    global temp_sheet
    temp_sheet = og_sheets[0]
def btn2():
    insert_tree(og_sheets[1])
    global temp_sheet
    temp_sheet = og_sheets[1]
def btn3():
    insert_tree(og_sheets[2])
    global temp_sheet
    temp_sheet = og_sheets[2]
def btn4():
    insert_tree(og_sheets[3])
    global temp_sheet
    temp_sheet = og_sheets[3]
def btn5():
    insert_tree(og_sheets[4])
    global temp_sheet
    temp_sheet=og_sheets[4]
def insert_tree(sheet): #자료형 변환해서 화면 표시, 저장
    del_t()
    print("inset_tree()")
    row = []
    modified_sheet = []

    for x in range(2, (sheet.max_row + 1)):
        for y in range(1, 5):
            if (sheet.cell(x, 1).value == None) | (sheet.cell(x, 1).value == '') | (sheet.cell(x, 1).value == 0):  # 물품명이 None, '', 0 이면 참조 끝
                break
            elif (y==4) & (sheet.cell(x, 4).value ==None): #None이면 빈셀로
                row.append("")
            elif sheet.cell(x, y).value == None: #None이면 0으로
                row.append(0)
            elif (y!=1) & (y!=4) & (type(sheet.cell(x,y).value)==str): #물품명이 아니면서 str일 경우
                print(1)
                row.append(int(float(sheet.cell(x,y).value)))
            else:
                row.append(sheet.cell(x, y).value)
        modified_sheet.append(row)
        row = []
        tree.insert('', 'end', text="", values=modified_sheet[x - 2])
    og_file.save(home)
def l_double(event): #왼쪽 물품 더블클릭
    def close():
        center_tree()
        count_item.quit()
        count_item.destroy()
    def go(): #확인 버튼
        num=int(amount.get()) #입력된 텍스트(수량)저장
        selectedItem = tree.selection()[0]  # tree 선택한 위치 받기


        row=[] #지역변수 리셋 필요 없음
        if(((tree.item(selectedItem)['values'][2])==None)| (tree.item(selectedItem)['values'][2]<num)):
            messagebox.showinfo("","수량보다 많이 입력하였습니다.")
        else:
            for row2 in temp_sheet.iter_rows(min_row=2):
                for cell in row2:
                    if cell.value == tree.item(selectedItem)['values'][0]:
                        row2[2].value = int(tree.item(selectedItem)['values'][2])-num
                        og_file.save(home)

            row.append(tree.item(selectedItem)['values'][0])  # 물품명
            row.append(tree.item(selectedItem)['values'][1])  # 단가
            row.append(num)  # 수량
            row.append(row[1] * num)  # 금액
            row.append(tree.item(selectedItem)['values'][3])
            new_p.append(row)  # new_p에 저장(선택한 값 모두 받기

            for i in range(len(new_p)-1):
                if (new_p[i][0] == tree.item(selectedItem)['values'][0]):
                    new_p[i][2] += num
                    new_p[i][3] = tree.item(selectedItem)['values'][1] * new_p[i][2]
                    new_p.remove(row)
            insert_tree(temp_sheet)
        close()

        # print(new_p[0][0])
    def go_enter(event): #엔터 사용을 위한 함수
        num = int(amount.get())  # 입력된 텍스트(수량)저장
        selectedItem = tree.selection()[0]  # tree 선택한 위치 받기

        row = []  # 지역변수 리셋 필요 없음
        if (((tree.item(selectedItem)['values'][2]) == None) | (tree.item(selectedItem)['values'][2] < num)):
            messagebox.showinfo("", "수량보다 많이 입력하였습니다.")
        else:
            for row2 in temp_sheet.iter_rows(min_row=2):
                for cell in row2:
                    if cell.value == tree.item(selectedItem)['values'][0]:
                        row2[2].value = int(tree.item(selectedItem)['values'][2]) - num
                        og_file.save(home)

            row.append(tree.item(selectedItem)['values'][0])  # 물품명
            row.append(tree.item(selectedItem)['values'][1])  # 단가
            row.append(num)  # 수량
            row.append(row[1] * num)  # 금액
            row.append(tree.item(selectedItem)['values'][3])
            new_p.append(row)  # new_p에 저장(선택한 값 모두 받기

            for i in range(len(new_p) - 1):
                if (new_p[i][0] == tree.item(selectedItem)['values'][0]):
                    new_p[i][2] += num
                    new_p[i][3] = tree.item(selectedItem)['values'][1] * new_p[i][2]
                    new_p.remove(row)
            insert_tree(temp_sheet)
        close()

    global temp_sheet
    print(temp_sheet)

    count_item = Tk()  # 불러오기 하면 나오는 화면

    count_item.geometry("200x150+500+300")  # 창의 크기
    count_item.title("수량 입력")  # 창의 제목
    count_item.option_add("*Font", "맑은고딕 14")  # 전체 폰트

    ontk = Label(count_item) #수량 레이블
    ontk.config(text="수량", width=10, relief="solid")
    ontk.pack(side="top", pady=10)

    amount = Entry(count_item) #수량 엔트리 go_enter 연결
    amount.config(width=10, relief="solid", borderwidth=0)
    amount.focus()
    amount.bind("<Return>", go_enter)
    amount.place(x=60,y=50)
    amount.pack()

    conf = Button(count_item, text="확인") #확인 버튼
    conf.config(width=10, height=3, command=go) #go 연결
    # conf.place(x=30,y=200)
    conf.pack(side="bottom",pady=10)
    count_item.mainloop()
def center_tree(): #중앙 트리뷰에 표시
    del_t2()
    for i in range(len(new_p)):
        tree2.insert('', 'end', text="", values=new_p[i])
    tree2.place(x=500, y=200)
    save()
def save(): #개인 정보 물품에 저장
    print(new_p)
    for i in range(len(new_p)):
        info_sheets[0].delete_rows(0) #-----------info_sheets[0]은 시트가 늘어나면 숭자가 바뀔 예정
    for x in range(1,(len(new_p)+1)):
        for y in range(1,6):
            info_sheets[0].cell(x,y).value=new_p[x-1][y-1]
            info_file.save(info_xl)

def exsave(): #빈소에 등록 된 물품명 읽어오기

    row = []

    for x in range(1,(readsh.max_row + 1)): # 행의 끝까지 반복
        row.append(readsh.cell(x, 1).value) # row에 값 넣기 / 1을 바꾸면 열이 바뀜

    print(*row) #row 내의 목록 전체 출력 (테스트용)

def exsave2(): #거래명세서에 등록 된 물품명 읽어오기
    row = []
    
    for x in range(7, (readtrade.max_row + 1)): # 7번행부터 (물품명 시작) 끝까지 반복
        row.append(readtrade.cell(x, 3).value)  # C열부터 (물품명) row에 값넣기

    print(*row) # row 내의 목록 전체 출력 (테스트용)

'''def exinsert():  #일치하는 값 입력

    row = []

    for x in range(7, (readtrade.max_row + 1)):
        row.append(readtrade.cell(x, 6).value)

    write_readtrade.append([row])
    #print(*row)'''

'''def compare():
    exsave() == exsave2()
    if True:
        print('2')
    else:
        print('1')'''



if __name__ == "__main__":
#시트기준


#빈소 특 101,102,201,202
    # sp101
    # sp102
    # sp201
    # sp202


    home = 'xl/test.xlsx'
    info_xl='xl/personal.xlsx'
    trade_xl='xl/trade.xlsx'

    og_file= openpyxl.load_workbook(home, data_only=True) #초기 시트 위치 저장(값으로)
    info_file=openpyxl.load_workbook(info_xl,data_only=True) #개인정보, 빈소별 물품정보 저장 공간(값으)
    trade_file=openpyxl.load_workbook(trade_xl,data_only=True)


    info_sheets=[info_file['빈소1']] #지금은 하나만 사용하지만 빈소 창이 생기면 9개로 늘어날 것임
    og_sheets=[og_file['식당판매'], og_file['매점판매'], og_file['장의용품'], og_file['상복'], og_file['기타']]  #시트 리스트에 저장 시트 이름 바꾸면 같이 바꿔야 함
    readsh=info_file['빈소1']  #Hong info_sheets에 리스트를 2개씩 사용해서 읽히지않아 추가 작성
    readtrade=trade_file['거래명세서'] #거래명세서 시트 읽어오기


    og_row=['','','','',''] #길이 저장


    exsave()
    exsave2()
    #compare()
    #exinsert()

    global temp_sheet

    global new_p #중앙 목록 폼 출력용
    new_p=[]

    win = tk.Tk() # 창 생성
    win.geometry("1200x720") # 창의 크기
    win.title("장례식장 재고관리 프로그램 Ver1.221123") # 창의 제목
    win.option_add("*Font", "맑은고딕 12") # 전체 폰트

    #-------------------------------------------------

    tree = tkinter.ttk.Treeview(win, columns=["one", "two", "three"],
                                displaycolumns=["one", "two", "three"], height=24)  # 3개 창 생성

    tree.column("#0", width=10, anchor="center")  # 1
    tree.heading("#0", text="", anchor="center")

    tree.column("#1", width=90, anchor="center")  # 2
    tree.heading("#1", text="물품명", anchor="center")

    tree.column("#2", width=100, anchor="center")  # 3
    tree.heading("#2", text="단가", anchor="center")

    tree.column("#3", width=100, anchor="center")  # 4
    tree.heading("#3", text="수량", anchor="center")

    #-------------------------------------------------

    tree2 = tkinter.ttk.Treeview(win, columns=["one", "two", "three","four","five"],
                                displaycolumns=["one", "two", "three","four","five"], height=24)  # 4개 창 생성

    tree2.column("#0", width=10, anchor="center")  # 0
    tree2.heading("#0", text="", anchor="center")

    tree2.column("#1", width=90, anchor="center")  # 1
    tree2.heading("#1", text="물품명", anchor="center")

    tree2.column("#2", width=100, anchor="center")  # 2
    tree2.heading("#2", text="단가", anchor="center")

    tree2.column("#3", width=100, anchor="center")  # 3
    tree2.heading("#3", text="수량", anchor="center")

    tree2.column("#4", width=100, anchor="center")  # 4
    tree2.heading("#4", text="금액", anchor="center")

    tree2.column("#5", width=100, anchor="center")  # 5
    tree2.heading("#5", text="단위", anchor="center")


    #-------------------------------------------------

    시트1 = Button(win, text = "식당판매")
    시트1.config(width=7,height=2,command=btn1)
    시트1.place(x=10,y=10)

    시트2 = Button(win, text = "매점판매")
    시트2.config(width=7,height=2,command=btn2)
    시트2.place(x=100,y=10)

    시트3 = Button(win, text = "장의용품")
    시트3.config(width=7,height=2,command=btn3)
    시트3.place(x=190,y=10)

    시트4 = Button(win, text = "상복")
    시트4.config(width=7,height=2,command=btn4)
    시트4.place(x=280,y=10)

    시트5 = Button(win, text = "기타")
    시트5.config(width=7,height=2,command=btn5)
    시트5.place(x=370,y=10)

    tree.place(x=10,y=200)
    tree.bind("<Double-Button-1>",l_double)
    tree2.place(x=500,y=200)
    # tree2.bind("<Double-Button-1>",center_double)


win.mainloop() # 창 실행