Move batchquery, listtables tools under pgtool.
from gi.repository import Gtk
from gi.repository import Gdk
from pgconsole.config import cfg
class Settings(Gtk.Window):
def __init__(self, parent):
super(Settings, self).__init__()
self._parent = parent
self.set_title('Settings')
self.set_modal(True)
self.set_transient_for(parent.win)
self.set_position(Gtk.WindowPosition.CENTER_ON_PARENT)
#self.set_border_width(10)
self.connect("key_press_event", self.on_keypress)
self.connect("destroy", self.on_destroy)
self.tabs = Gtk.Notebook()
self.add(self.tabs)
### Servers
vbox = Gtk.VBox(spacing=10)
vbox.set_border_width(10)
self.tabs.append_page(vbox, Gtk.Label('Servers'))
hbox = Gtk.HBox(spacing=10)
vbox.pack_start(hbox, expand=True, fill=True, padding=0)
sw = Gtk.ScrolledWindow()
sw.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
sw.set_shadow_type(Gtk.ShadowType.IN)
hbox.pack_start(sw, expand=True, fill=True, padding=0)
tree = Gtk.TreeView()
tree.set_headers_visible(False)
sw.add(tree)
tree.get_selection().set_mode(Gtk.SelectionMode.BROWSE)
model = Gtk.ListStore(str, object)
tree.set_model(model)
column = Gtk.TreeViewColumn("Server", Gtk.CellRendererText(), text=0)
tree.append_column(column)
tree.get_selection().connect('changed', self.item_change)
tree.set_size_request(100, -1)
try:
for server in cfg.servers.server:
if str(server.name):
title = '%s (%s)' % (str(server.name), str(server.host))
else:
title = str(server.host)
model.append([title, server])
except AttributeError:
pass
self.itemtree = tree
vbox = Gtk.VBox(spacing=10)
hbox.pack_start(vbox, expand=True, fill=True, padding=0)
table = Gtk.Table(5, 2)
table.set_row_spacings(4)
table.set_col_spacings(10)
vbox.pack_start(table, expand=True, fill=True, padding=0)
label = Gtk.Label("Name:")
label.set_alignment(0.0, 0.5)
table.attach(label, 0, 1, 0, 1)
entry = Gtk.Entry()
table.attach(entry, 1, 2, 0, 1)
self.entry_name = entry
label = Gtk.Label("Host:")
label.set_alignment(0.0, 0.5)
table.attach(label, 0, 1, 1, 2)
entry = Gtk.Entry()
entry.set_text('127.0.0.1')
table.attach(entry, 1, 2, 1, 2)
self.entry_host = entry
label = Gtk.Label("Port:")
label.set_alignment(0.0, 0.5)
table.attach(label, 0, 1, 2, 3)
entry = Gtk.Entry()
entry.set_text('5432')
table.attach(entry, 1, 2, 2, 3)
self.entry_port = entry
label = Gtk.Label("User:")
label.set_alignment(0.0, 0.5)
table.attach(label, 0, 1, 3, 4)
entry = Gtk.Entry()
entry.set_text('postgres')
table.attach(entry, 1, 2, 3, 4)
self.entry_user = entry
label = Gtk.Label("Password:")
label.set_alignment(0.0, 0.5)
table.attach(label, 0, 1, 4, 5)
entry = Gtk.Entry()
table.attach(entry, 1, 2, 4, 5)
self.entry_password = entry
hbox = Gtk.HBox()
vbox.pack_start(hbox, expand=True, fill=True, padding=0)
btn = Gtk.Button('Add')
btn.connect('clicked', self.item_add)
hbox.pack_start(btn, expand=True, fill=True, padding=0)
btn = Gtk.Button('Save')
btn.connect('clicked', self.item_save)
hbox.pack_start(btn, expand=True, fill=True, padding=0)
btn = Gtk.Button('Remove')
btn.connect('clicked', self.item_remove)
hbox.pack_start(btn, expand=True, fill=True, padding=0)
### Editor
vbox = Gtk.VBox(spacing=10)
vbox.set_border_width(10)
self.tabs.append_page(vbox, Gtk.Label('Editor'))
self.show_all()
self.itemtree.grab_focus()
def item_change(self, w):
model, sel = self.itemtree.get_selection().get_selected_rows()
if not sel:
return
el, = model.get(model.get_iter(sel[0]), 1)
self.entry_name.set_text(str(el.name))
self.entry_host.set_text(str(el.host))
self.entry_port.set_text(str(el.port))
self.entry_user.set_text(str(el.user))
self.entry_password.set_text(str(el.password))
def item_add(self, w):
el = cfg.add_server(
self.entry_name.get_text(),
self.entry_host.get_text(),
self.entry_port.get_text(),
self.entry_user.get_text(),
self.entry_password.get_text())
if str(el.name):
title = '%s (%s)' % (str(el.name), str(el.host))
else:
title = str(el.host)
iter = self.itemtree.get_model().append([title, el])
self.itemtree.get_selection().select_iter(iter)
cfg.save()
def item_save(self, w):
model, sel = self.itemtree.get_selection().get_selected_rows()
el = cfg.servers.server[sel[0][0]]
el.name = self.entry_name.get_text()
el.host = self.entry_host.get_text()
el.port = self.entry_port.get_text()
el.user = self.entry_user.get_text()
el.password = self.entry_password.get_text()
cfg.save()
if str(el.name):
title = '%s (%s)' % (str(el.name), str(el.host))
else:
title = str(el.host)
model.set(model.get_iter(sel[0]), 0, title)
def item_remove(self, w):
model, sel = self.itemtree.get_selection().get_selected_rows()
model.remove(model.get_iter(sel[0]))
if model.get_iter_first():
self.itemtree.get_selection().select_iter(model.get_iter_first())
del cfg.servers.server[sel[0][0]]
cfg.save()
def on_keypress(self, w, event):
keyname = Gdk.keyval_name(event.keyval)
if keyname == 'Escape':
self.destroy()
return True
return False
def on_destroy(self, w):
model, sel = self.itemtree.get_selection().get_selected_rows()
if sel:
sel = sel[0].get_indices()[0]
else:
sel = None
self._parent.reload_server_list(sel)