pythonScripts/pp-idf-gtk.py at master · alnjxn/pythonScripts · 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
import pygtk
pygtk.require('2.0')

import gtk

# Check for new pygtk: this is new class in PyGtk 2.4
if gtk.pygtk_version < (2,3,90):
   print "PyGtk 2.3.90 or later required for this example"
   raise SystemExit

# Provide user with dialog box to select the IDF file to process
def userDialog():
	dialog = gtk.FileChooserDialog("Select IDF File to Process..", None,
		gtk.FILE_CHOOSER_ACTION_OPEN,
		(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
		gtk.STOCK_OPEN, gtk.RESPONSE_OK))
	dialog.set_default_response(gtk.RESPONSE_OK)

	filter = gtk.FileFilter()
	filter.set_name("IDF files")
	filter.add_pattern("*.idf")
	dialog.add_filter(filter)

	response = dialog.run()

	if response == gtk.RESPONSE_OK:
	    #print dialog.get_filename(), 'selected'
	    getFile = dialog.get_filename()
	    userEntry()
	    #userSave()
	    #idfReplace(getFile)
	elif response == gtk.RESPONSE_CANCEL:
	    print 'Closed, no files selected'
	dialog.destroy()

# Have user select the search string to be replaced
class userEntry:
    def enter_callback(self, widget, entry):
        entry_text = entry.get_text()
        searchString = entry_text
        print "Entry contents: %s\n" % entry_text
        userSave()

    def entry_toggle_editable(self, checkbutton, entry):
        entry.set_editable(checkbutton.get_active())

    def entry_toggle_visibility(self, checkbutton, entry):
        entry.set_visibility(checkbutton.get_active())

    def __init__(self):
        # create a new window
        window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        window.set_size_request(225, 100)
        window.set_title("Search String")
        window.connect("delete_event", lambda w,e: gtk.main_quit())

        vbox = gtk.VBox(False, 0)
        window.add(vbox)
        vbox.show()

        label = gtk.Label(str="Enter Search String to be Replaced:")
        vbox.pack_start(label, True, True, 0)
        label.show()

        entry = gtk.Entry()
        entry.set_max_length(50)
        entry.connect("activate", self.enter_callback, entry)
        entry.set_text("$$ReplaceThis$$")
        entry.select_region(0, len(entry.get_text()))
        vbox.pack_start(entry, True, True, 0)
        entry.show()

        hbox = gtk.HBox(False, 0)
        vbox.add(hbox)
        hbox.show()

        button = gtk.Button(stock=gtk.STOCK_OK)
        button.connect("clicked", self.enter_callback, entry)
        vbox.pack_start(button, True, True, 0)
        button.set_flags(gtk.CAN_DEFAULT)
        button.grab_default()
        button.show()
        window.show()

# Provide user with dialog box to select folder where processed
# files should be placed
def userSave():
	saveDialog = gtk.FileChooserDialog("Select output directory..", None,
	gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER,
	(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
	gtk.STOCK_OPEN, gtk.RESPONSE_OK))

	saveDialog.set_default_response(gtk.RESPONSE_OK)
	saveResponse = saveDialog.run()

	if saveResponse == gtk.RESPONSE_OK:
		getSave = saveDialog.get_folder()
		idfReplace(getFile)
	elif response == gtk.RESPONSE_CANCEL:
	    print 'Closed, no folder selected'
	saveDialog.destroy()


def idfReplace(inFile):
	# For every replacement value in list of values:
	for thisAngle in angles:
	    # Open the input file
	    idfFile = open(inFile)
	    # Open the output file
	    outFileName = str(getSave) + "output-" + str(thisAngle) + "-degree-rotation.idf";
	    outFile = open(outFileName,'w')
	    print "Now writing file:", outFileName

	    # For every line in the input file:
	    for line in idfFile:
	        # If searchString is found, replace it with thisAngle
	        newLine = line.replace(searchString, str(thisAngle))
	        # Write the line to the output file
	        outFile.write(newLine)
	    outFile.close()
	    print "Finished writing!"
	idfFile.close()
	print "Finished the whole script!"


# What do we want to replace it with?
angles = (0, 90, 180, 270)

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

if __name__ == "__main__":
    userDialog()
    main()