from tkinter import *
from tkinter import font
from tkinter import messagebox
from os import path
from os import remove


# 添加、编辑联系人弹出框类
class PopupWindow(object):
    # 初始化构造及添加组件到弹出框
    def __init__(self, master, main_window, title, contact=None):
        self.main_window = main_window
        top = self.top = Toplevel(master)
        top.title(title)
        top.resizable(False, False)
        w = 280
        h = 320
        top.geometry('%dx%d+%d+%d' % (w, h, (ws - w) / 2, (hs - h) / 2))
        top.bind('<Escape>', lambda event: top.destroy())

        m_font = font.Font(size=16)
        l = Label(top, text="姓名：", font=m_font)
        l.pack(side=TOP, pady=5)
        self.e1 = Entry(top)
        self.e1.pack(side=TOP, padx=16, ipady=3, fill=X)
        self.e1.focus()
        if contact is not None:
            self.e1.insert(0, contact.name)

        l2 = Label(top, text="电话号码：", font=m_font)
        l2.pack(side=TOP, pady=5)
        self.e2 = Entry(top)
        self.e2.pack(side=TOP, padx=16, ipady=3, fill=X)
        if contact is not None:
            self.e2.insert(0, contact.phone_number)

        l2 = Label(top, text="工作单位：", font=m_font)
        l2.pack(side=TOP, pady=5)
        self.e3 = Entry(top)
        self.e3.pack(side=TOP, padx=16, ipady=3, fill=X)
        if contact is not None:
            self.e3.insert(0, contact.work_place)

        l2 = Label(top, text="地址：", font=m_font)
        l2.pack(side=TOP, pady=5)
        self.e4 = Entry(top)
        self.e4.pack(side=TOP, padx=16, ipady=3, fill=X)
        if contact is not None:
            self.e4.insert(0, contact.address)

        if contact is None:
            b2 = Button(top, text='添加', width=12, command=lambda: self.add_click(None))
            self.e4.bind('<Return>', self.add_click)
        else:
            b2 = Button(top, text='编辑', width=12, command=lambda: self.edit_click(None))
            self.e4.bind('<Return>', self.edit_click)
        b2.pack(side=LEFT, pady=10, padx=20)
        b3 = Button(top, text='取消', width=12, command=lambda: top.destroy())
        b3.pack(side=RIGHT, pady=10, padx=20)

        top.grab_set()

    # 点击编辑联系人按钮
    def edit_click(self, event):
        e1_name = self.e1.get()
        if not e1_name:
            messagebox.showinfo("出错了", '名字不能为空！')
            return
        e2_name = self.e2.get()
        if not e2_name:
            messagebox.showinfo("出错了", '电话号码不能为空！')
            return
        e3_name = self.e3.get()
        e4_name = self.e4.get()
        self.main_window.edit_value(e1_name, e2_name, e3_name, e4_name)
        self.top.destroy()

    # 点击添加联系人按钮
    def add_click(self, event):
        e1_name = self.e1.get()
        if not e1_name:
            messagebox.showinfo("出错了", '名字不能为空！')
            return
        e2_name = self.e2.get()
        if not e2_name:
            messagebox.showinfo("出错了", '电话号码不能为空！')
            return
        e3_name = self.e3.get()
        e4_name = self.e4.get()
        self.main_window.add_value(e1_name, e2_name, e3_name, e4_name)
        self.top.destroy()


