class QuickNoteDialog(BoundQuickNoteDialog):
'''Dialog which includes a notebook chooser'''
def __init__(self,
window,
notebook=None,
page=None,
namespace=None,
basename=None,
append=None,
text=None,
template_options=None,
attachments=None):
assert page is None, 'TODO'
manager = ConfigManager() # FIXME should be passed in
self.config = manager.get_config_dict('quicknote.conf')
self.uistate = self.config['QuickNoteDialog']
Dialog.__init__(self, window, _('Quick Note'))
self._updating_title = False
self._title_set_manually = not basename is None
self.attachments = attachments
if notebook and not isinstance(notebook, basestring):
notebook = notebook.uri
self.uistate.setdefault('lastnotebook', None, basestring)
if self.uistate['lastnotebook']:
notebook = notebook or self.uistate['lastnotebook']
self.config['Namespaces'].setdefault(notebook, None, basestring)
namespace = namespace or self.config['Namespaces'][notebook]
self.form = InputForm()
self.vbox.pack_start(self.form, False)
# TODO dropdown could use an option "Other..."
label = gtk.Label(_('Notebook') + ': ')
label.set_alignment(0.0, 0.5)
self.form.attach(label, 0, 1, 0, 1, xoptions=gtk.FILL)
# T: Field to select Notebook from drop down list
self.notebookcombobox = NotebookComboBox(current=notebook)
self.notebookcombobox.connect('changed', self.on_notebook_changed)
self.form.attach(self.notebookcombobox, 1, 2, 0, 1)
self._init_inputs(namespace, basename, append, text, template_options)
self.uistate['lastnotebook'] = notebook
self._set_autocomplete(notebook)
def save_uistate(self):
notebook = self.notebookcombobox.get_notebook()
self.uistate['lastnotebook'] = notebook
self.uistate['new_page'] = self.form['new_page']
self.uistate['open_page'] = self.open_page.get_active()
if notebook is not None:
if self.uistate['new_page']:
self.config['Namespaces'][notebook] = self.form['namespace']
else:
self.config['Namespaces'][notebook] = self.form['page']
self.config.write()
def on_notebook_changed(self, o):
notebook = self.notebookcombobox.get_notebook()
if not notebook or notebook == self.uistate['lastnotebook']:
return
self.uistate['lastnotebook'] = notebook
self.config['Namespaces'].setdefault(notebook, None, basestring)
namespace = self.config['Namespaces'][notebook]
if namespace:
self.form['namespace'] = namespace
self._set_autocomplete(notebook)
def _set_autocomplete(self, notebook):
if notebook:
if isinstance(notebook, basestring):
notebook = NotebookInfo(notebook)
obj, x = build_notebook(notebook)
self.form.widgets['namespace'].notebook = obj
self.form.widgets['page'].notebook = obj
logger.debug('Notebook for autocomplete: %s (%s)', obj, notebook)
else:
self.form.widgets['namespace'].notebook = None
self.form.widgets['page'].notebook = None
logger.debug('Notebook for autocomplete unset')
def _get_ui(self):
start_server_if_not_running()
notebook = self.notebookcombobox.get_notebook()
if notebook:
return ServerProxy().get_notebook(notebook)
else:
return None
class QuickNoteDialog(Dialog):
'''Dialog bound to a specific notebook'''
def __init__(self,
window,
notebook=None,
page=None,
namespace=None,
basename=None,
append=None,
text=None,
template_options=None,
attachments=None):
assert page is None, 'TODO'
manager = ConfigManager() # FIXME should be passed in
self.config = manager.get_config_dict('quicknote.conf')
self.uistate = self.config['QuickNoteDialog']
Dialog.__init__(self, window, _('Quick Note'))
self._updating_title = False
self._title_set_manually = not basename is None
self.attachments = attachments
if notebook and not isinstance(notebook, str):
notebook = notebook.uri
self.uistate.setdefault('lastnotebook', None, str)
if self.uistate['lastnotebook']:
notebook = notebook or self.uistate['lastnotebook']
self.config['Namespaces'].setdefault(notebook, None, str)
namespace = namespace or self.config['Namespaces'][notebook]
self.form = InputForm()
self.vbox.pack_start(self.form, False, True, 0)
# TODO dropdown could use an option "Other..."
label = Gtk.Label(label=_('Notebook') + ': ')
label.set_alignment(0.0, 0.5)
self.form.attach(label, 0, 1, 0, 1, xoptions=Gtk.AttachOptions.FILL)
# T: Field to select Notebook from drop down list
self.notebookcombobox = NotebookComboBox(current=notebook)
self.notebookcombobox.connect('changed', self.on_notebook_changed)
self.form.attach(self.notebookcombobox, 1, 2, 0, 1)
self._init_inputs(namespace, basename, append, text, template_options)
self.uistate['lastnotebook'] = notebook
self._set_autocomplete(notebook)
def _init_inputs(self,
namespace,
basename,
append,
text,
template_options,
custom=None):
if template_options is None:
template_options = {}
else:
template_options = template_options.copy()
if namespace is not None and basename is not None:
page = namespace + ':' + basename
else:
page = namespace or basename
self.form.add_inputs((
('page', 'page', _('Page')),
('namespace', 'namespace',
_('Page section')), # T: text entry field
('new_page', 'bool', _('Create a new page for each note')
), # T: checkbox in Quick Note dialog
('basename', 'string', _('Title')) # T: text entry field
))
self.form.update({
'page': page,
'namespace': namespace,
'new_page': True,
'basename': basename,
})
self.uistate.setdefault('open_page', True)
self.uistate.setdefault('new_page', True)
if basename:
self.uistate['new_page'] = True # Be consistent with input
# Set up the inputs and set page/ namespace to switch on
# toggling the checkbox
self.form.widgets['page'].set_no_show_all(True)
self.form.widgets['namespace'].set_no_show_all(True)
if append is None:
self.form['new_page'] = bool(self.uistate['new_page'])
else:
self.form['new_page'] = not append
def switch_input(*a):
if self.form['new_page']:
self.form.widgets['page'].hide()
self.form.widgets['namespace'].show()
self.form.widgets['basename'].set_sensitive(True)
else:
self.form.widgets['page'].show()
self.form.widgets['namespace'].hide()
self.form.widgets['basename'].set_sensitive(False)
switch_input()
self.form.widgets['new_page'].connect('toggled', switch_input)
self.open_page_check = Gtk.CheckButton.new_with_mnemonic(
_('Open _Page')) # T: Option in quicknote dialog
# Don't use "O" as accelerator here to avoid conflict with "Ok"
self.open_page_check.set_active(self.uistate['open_page'])
self.action_area.pack_start(self.open_page_check, False, True, 0)
self.action_area.set_child_secondary(self.open_page_check, True)
# Add the main textview and hook up the basename field to
# sync with first line of the textview
window, textview = ScrolledTextView()
self.textview = textview
self.textview.set_editable(True)
self.vbox.pack_start(window, True, True, 0)
self.form.widgets['basename'].connect('changed', self.on_title_changed)
self.textview.get_buffer().connect('changed', self.on_text_changed)
# Initialize text from template
template = get_template('plugins', 'quicknote.txt')
template_options['text'] = text or ''
template_options.setdefault('url', '')
lines = []
template.process(lines, template_options)
buffer = self.textview.get_buffer()
buffer.set_text(''.join(lines))
begin, end = buffer.get_bounds()
buffer.place_cursor(begin)
buffer.set_modified(False)
self.connect('delete-event', self.do_delete_event)
def on_notebook_changed(self, o):
notebook = self.notebookcombobox.get_notebook()
if not notebook or notebook == self.uistate['lastnotebook']:
return
self.uistate['lastnotebook'] = notebook
self.config['Namespaces'].setdefault(notebook, None, str)
namespace = self.config['Namespaces'][notebook]
if namespace:
self.form['namespace'] = namespace
self._set_autocomplete(notebook)
def _set_autocomplete(self, notebook):
if notebook:
try:
if isinstance(notebook, str):
notebook = NotebookInfo(notebook)
obj, x = build_notebook(notebook)
self.form.widgets['namespace'].notebook = obj
self.form.widgets['page'].notebook = obj
logger.debug('Notebook for autocomplete: %s (%s)', obj,
notebook)
except:
logger.exception('Could not set notebook: %s', notebook)
else:
self.form.widgets['namespace'].notebook = None
self.form.widgets['page'].notebook = None
logger.debug('Notebook for autocomplete unset')
def do_response(self, id):
if id == Gtk.ResponseType.DELETE_EVENT:
if self.textview.get_buffer().get_modified():
ok = QuestionDialog(self, _('Discard note?')).run()
# T: confirm closing quick note dialog
if ok:
Dialog.do_response(self, id)
# else pass
else:
Dialog.do_response(self, id)
else:
Dialog.do_response(self, id)
def do_delete_event(self, *a):
# Block deletion if do_response did not yet destroy the dialog
return True
def run(self):
self.textview.grab_focus()
Dialog.run(self)
def show(self):
self.textview.grab_focus()
Dialog.show(self)
def save_uistate(self):
notebook = self.notebookcombobox.get_notebook()
self.uistate['lastnotebook'] = notebook
self.uistate['new_page'] = self.form['new_page']
self.uistate['open_page'] = self.open_page_check.get_active()
if notebook is not None:
if self.uistate['new_page']:
self.config['Namespaces'][notebook] = self.form['namespace']
else:
self.config['Namespaces'][notebook] = self.form['page']
self.config.write()
def on_title_changed(self, o):
o.set_input_valid(True)
if not self._updating_title:
self._title_set_manually = True
def on_text_changed(self, buffer):
if not self._title_set_manually:
# Automatically generate a (valid) page name
self._updating_title = True
start, end = buffer.get_bounds()
title = start.get_text(end).strip()[:50]
# Cut off at 50 characters to prevent using a whole paragraph
title = title.replace(':', '')
if '\n' in title:
title, _ = title.split('\n', 1)
try:
title = Path.makeValidPageName(title.replace(':', ''))
self.form['basename'] = title
except ValueError:
pass
self._updating_title = False
def do_response_ok(self):
buffer = self.textview.get_buffer()
start, end = buffer.get_bounds()
text = start.get_text(end)
# HACK: change "[]" at start of line into "[ ]" so checkboxes get inserted correctly
text = re.sub(r'(?m)^(\s*)\[\](\s)', r'\1[ ]\2', text)
# Specify "(?m)" instead of re.M since "flags" keyword is not
# supported in python 2.6
notebook = self._get_notebook()
if notebook is None:
return False
if self.form['new_page']:
if not self.form.widgets['namespace'].get_input_valid() \
or not self.form['basename']:
if not self.form['basename']:
entry = self.form.widgets['basename']
entry.set_input_valid(False, show_empty_invalid=True)
return False
path = self.form['namespace'] + self.form['basename']
self.create_new_page(notebook, path, text)
else:
if not self.form.widgets['page'].get_input_valid() \
or not self.form['page']:
return False
path = self.form['page']
self.append_to_page(notebook, path, '\n------\n' + text)
if self.attachments:
self.import_attachments(notebook, path, self.attachments)
if self.open_page_check.get_active():
self.hide()
ZIM_APPLICATION.present(notebook, path)
return True
def _get_notebook(self):
uri = self.notebookcombobox.get_notebook()
notebook, p = build_notebook(Dir(uri))
return notebook
def create_new_page(self, notebook, path, text):
page = notebook.get_new_page(path)
page.parse('wiki', text) # FIXME format hard coded
notebook.store_page(page)
def append_to_page(self, notebook, path, text):
page = notebook.get_page(path)
page.parse('wiki', text, append=True) # FIXME format hard coded
notebook.store_page(page)
def import_attachments(self, notebook, path, dir):
attachments = notebook.get_attachments_dir(path)
attachments = Dir(attachments.path) # XXX
for name in dir.list():
# FIXME could use list objects, or list_files()
file = dir.file(name)
if not file.isdir():
file.copyto(attachments)
class QuickNoteDialog(BoundQuickNoteDialog):
'''Dialog which includes a notebook chooser'''
def __init__(self, ui, notebook=None, namespace=None, basename=None, append=None, text=None, template_options=None, attachments=None):
self.config = get_config('quicknote.conf')
self.uistate = self.config['QuickNoteDialog']
Dialog.__init__(self, ui, _('Quick Note'))
self._updating_title = False
self._title_set_manually = not basename is None
self.attachments = attachments
if notebook and not isinstance(notebook, basestring):
notebook = notebook.uri
self.uistate.setdefault('lastnotebook', None, basestring)
if self.uistate['lastnotebook']:
notebook = notebook or self.uistate['lastnotebook']
self.config['Namespaces'].setdefault(notebook, None, basestring)
namespace = namespace or self.config['Namespaces'][notebook]
self.form = InputForm()
self.vbox.pack_start(self.form, False)
# TODO dropdown could use an option "Other..."
label = gtk.Label(_('Notebook')+': ')
label.set_alignment(0.0, 0.5)
self.form.attach(label, 0,1, 0,1, xoptions=gtk.FILL)
# T: Field to select Notebook from drop down list
self.notebookcombobox = NotebookComboBox(current=notebook)
self.notebookcombobox.connect('changed', self.on_notebook_changed)
self.form.attach(self.notebookcombobox, 1,2, 0,1)
self._init_inputs(namespace, basename, append, text, template_options)
self.uistate['lastnotebook'] = notebook
self._set_autocomplete(notebook)
def save_uistate(self):
notebook = self.notebookcombobox.get_notebook()
self.uistate['lastnotebook'] = notebook
self.uistate['new_page'] = self.form['new_page']
self.uistate['open_page'] = self.open_page.get_active()
if notebook is not None:
if self.uistate['new_page']:
self.config['Namespaces'][notebook] = self.form['namespace']
else:
self.config['Namespaces'][notebook] = self.form['page']
self.config.write()
def on_notebook_changed(self, o):
notebook = self.notebookcombobox.get_notebook()
if not notebook or notebook == self.uistate['lastnotebook']:
return
self.uistate['lastnotebook'] = notebook
self.config['Namespaces'].setdefault(notebook, None, basestring)
namespace = self.config['Namespaces'][notebook]
if namespace:
self.form['namespace'] = namespace
self._set_autocomplete(notebook)
def _set_autocomplete(self, notebook):
if notebook:
obj = get_notebook(notebook)
self.form.widgets['namespace'].notebook = obj
self.form.widgets['page'].notebook = obj
# Could still be None, e.g. if the notebook folder is not mounted
logger.debug('Notebook for autocomplete: %s (%s)', obj, notebook)
else:
self.form.widgets['namespace'].notebook = None
self.form.widgets['page'].notebook = None
logger.debug('Notebook for autocomplete unset')
def do_response_ok(self):
def get_ui():
start_server_if_not_running()
notebook = self.notebookcombobox.get_notebook()
if notebook:
return ServerProxy().get_notebook(notebook)
else:
return None
return BoundQuickNoteDialog.do_response_ok(self, get_ui)
