def testCopyLocation(self):
from zim.gui.clipboard import Clipboard
Clipboard.set_text('XXX')
self.assertEqual(Clipboard.get_text(), 'XXX')
self.uiactions.copy_location()
self.assertEqual(Clipboard.get_text(), 'Test')
def get_text(self):
if 'input' in self.opts:
if self.opts['input'] == 'stdin':
import sys
text = sys.stdin.read()
elif self.opts['input'] == 'clipboard':
text = \
SelectionClipboard.get_text() \
or Clipboard.get_text()
else:
raise AssertionError('Unknown input type: %s' %
self.opts['input'])
else:
text = self.opts.get('text')
if text and 'encoding' in self.opts:
if self.opts['encoding'] == 'base64':
import base64
text = base64.b64decode(text).decode('UTF-8')
elif self.opts['encoding'] == 'url':
from zim.parsing import url_decode, URL_ENCODE_DATA
text = url_decode(text, mode=URL_ENCODE_DATA)
else:
raise AssertionError('Unknown encoding: %s' %
self.opts['encoding'])
assert isinstance(text, str), '%r is not decoded' % text
return text
def rename_bookmark(self, button):
'''
Change label of the button.
New name is taken from the clipboard.
If button's name has been changed before,
change it back to its initial state.
'''
_full, _short = button.zim_path, self._get_short_page_name(button.zim_path)
if button.get_label() in (_short, _full):
# Change the button to new name.
new_name = None
try:
# Take from clipboard.
new_name = self._convert_path_name(Clipboard.get_text())
except:
logger.error('BookmarksBar: Error while converting from buffer.')
if new_name:
self.paths_names[_full] = new_name
button.set_label(new_name)
else:
# Change the button back to its initial state.
new_name = _full if self.uistate['show_full_page_name'] else _short
button.set_label(new_name)
self.paths_names.pop(_full, None)
if self.save_flag:
self.uistate['bookmarks_names'] = self.paths_names
def testTaskListTreeView(self):
klass = PluginManager.get_plugin_class('tasklist')
plugin = klass()
notebook = tests.new_notebook()
plugin.extend(notebook.index)
index_ext = plugin.get_extension(IndexExtension)
self.assertIsNotNone(index_ext)
notebook.index.flush()
notebook.index.update()
from zim.plugins.tasklist import TaskListTreeView
opener = tests.MockObject()
treeview = TaskListTreeView(index_ext, opener)
menu = treeview.get_popup()
# Check these do not cause errors - how to verify state ?
tests.gtk_activate_menu_item(menu, _("Expand _All"))
tests.gtk_activate_menu_item(menu, _("_Collapse All"))
# Copy tasklist -> csv
from zim.gui.clipboard import Clipboard
tests.gtk_activate_menu_item(menu, 'gtk-copy')
text = Clipboard.get_text()
lines = text.splitlines()
self.assertTrue(len(lines) > 10)
self.assertTrue(len(lines[0].split(',')) > 3)
self.assertFalse(any('<span' in l for l in lines)) # make sure encoding is removed
def testTaskListTreeView(self):
plugin = PluginManager.load_plugin('tasklist')
notebook = self.setUpNotebook(content=tests.FULL_NOTEBOOK)
notebook.index.check_and_update()
from zim.plugins.tasklist.gui import TaskListTreeView
view = TasksView.new_from_index(notebook.index)
opener = tests.MockObject()
treeview = TaskListTreeView(view,
opener,
task_labels=['TODO', 'FIXME'])
menu = treeview.get_popup()
# Check these do not cause errors - how to verify state ?
tests.gtk_activate_menu_item(menu, _("Expand _All"))
tests.gtk_activate_menu_item(menu, _("_Collapse All"))
# Copy tasklist -> csv
from zim.gui.clipboard import Clipboard
tests.gtk_activate_menu_item(menu, _('_Copy'))
text = Clipboard.get_text()
lines = text.splitlines()
self.assertTrue(len(lines) > 10)
self.assertTrue(len(lines[0].split(',')) > 3)
self.assertFalse(any('<span' in l
for l in lines)) # make sure encoding is removed
# Test tags
tags = treeview.get_tags()
for tag in ('home', 'FIXME', '__no_tags__', 'tags'):
self.assertIn(tag, tags)
self.assertGreater(tags[tag], 0)
def rename_bookmark(self, button):
'''
Change label of the button.
New name is taken from the clipboard.
If button's name has been changed before,
change it back to its initial state.
'''
_full, _short = button.zim_path, self._get_short_page_name(
button.zim_path)
if button.get_label() in (_short, _full):
# Change the button to new name.
new_name = None
try:
# Take from clipboard.
new_name = self._convert_path_name(Clipboard.get_text())
except:
logger.error(
'BookmarksBar: Error while converting from buffer.')
if new_name:
self.paths_names[_full] = new_name
button.set_label(new_name)
else:
# Change the button back to its initial state.
new_name = _full if self.uistate['show_full_page_name'] else _short
button.set_label(new_name)
self.paths_names.pop(_full, None)
if self.save_flag:
self.uistate['bookmarks_names'] = self.paths_names
def testTaskListTreeView(self):
klass = zim.plugins.get_plugin_class('tasklist')
plugin = klass()
notebook = tests.new_notebook()
plugin.extend(notebook.index)
index_ext = plugin.get_extension(IndexExtension)
self.assertIsNotNone(index_ext)
notebook.index.flush()
notebook.index.update()
from zim.plugins.tasklist import TaskListTreeView
opener = tests.MockObject()
treeview = TaskListTreeView(index_ext, opener)
menu = treeview.get_popup()
# Check these do not cause errors - how to verify state ?
tests.gtk_activate_menu_item(menu, _("Expand _All"))
tests.gtk_activate_menu_item(menu, _("_Collapse All"))
# Copy tasklist -> csv
from zim.gui.clipboard import Clipboard
tests.gtk_activate_menu_item(menu, 'gtk-copy')
text = Clipboard.get_text()
lines = text.splitlines()
self.assertTrue(len(lines) > 10)
self.assertTrue(len(lines[0].split(',')) > 3)
self.assertFalse(any('<span' in l
for l in lines)) # make sure encoding is removed
def testTaskListTreeView(self):
klass = zim.plugins.get_plugin("tasklist")
ui = MockUI()
plugin = klass(ui)
ui.notebook.index.flush()
ui.notebook.index.update()
from zim.plugins.tasklist import TaskListTreeView
treeview = TaskListTreeView(ui, plugin, filter_actionable=False)
menu = treeview.get_popup()
# Check these do not cause errors - how to verify state ?
tests.gtk_activate_menu_item(menu, _("Expand _All"))
tests.gtk_activate_menu_item(menu, _("_Collapse All"))
# Copy tasklist -> csv
from zim.gui.clipboard import Clipboard
tests.gtk_activate_menu_item(menu, "gtk-copy")
text = Clipboard.get_text()
lines = text.splitlines()
self.assertTrue(len(lines) > 10)
self.assertTrue(len(lines[0].split(",")) > 3)
self.assertFalse(any("<span" in l for l in lines)) # make sure encoding is removed
def get_text(self):
if 'input' in self.opts:
if self.opts['input'] == 'stdin':
import sys
text = sys.stdin.read()
elif self.opts['input'] == 'clipboard':
text = \
SelectionClipboard.get_text() \
or Clipboard.get_text()
else:
raise AssertionError, 'Unknown input type: %s' % self.opts['input']
else:
text = self.opts.get('text')
if text and 'encoding' in self.opts:
if self.opts['encoding'] == 'base64':
import base64
text = base64.b64decode(text)
elif self.opts['encoding'] == 'url':
from zim.parsing import url_decode, URL_ENCODE_DATA
text = url_decode(text, mode=URL_ENCODE_DATA)
else:
raise AssertionError, 'Unknown encoding: %s' % self.opts['encoding']
if text and not isinstance(text, unicode):
text = text.decode('utf-8')
return text
def testContextMenu(self):
menu = self.treeview.get_popup()
# Check these do not cause errors - how to verify state ?
tests.gtk_activate_menu_item(menu, _("Expand _All"))
tests.gtk_activate_menu_item(menu, _("_Collapse All"))
# Copy item
tests.gtk_activate_menu_item(menu, 'gtk-copy')
self.assertEqual(Clipboard.get_text(), 'Test')
def testContextMenu(self):
menu = self.treeview.get_popup()
# Check these do not cause errors - how to verify state ?
tests.gtk_activate_menu_item(menu, _("Expand _All"))
tests.gtk_activate_menu_item(menu, _("_Collapse All"))
# Copy item
tests.gtk_activate_menu_item(menu, 'gtk-copy')
self.assertEqual(Clipboard.get_text(), 'Test')
def testClipboard(self): self.ui.copy_location() self.assertEqual(Clipboard.get_text(), 'Test:foo:bar')
def testClipboard(self):
self.ui.copy_location()
self.assertEqual(Clipboard.get_text(), 'Test:foo:bar')
def main(*args):
options = {}
template_options = {}
for arg in args:
if arg.startswith('option:'):
arg = arg[7:]
dict = template_options
else:
dict = options
if '=' in arg:
key, value = arg.split('=', 1)
dict[key] = value
else:
dict[arg] = True
#~ print 'OPTIONS:', options, template_options
if 'help' in options:
print usagehelp
return
if 'notebook' in options:
notebook, page = resolve_notebook(options['notebook'])
else:
notebook = None
if 'append' in options:
if options['append'].lower() == 'true':
options['append'] = True
else:
options['append'] = False
if 'input' in options:
if options['input'] == 'stdin':
import sys
text = sys.stdin.read()
elif options['input'] == 'clipboard':
text = \
SelectionClipboard.get_text() \
or Clipboard.get_text()
else:
text = options.get('text')
if text and options.get('encoding'):
if options['encoding'] == 'base64':
import base64
text = base64.b64decode(text)
elif options['encoding'] == 'url':
from zim.parsing import url_decode, URL_ENCODE_DATA
text = url_decode(text, mode=URL_ENCODE_DATA)
else:
raise AssertionError, 'Unknown encoding: %s' % options['encoding']
if text and not isinstance(text, unicode):
text = text.decode('utf-8')
icon = data_file('zim.png').path
gtk_window_set_default_icon()
dialog = QuickNoteDialog(None,
notebook,
options.get('namespace'), options.get('basename'),
options.get('append'),
text,
template_options,
options.get('attachments')
)
dialog.run()
def run(self):
if not self.opts or self.opts.get('help'):
print usagehelp
else:
_raise = 'raise' in self.opts
_show = 'show' in self.opts
if 'notebook' in self.opts:
notebookInfo = resolve_notebook(self.opts['notebook'])
else:
notebookInfo = resolve_notebook(defaultNotebook)
print 'NotebookInfo=', notebookInfo
# The notion of 'today' might extend into the wee hours of the morning.
offset_time = datetime.today() - timedelta(
hours=hours_past_midnight)
todaysJournal = offset_time.strftime(':Journal:%Y:%m:%d')
if 'page' in self.opts:
pagename = self.opts['page']
elif 'journal' in self.opts:
pagename = todaysJournal
elif 'date' in self.opts:
pagename = parse(
self.opts['date']).strftime(':Journal:%Y:%m:%d')
else:
print self.opts
raise Exception, 'you must somehow identify a page to modify'
print 'Pagename=', pagename
ui = None
notebook = None
if (zim66):
server = ZIM_APPLICATION
#print ZIM_APPLICATION._running
for window in ZIM_APPLICATION._windows:
if window.ui.notebook.uri == notebookInfo.uri:
notebook = window.ui.notebook
ui = window.ui
break
else:
logger.debug("not it: '%s' != '%s'",
window.ui.notebook.uri, notebookInfo.uri)
else:
start_server_if_not_running()
server = ServerProxy()
pprint.pprint(server.list_objects())
ui = server.get_notebook(notebookInfo, _raise or _show)
notebook = ui.notebook
print 'Server=', server
print 'UI=', ui
print 'Notebook?=', notebook
quoting = ('quote' in self.opts)
text = ''
emptyString = False
if 'literal' in self.opts:
if type(self.opts['literal']) == bool:
emptyString = True
else:
text += self.opts['literal']
if 'time' in self.opts:
print "time(): ", time.time()
print "timezone: ", time.tzname
print "localtime: ", time.localtime()
if pagename == todaysJournal:
# It's log-like... all the same day... so don't include the full date...
text = strftime('%I:%M%P - ') + text
else:
text = strftime('%Y-%m-%d @ %I:%M%P - ') + text
if 'file' in self.opts:
#if not quoting:
# text += '\n{0}:\n'.format(self.opts['file'])
text += open(self.opts['file']).read()
if 'clipboard' in self.opts:
text += SelectionClipboard.get_text() or Clipboard.get_text()
# BUG: This does not work, because this code executes in the main zim process...
# we need a pre-handoff hook to convert it to a '--literal', or similar.
if 'stdin' in self.opts:
import sys
text += sys.stdin.read()
# --------------------------------------------------------------------------------
if text and quoting:
text = "'''\n{0}\n'''".format(text)
didSomething = False
if text or emptyString:
# BUG: the journal template is not used for new pages...
# BUG: If the page does not exist, then we assume that it cannot be 'open'... while generally true, this is probably technically incorrect for stub pages just-navigated-to (but not yet saved).
if self.pageExists(notebookInfo, pagename):
print 'Page exists...'
if 'create' in self.opts:
raise Exception, 'Page already exists: ' + pagename
if ui is None:
self._direct_append(notebookInfo, pagename, text)
else:
ui.append_text_to_page(pagename, text)
elif ui is None:
self._direct_create(notebookInfo, pagename, text)
elif self.likelyHasChildPages(ui, notebookInfo, pagename):
print 'Page dne, but has children... yuck...'
# The new_page_from_text function would create enumerated side-pages...
# so we can't use the template when creating a new page... :-(
#text = (
# "====== " + pagename + " ======\n"
# "https://github.com/Osndok/zim-plugin-append/issues/1\n\n"
# +text
#)
#ui.append_text_to_page(pagename, text);
path = ui.notebook.pages.lookup_from_user_input(pagename)
page = ui.notebook.get_page(
path) # as opposed to 'get_new_page' (!!!!)
parsetree = ui.notebook.get_template(page)
page.set_parsetree(parsetree)
page.parse('wiki', text,
append=True) # FIXME format hard coded
ui.notebook.store_page(page)
else:
print 'Page does not exist'
if 'exists' in self.opts:
raise Exception, 'Page does not exist: ' + pagename
ui.new_page_from_text(text,
name=pagename,
use_template=True)
didSomething = True
#BUG: these features don't work without 'ui'...
if 'directory' in self.opts:
ui.import_attachments(path, self.opts['directory'])
didSomething = True
if 'attach' in self.opts:
if zim66:
attachments = notebook.get_attachments_dir(path)
file = dir.file(name)
if file.isdir():
print 'BUG: dont know how to handle folders in 0.66'
else:
file.copyto(attachments)
else:
ui.do_attach_file(path, self.opts['attach'])
if _raise or _show:
if ui:
ui.present(pagename)
didSomething = True
else:
print 'ERROR: unable to raise/show window, no UI running'
def main(*args):
options = {}
template_options = {}
for arg in args:
if arg.startswith("option:"):
arg = arg[7:]
dict = template_options
else:
dict = options
if "=" in arg:
key, value = arg.split("=", 1)
dict[key] = value
else:
dict[arg] = True
# ~ print 'OPTIONS:', options, template_options
if "help" in options:
print usagehelp
return
if "notebook" in options:
notebook, page = resolve_notebook(options["notebook"])
else:
notebook = None
if "append" in options:
if options["append"].lower() == "true":
options["append"] = True
else:
options["append"] = False
if "input" in options:
if options["input"] == "stdin":
import sys
text = sys.stdin.read()
elif options["input"] == "clipboard":
text = SelectionClipboard.get_text() or Clipboard.get_text()
else:
text = options.get("text")
if text and options.get("encoding"):
if options["encoding"] == "base64":
import base64
text = base64.b64decode(text)
elif options["encoding"] == "url":
from zim.parsing import url_decode, URL_ENCODE_DATA
text = url_decode(text, mode=URL_ENCODE_DATA)
else:
raise AssertionError, "Unknown encoding: %s" % options["encoding"]
if text and not isinstance(text, unicode):
text = text.decode("utf-8")
icon = data_file("zim.png").path
gtk_window_set_default_icon()
dialog = QuickNoteDialog(
None,
notebook,
options.get("namespace"),
options.get("basename"),
options.get("append"),
text,
template_options,
options.get("attachments"),
)
dialog.run()
