def __init__(self, win):
super(Worksheet, self).__init__()
self.win = win
self.app = win.app
self.connection = None
self.editor = Editor(self)
self.editor_context = EditorContext(self)
self.results = Results(self)
sw = Gtk.ScrolledWindow()
sw.add(self.editor)
sw_context = Gtk.ScrolledWindow()
sw_context.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
sw_context.add(self.editor_context)
self._context_paned = Gtk.HPaned()
if self.app.args.experimental:
self._context_paned.add1(sw_context)
self._context_paned.add2(sw)
self._context_paned.set_position(200)
self.add1(self._context_paned)
self.add2(self.results)
self.connect('realize', self.on_map_event)
self.connect('size-allocate', self.on_size_allocate)
self.app.connection_manager.connect('connection-deleted',
self.on_connection_deleted)
def __init__(self, win):
super(Worksheet, self).__init__()
self.win = win
self.app = win.app
self.connection = None
self.editor = Editor(self)
self.results = Results(self)
sw = Gtk.ScrolledWindow()
sw.add(self.editor)
self.add1(sw)
self.add2(self.results)
self.connect('realize', self.on_map_event)
self.app.connection_manager.connect(
'connection-deleted', self.on_connection_deleted)
class Worksheet(Gtk.VPaned):
__gsignals__ = {
'connection-changed': (GObject.SIGNAL_RUN_LAST, None, ()),
'active-query-changed': (GObject.SIGNAL_RUN_LAST, None, ()),
}
def __init__(self, win):
super(Worksheet, self).__init__()
self.win = win
self.app = win.app
self.connection = None
self.editor = Editor(self)
self.editor_context = EditorContext(self)
self.results = Results(self)
sw = Gtk.ScrolledWindow()
sw.add(self.editor)
sw_context = Gtk.ScrolledWindow()
sw_context.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)
sw_context.add(self.editor_context)
self._context_paned = Gtk.HPaned()
if self.app.args.experimental:
self._context_paned.add1(sw_context)
self._context_paned.add2(sw)
self._context_paned.set_position(200)
self.add1(self._context_paned)
self.add2(self.results)
self.connect('realize', self.on_map_event)
self.connect('size-allocate', self.on_size_allocate)
self.app.connection_manager.connect('connection-deleted',
self.on_connection_deleted)
def on_map_event(self, *args):
if self.get_position() != 0:
return
height = self.get_parent().get_allocation().height
self.set_position(int(height / 3) * 2)
def on_size_allocate(self, widget, allocation, *args):
# adjust position the always show results
if self.get_position() > allocation.height - 10:
self.set_position(allocation.height - 20)
def on_connection_deleted(self, manager, key):
if self.connection is not None and self.connection.key == key:
self.set_connection(None)
def save_state(self):
if self.connection is not None:
conn_key = self.connection.key
else:
conn_key = None
return {
'content': self.editor.get_text(),
'split_pos': self.get_position(),
'splitpos_context': self._context_paned.get_position(),
'connection': conn_key,
'cursor': self.editor.get_cursor_position(),
}
def restore_state(self, state):
self.editor.set_text(state['content'])
self.set_position(state.get('split_pos', 0))
self._context_paned.set_position(state.get('splitpos_context', -1))
self.set_connection(state.get('connection', None))
self.editor.set_cursor_position(state.get('cursor', None))
def get_tab_label(self):
return Gtk.Label('SQL Editor')
def assume_connection(self, force=False):
if self.connection is not None and not force:
return True
dlg = ConnectionDialog(self.win)
if dlg.run() == Gtk.ResponseType.OK:
self.connection = dlg.get_connection()
self.emit('connection-changed')
dlg.destroy()
return self.connection is not None
def assume_password(self):
if self.connection is None:
return False
result = True
if self.connection.requires_password():
result = False
dlg = Gtk.Dialog('Enter password',
self.win,
Gtk.DialogFlags.DESTROY_WITH_PARENT
| Gtk.DialogFlags.MODAL,
use_header_bar=True)
dlg.add_button('_Cancel', Gtk.ResponseType.CANCEL)
dlg.add_button('_Ok', Gtk.ResponseType.OK)
dlg.set_default_response(Gtk.ResponseType.OK)
box = dlg.get_content_area()
box.set_border_width(12)
box.set_spacing(6)
lbl = Gtk.Label()
lbl.set_markup('Password required for <b>{}</b>.'.format(
GObject.markup_escape_text(self.connection.get_label())))
box.pack_start(lbl, True, True, 0)
entry = Gtk.Entry()
entry.set_visibility(False)
entry.set_invisible_char('*')
entry.connect('activate',
lambda *a: dlg.response(Gtk.ResponseType.OK))
box.pack_start(entry, True, True, 0)
box.show_all()
if dlg.run() == Gtk.ResponseType.OK:
self.connection.set_session_password(entry.get_text())
result = True
dlg.destroy()
return result
def open_connection(self):
if not self.assume_connection():
return
if not self.assume_password():
return
self.connection.request_open()
def set_connection(self, key):
if key is None:
self.connection = None
else:
self.connection = self.app.connection_manager.get_connection(key)
self.emit('connection-changed')
def get_active_query(self):
return self.results.get_active_query()
def run_query(self):
stmt = self.editor.get_statement_at_cursor()
if stmt is None:
return
if not self.assume_connection():
return
if not self.assume_password():
return
query = Query(stmt)
self.results.set_query(query)
self.emit('active-query-changed')
self.connection.run_query(query)
class Worksheet(Gtk.VPaned):
__gsignals__ = {
'connection-changed': (GObject.SIGNAL_RUN_LAST, None, ()),
'active-query-changed': (GObject.SIGNAL_RUN_LAST, None, ()),
}
def __init__(self, win):
super(Worksheet, self).__init__()
self.win = win
self.app = win.app
self.connection = None
self.editor = Editor(self)
self.results = Results(self)
sw = Gtk.ScrolledWindow()
sw.add(self.editor)
self.add1(sw)
self.add2(self.results)
self.connect('realize', self.on_map_event)
self.app.connection_manager.connect(
'connection-deleted', self.on_connection_deleted)
def on_map_event(self, *args):
if self.get_position() != 0:
return
height = self.get_parent().get_allocation().height
self.set_position(int(height / 3) * 2)
def on_connection_deleted(self, manager, key):
if self.connection is not None and self.connection.key == key:
self.set_connection(None)
def save_state(self):
if self.connection is not None:
conn_key = self.connection.key
else:
conn_key = None
return {
'content': self.editor.get_text(),
'split_pos': self.get_position(),
'connection': conn_key,
'cursor': self.editor.get_cursor_position(),
}
def restore_state(self, state):
self.editor.set_text(state['content'])
self.set_position(state.get('split_pos', 0))
self.set_connection(state.get('connection', None))
self.editor.set_cursor_position(state.get('cursor', None))
def get_tab_label(self):
return Gtk.Label('SQL Editor')
def assume_connection(self, force=False):
if self.connection is not None and not force:
return True
dlg = ConnectionDialog(self.win)
if dlg.run() == Gtk.ResponseType.OK:
self.connection = dlg.get_connection()
self.emit('connection-changed')
dlg.destroy()
return self.connection is not None
def assume_password(self):
if self.connection is None:
return False
result = True
if self.connection.requires_password():
result = False
dlg = Gtk.Dialog('Enter password', self.win,
Gtk.DialogFlags.DESTROY_WITH_PARENT |
Gtk.DialogFlags.MODAL,
use_header_bar=True)
dlg.add_button('_Cancel', Gtk.ResponseType.CANCEL)
dlg.add_button('_Ok', Gtk.ResponseType.OK)
dlg.set_default_response(Gtk.ResponseType.OK)
box = dlg.get_content_area()
box.set_border_width(12)
box.set_spacing(6)
lbl = Gtk.Label()
lbl.set_markup('Password required for <b>{}</b>.'.format(
GObject.markup_escape_text(
self.connection.get_label())))
box.pack_start(lbl, True, True, 0)
entry = Gtk.Entry()
entry.set_visibility(False)
entry.set_invisible_char('*')
entry.connect(
'activate', lambda *a: dlg.response(Gtk.ResponseType.OK))
box.pack_start(entry, True, True, 0)
box.show_all()
if dlg.run() == Gtk.ResponseType.OK:
self.connection.set_session_password(entry.get_text())
result = True
dlg.destroy()
return result
def open_connection(self):
if not self.assume_connection():
return
if not self.assume_password():
return
self.connection.request_open()
def set_connection(self, key):
if key is None:
self.connection = None
else:
self.connection = self.app.connection_manager.get_connection(key)
self.emit('connection-changed')
def get_active_query(self):
return self.results.get_active_query()
def run_query(self):
stmt = self.editor.get_statement_at_cursor()
if stmt is None:
return
if not self.assume_connection():
return
if not self.assume_password():
return
query = Query(stmt)
self.results.set_query(query)
self.emit('active-query-changed')
self.connection.run_query(query)