# 主界面类
class MainWindow(object):
    # 默认初始化构造
    def __init__(self, root):
        self.contacts = []
        self.root = root
        self.add_btn_widget()
        self.add_search_widget()
        self.add_listbox_widget()
        self.add_statusbar_widget()
        self.read_save_contacts()
        self.sel_item = 0

    # 添加操作按钮
    def add_btn_widget(self):
        frame = Frame(self.root)
        frame.pack(pady=8)
        self.addBtn = Button(frame, text='添加联系人', width=15, command=lambda: self.popup("添加联系人"))
        self.addBtn.pack(padx=5, fill=X, side=LEFT)
        self.delAllBtn = Button(frame, text='删除所有联系人', width=15, command=self.del_all_contacts)
        self.delAllBtn.pack(padx=5, fill=X, side=LEFT)
        self.saveAllBtn = Button(frame, text='保存所有联系人', width=15, command=self.save_all_contacts)
        self.saveAllBtn.pack(padx=5, fill=X, side=LEFT)

    # 添加搜索框
    def add_search_widget(self):
        frame = Frame(self.root)
        frame.pack(pady=8)
        entry1 = self.input_view = Entry(frame, width=34)
        entry1.insert(0, '输入部分姓名或电话号码按回车查询')
        entry1.bind("<Button-1>", self.click_input)
        entry1.bind("<FocusOut>", self.focusout_input)
        entry1.bind('<Return>', self.search_contact)
        entry1.bind('<Escape>', self.cancel_search)
        entry1.pack(ipady=3, padx=5, side=LEFT)
        entry1.selection_range(0, len(entry1.get()))
        entry1.focus()
        command4 = self.search_btn = Button(frame, text='清空输入', width=15, command=lambda: self.cancel_search(None))
        command4["state"] = "disabled"
        command4.pack(padx=5, side=LEFT)

    # 点击输入框清空内容
    def click_input(self, event):
        if self.input_view.get() == '输入部分姓名或电话号码按回车查询':
            self.input_view.delete(0, END)

    # 输入框失去焦点时
    def focusout_input(self, event):
        if len(self.input_view.get()) == 0:
            self.input_view.insert(0, '输入部分姓名或电话号码按回车查询')

    # 添加列表及滚动条
    def add_listbox_widget(self):
        frame = Frame(self.root)
        frame.pack(pady=8)
        bolded = font.Font(size=20)
        self.lb = Listbox(frame, font=bolded, height=14, width=25, borderwidth=0)
        scrollbar = Scrollbar(frame, orient=VERTICAL)
        scrollbar.config(command=self.lb.yview)
        scrollbar.pack(side=RIGHT, fill=Y)
        self.lb.config(yscrollcommand=scrollbar.set, activestyle='none')
        scrollbar2 = Scrollbar(frame, orient=HORIZONTAL)
        scrollbar2.config(command=self.lb.xview)
        scrollbar2.pack(side=BOTTOM, fill=X)
        self.lb.config(xscrollcommand=scrollbar2.set, activestyle='none')
        self.lb.pack(fill=BOTH)
        self.lb.bind('<Double-1>', self.dbclick)
        self.lb.bind('<Button-3>', self.rclick_popup)

    # 添加界面底部联系人数
    def add_statusbar_widget(self):
        frame = Frame(self.root)
        frame.pack(pady=8, side=LEFT)
        self.label = Label(frame, text='>系统现有 0 位联系人<')
        self.label.pack()

    # 右键菜单
    def rclick_popup(self, event):
        a_menu = Menu(self.root, tearoff=0)
        a_menu.add_command(label='编辑选中的联系人', command=self.edit_contact)
        a_menu.add_command(label='删除选中的联系人', command=self.del_contact)
        a_menu.post(event.x_root, event.y_root)

    # 右键编辑选中的联系人
    def edit_contact(self):
        selection = self.lb.curselection()
        if len(selection) == 0:
            messagebox.showerror("出错了", '请先左键选中待操作的联系人！')
            return
        self.sel_item = selection[0]
        self.right_clidk_reset()
        contact = self.contacts[self.sel_item]
        self.popup("编辑联系人", contact=contact)

    # 右键删除选中的联系人
    def del_contact(self):
        selection = self.lb.curselection()
        if len(selection) == 0:
            messagebox.showerror("出错了", '请先左键选中待操作的联系人！')
            return
        self.right_clidk_reset()
        answer = messagebox.askyesno("提示", "您确定要删除此联系人吗？")
        if answer:
            self.lb.delete(self.sel_item, self.sel_item)
            self.contacts.pop(self.sel_item)
            self.label.config(text='系统现有 %d 位联系人' % len(self.contacts))
            messagebox.showinfo('提示', '联系人从列表删除成功！\n若需要保存操作结果，请点击“保存所有联系人”')

    # 若是搜索后右键，则操作重置列表
    def right_clidk_reset(self, is_dbclick=False):
        b_text = self.search_btn["state"]
        if b_text == "normal":
            ic = -1
            item = self.lb.selection_get()
            if not is_dbclick:
                self.cancel_search(None)
            for ct in self.contacts:
                ic += 1
                if (ct.name in item) and (ct.phone_number in item):
                    break
            self.sel_item = ic
            self.lb.selection_set(ic, ic)

    # 双击联系人条目
    def dbclick(self, event):
        selection = self.lb.curselection()
        self.sel_item = selection[0]
        self.right_clidk_reset(is_dbclick=True)
        contact = self.contacts[self.sel_item]
        wp = contact.work_place if len(contact.work_place) != 0 else '空'
        ad = contact.address if len(contact.address) != 0 else '空'
        msg = '姓名：%s\n电话：%s\n工作单位：%s\n地址：%s' % (contact.name, contact.phone_number, wp, ad)
        messagebox.showinfo("详细信息", msg)

    # 添加、编辑联系人弹窗
    def popup(self, title, contact=None):
        self.cancel_search(None)
        self.w = PopupWindow(self.root, self, title, contact)
        self.addBtn["state"] = "disabled"
        self.root.wait_window(self.w.top)
        self.addBtn["state"] = "normal"

    # 删除所有联系人
    def del_all_contacts(self):
        self.cancel_search(None)
        answer = messagebox.askyesno("提示", "您确定要删除所有联系人吗？")
        if answer:
            self.contacts.clear()
            self.lb.delete(0, END)
            remove("contacts.csv")
            self.label.config(text='系统现有 %d 位联系人' % len(self.contacts))

    # 保存联系人到文件
    def save_all_contacts(self):
        self.cancel_search(None)
        f = open("contacts.csv", "w", encoding='utf-8')
        for contact in self.contacts:
            str = '%s,%s,%s,%s\n' % (contact.name, contact.phone_number, contact.work_place, contact.address)
            f.write(str)
        f.close()
        messagebox.showinfo('提示', '保存 %d 位联系人到文件成功！' % len(self.contacts))

    # 读取保存在文件的联系人
    def read_save_contacts(self):
        if not path.exists('contacts.csv'):
            return
        f = open("contacts.csv", "r", encoding='utf-8')
        for line in f:
            array = line.strip().split(',')
            contact = Contact(array[0], array[1], array[2], array[3])
            self.contacts.append(contact)
            self.lb.insert(END, '%s Tel:%s' % (contact.name, contact.phone_number))
        self.label.config(text='系统现有 %d 位联系人' % len(self.contacts))
        f.close()

    # 添加联系人回调
    def add_value(self, name, phone_number, work_place, address):
        contact = Contact(name, phone_number, work_place, address)
        self.contacts.append(contact)
        self.lb.insert(END, '%s Tel:%s' % (name, phone_number))
        self.label.config(text='系统现有 %d 位联系人' % len(self.contacts))

    # 编辑联系回调
    def edit_value(self, name, phone_number, work_place, address):
        contact = self.contacts[self.sel_item]
        contact.name = name
        contact.phone_number = phone_number
        contact.work_place = work_place
        contact.address = address
        self.lb.delete(0, END)
        for contact in self.contacts:
            self.lb.insert(END, '%s Tel:%s' % (contact.name, contact.phone_number))
        self.label.config(text='系统现有 %d 位联系人' % len(self.contacts))

    # 搜索联系人方法
    def search_contact(self, event):
        self.search_btn["state"] = "normal"
        self.lb.delete(0, END)
        key = self.input_view.get().strip()
        ci = 0
        for contact in self.contacts:
            if (key in contact.name) or (key in contact.phone_number):
                self.lb.insert(END, '%s Tel:%s' % (contact.name, contact.phone_number))
                ci += 1
        self.label.config(text='查询到 %d 位联系人' % ci)

    # 取消搜索
    def cancel_search(self, event):
        b_state = self.search_btn["state"]
        if b_state == "normal":
            self.search_btn["state"] = "disabled"
            self.lb.delete(0, END)
            self.input_view.delete(0, END)
            self.input_view.insert(0, '输入部分姓名或电话号码按回车查询')
            for contact in self.contacts:
                self.lb.insert(END, '%s Tel:%s' % (contact.name, contact.phone_number))
            self.label.config(text='系统现有 %d 位联系人' % len(self.contacts))
            self.input_view.selection_range(0, len(self.input_view.get()))


# 联系人类对象
class Contact:
    def __init__(self, name, phone_number, work_place, address):
        self.name = name
        self.phone_number = phone_number
        self.work_place = work_place
        self.address = address


# 程序启动入口
if __name__ == "__main__":
    root = Tk()
    root.wm_resizable(False, False)
    root.title('通讯录管理系统')
    w = 380
    h = 560
    ws = root.winfo_screenwidth()
    hs = root.winfo_screenheight()
    root.geometry('%dx%d+%d+%d' % (w, h, (ws - w) / 2, (hs - h) / 2))
    m = MainWindow(root)
    root.mainloop()