sniff2/queryEditWindow.py at master · g0dd4rd/sniff2 · 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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
#!/usr/bin/env python

import gi
gi.require_version('Gtk', '3.0')
gi.require_version('Gdk', '3.0')
import sys
import pyatspi
import Accessibility
from gi.repository import Gtk
from gi.repository import Gdk
from gi.repository import Gio
from gi.repository import GdkPixbuf
from gi.repository import GObject
from gi.repository import GLib

class QueryEditWindow:
    def __init__(self):
        self.app_name = 'example-app'
        self.projectDir = ''
        self.window_title = 'Dogtail Query - Behave Steps Edit Window'

        self.query_text_view = Gtk.TextView()
        self.query_text_buffer = self.query_text_view.get_buffer()
        self.query_text_view.show()

        query_scroll_window = Gtk.ScrolledWindow()
        query_scroll_window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
        query_scroll_window.add(self.query_text_view)
        query_scroll_window.show()

        query_cancel_button = Gtk.Button('Clear query')
        query_cancel_button.connect('clicked', self.clear_query_text_buffer)
        query_cancel_button.show()

        query_write_button = Gtk.Button('Write Dogtail Query')
        query_write_button.connect('clicked', self.write_dogtail_query)
        query_write_button.show()

        query_execute_button = Gtk.Button('Execute Dogtail Query')
        query_execute_button.connect('clicked', self.execute_query_selection)
        query_execute_button.show()

        query_h_button_box = Gtk.HButtonBox()
        query_h_button_box.pack_start(query_write_button, True, True, 0)
        query_h_button_box.pack_start(query_execute_button, True, True, 0)
        query_h_button_box.pack_start(query_cancel_button, True, True, 0)
        query_h_button_box.show()

        steps_text_view = Gtk.TextView()
        self.steps_text_buffer = steps_text_view.get_buffer()
        steps_text_view.show()

        steps_scroll_window = Gtk.ScrolledWindow()
        steps_scroll_window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
        steps_scroll_window.add(steps_text_view)
        steps_scroll_window.show()

        steps_cancel_button = Gtk.Button('Clear query')
        steps_cancel_button.connect('clicked', self.clear_steps_text_buffer)
        steps_cancel_button.show()

        steps_write_button = Gtk.Button('Write Behave Steps')
        steps_write_button.connect('clicked', self.write_behave_steps)
        steps_write_button.show()

        steps_h_button_box = Gtk.HButtonBox()
        steps_h_button_box.pack_start(steps_write_button, True, True, 0)
        steps_h_button_box.pack_start(steps_cancel_button, True, True, 0)
        steps_h_button_box.show()

        vbox = Gtk.VBox(False, 0)
        vbox.add(query_scroll_window)
        vbox.pack_start(query_h_button_box, False, False, 0)
        vbox.add(steps_scroll_window)
        vbox.pack_start(steps_h_button_box, False, False, 0)
        vbox.show()

        self.window = Gtk.Window()
        self.window.set_resizable(True)
        self.window.connect('destroy', self.close_edit_window)
        self.window.set_border_width(0)
        self.window.add(vbox)
        self.window.show()


    def clear_query_text_buffer(self, widget):
        self.query_text_buffer.set_text('', 0)


    def clear_steps_text_buffer(self, widget):
        self.steps_text_buffer.set_text('', 0)


    def close_edit_window(self, widget):
        self.window.destroy()

    def execute_query_selection(self, widget):
        execute_query_selection_prefix = "from dogtail.tree import *; from dogtail.utils import *; app = root.application(\'"+ self.app_name +"\'); app." # make it generic
        execute_query_selection_postfix = '[0].blink()'

        from subprocess import Popen
        Popen(['python', '-c', execute_query_selection_prefix + self.query_text_buffer.get_text(self.query_text_buffer.get_selection_bounds()[0], self.query_text_buffer.get_selection_bounds()[1], include_hidden_chars = True) + execute_query_selection_postfix])
        print(execute_query_selection_prefix + self.query_text_buffer.get_text(self.query_text_buffer.get_selection_bounds()[0], self.query_text_buffer.get_selection_bounds()[1], include_hidden_chars = True) + execute_query_selection_postfix)


    def fill_query_text_buffer(self, widget, app_name, content):
        self.app_name = app_name
        self.window.set_title(self.window_title +' - '+ self.app_name)
        self.query_text_buffer.set_text(content, len(content))


    def fill_steps_text_buffer(self, widget, app_name, content):
        self.app_name = app_name
        self.steps_text_buffer.set_text(content, len(content))

    def set_project_dir(self):
        import os
        from os.path import expanduser, join, exists

        if not os.path.exists(join(expanduser('~'), '.config', 'sniff2', 'preferences.csv')):
            os.makedirs(join(expanduser('~'), '.config', 'sniff2'))
            open(join(expanduser('~'), '.config', 'sniff2', 'preferences.csv'), 'a').close()

        if os.stat(join(expanduser('~'), '.config', 'sniff2', 'preferences.csv')).st_size != 0:
            configFile = open(join(expanduser('~'), '.config', 'sniff2', 'preferences.csv'), 'r')
            self.projectDir = configFile.readline()
        else:
            self.projectDir = join(expanduser('~'), 'dogtail-behave-projects', self.app_name)


    def write_dogtail_query(self, widget):
        print('=== writing dogtail query ===')
        import os
        from os.path import join, exists

        self.set_project_dir()
        if not exists(join(self.projectDir, 'features', 'steps')):
            os.makedirs(join(self.projectDir, 'features', 'steps'))

        query = open(join(self.projectDir, 'features', 'steps', 'steps.py'), 'wb') #dont forget to change it to 'ab' mode
        query.write(self.query_text_buffer.get_text(self.query_text_buffer.get_start_iter(), self.query_text_buffer.get_end_iter(), include_hidden_chars = True))
        query.close()
        print('=== done writing dogtail query ===\n')


    def write_behave_steps(self, widget):
        print('=== writing behave steps ===')
        import os
        from os.path import join, exists

        self.set_project_dir()
        if not exists(join(self.projectDir, 'features')):
            os.makedirs(join(self.projectDir, 'features'))

        steps = open(join(self.projectDir, 'features', 'general.feature'), 'wb') #dont forget to change it to 'ab' mode
        steps.write(self.steps_text_buffer.get_text(self.steps_text_buffer.get_start_iter(), self.steps_text_buffer.get_end_iter(), include_hidden_chars = True))
        steps.close()
        print('=== done writing behave steps ===\n')


def main():
    Gtk.main()
    return 0

if __name__ == '__main__':
    QueryEditWindow()
    main()