TreeDataView/example.py at master · vsantiago113/TreeDataView · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import sys
from TreeDataView import TreeDataView
import random
import tkinter as tk
from tkinter import ttk
from tkinter import messagebox
import pyperclip

my_event = None


def main():
    root = tk.Tk()
    root.wm_title('TreeDataView as a Table')

    def copy_to_clipboard():
        global my_event
        column = tdv1.identify_column(my_event.x)
        selitem = tdv1.selection()
        if selitem:
            text = tdv1.item(selitem[0], 'values')
            cell = int(column[1]) - 1
            my_event = None
            pyperclip.copy(text[cell])
        else:
            messagebox.showwarning('Warning', 'Please make a selection and try again!')

    def double_click(event):
        rowid = tdv1.identify_row(event.y)
        column = tdv1.identify_column(event.x)
        selitems = tdv1.selection()
        if selitems:
            selitem = selitems[0]
            text = tdv1.item(selitem, 'values')
            cell = int(column[1])-1
            print('Click on row:', rowid[0])
            print('Row data:', text)
            print('Clicked on Cell:', cell)
            print('Cell data:', text[cell])

    def right_click_func():
        global my_event
        column = tdv1.identify_column(my_event.x)
        selitem = tdv1.selection()
        if selitem:
            text = tdv1.item(selitem[0], 'values')
            cell = int(column[1]) - 1
            my_event = None
            print(text)
            print(cell)
        else:
            messagebox.showwarning('Warning', 'Please make a selection and try again!')

    def popup_menu(event):
        global my_event
        my_event = event
        row_id = tdv1.identify_row(event.y)
        tdv1.selection_set(row_id)
        right_click_menu.post(event.x_root, event.y_root)

    def get_row_data():
        selected_row = tdv1.selection()
        prev_item = tdv1.prev(selected_row)
        text = tdv1.item(selected_row, 'values')
        tags = tdv1.item(prev_item, 'tags')
        if tags:
            print(tags[0])
        else:
            pass
        print(text)

    def delete_row():
        selected_row = tdv1.selection()
        tdv1.delete(selected_row)

    menu = tk.Menu(root, tearoff=0)
    menu.add_command(label='Print row', command=get_row_data)
    menu.add_command(label='Add', command=None)
    menu.add_command(label='Edit', command=None)
    menu.add_command(label='Delete', command=delete_row)

    # Right Click Pop-up Menu
    right_click_menu = tk.Menu(root, tearoff=0)
    right_click_menu.add_command(label='Copy', command=copy_to_clipboard)
    right_click_menu.add_command(label='Item1', command=right_click_func)
    right_click_menu.add_command(label='Item2', command=right_click_func)
    right_click_sub_menu = tk.Menu(right_click_menu, tearoff=0)
    right_click_menu.add_cascade(label='Sub-Menu', menu=right_click_sub_menu)
    right_click_sub_menu.add_command(label='Sub Item1', command=right_click_func)
    right_click_sub_menu.add_command(label='Sub Item2', command=right_click_func)

    def mymenu(event):
        row_id = tdv1.identify_row(event.y)
        tdv1.selection_set(row_id)
        menu.post(event.x_root, event.y_root)

    tree_columns = ['Name', 'Phone', 'Email', 'Company', 'Date']
    tdv1 = TreeDataView(root, tree_columns, scrollbar_x=True, scrollbar_y=True, double_click=double_click,
                        right_click=popup_menu)
    tdv1.pack(fill='both', expand=1)

    def insert_data():
        # Lets generate a random phone number, and a date.
        phone = '(800) {}-{}'.format(random.randint(300, 900), random.randint(5000, 9000))
        date = '{:02d}/{:02d}/{}'.format(random.randint(1, 12), random.randint(1, 31), random.randint(2010, 2017))

        # Lets pick a random name, email and company name from the lists then create the email address.
        name_list = ['Kari Turner', 'Melinda Blanks', 'Russell Wise', 'James Garcia']
        company_list = ['Mansmann', 'Merrymaking', 'Monk Real Estate']
        name = random.choice(name_list)
        company = random.choice(company_list)
        email = '{}@{}'.format(name[0].lower() + name.split()[-1].lower(), company.replace(' ', '').lower())

        # Now lets insert a row with random data into the table.
        new_item = tdv1.insert('', 'end', values=(name, phone, email, company, date))

    button1 = ttk.Button(root, text='Insert random data', command=insert_data)
    button1.pack(side='bottom')

    root.mainloop()


if __name__ == '__main__':
    main()