def testMain(self): def has_text(text): # create the actual check function def my_has_text(dialog): self.assertIsInstance(dialog, QuickNoteDialog) buffer = dialog.textview.get_buffer() result = buffer.get_text(*buffer.get_bounds()) #~ print result self.assertTrue(text in result) return my_has_text # Text on commandline text = 'foo bar baz\ndus 123' with tests.DialogContext(has_text(text)): main('text=' + text) # Clipboard input text = 'foo bar baz\ndus 123' SelectionClipboard.clipboard.clear() # just to be sure Clipboard.set_text(text) with tests.DialogContext(has_text(text)): main('input=clipboard') text = 'foo bar baz\ndus 456' SelectionClipboard.set_text(text) with tests.DialogContext(has_text(text)): main('input=clipboard')
def testMain(self): # Text on commandline text = 'foo bar baz\ndus 123' self.assertRun(('text=' + text,), text) self.assertRun(('--text', text), text) encoded = 'Zm9vIGJhciBiYXoKZHVzIDEyMwo=' self.assertRun(('--text', encoded, '--encoding', 'base64'), text) encoded = 'foo%20bar%20baz%0Adus%20123' self.assertRun(('--text', encoded, '--encoding', 'url'), text) # Clipboard input text = 'foo bar baz\ndus 123' SelectionClipboard.clipboard.clear() # just to be sure Clipboard.set_text(text) self.assertRun(('input=clipboard',), text) self.assertRun(('--input', 'clipboard',), text) text = 'foo bar baz\ndus 456' SelectionClipboard.set_text(text) self.assertRun(('input=clipboard',), text) self.assertRun(('--input', 'clipboard',), text) # Template options cmd = QuickNotePluginCommand('quicknote') cmd.parse_options('option:url=foo') self.assertEqual(cmd.template_options, {'url': 'foo'}) cmd = QuickNotePluginCommand('quicknote') cmd.parse_options('--option', 'url=foo') self.assertEqual(cmd.template_options, {'url': 'foo'})
def testMain(self): def main(*args): cmd = QuickNotePluginCommand('quicknote') cmd.parse_options(*args) cmd.run() def has_text(text): # create the actual check function def my_has_text(dialog): self.assertIsInstance(dialog, QuickNoteDialog) buffer = dialog.textview.get_buffer() result = buffer.get_text(*buffer.get_bounds()) self.assertTrue(text in result) return my_has_text # Text on commandline text = 'foo bar baz\ndus 123' with tests.DialogContext(has_text(text)): main('text=' + text) with tests.DialogContext(has_text(text)): main('--text', text) encoded = 'Zm9vIGJhciBiYXoKZHVzIDEyMwo=' with tests.DialogContext(has_text(text)): main('--text', encoded, '--encoding', 'base64') encoded = 'foo%20bar%20baz%0Adus%20123' with tests.DialogContext(has_text(text)): main('--text', encoded, '--encoding', 'url') # Clipboard input text = 'foo bar baz\ndus 123' SelectionClipboard.clipboard.clear() # just to be sure Clipboard.set_text(text) with tests.DialogContext(has_text(text)): main('input=clipboard') with tests.DialogContext(has_text(text)): main('--input', 'clipboard') text = 'foo bar baz\ndus 456' SelectionClipboard.set_text(text) with tests.DialogContext(has_text(text)): main('input=clipboard') with tests.DialogContext(has_text(text)): main('--input', 'clipboard') # Template options cmd = QuickNotePluginCommand('quicknote') cmd.parse_options('option:url=foo') self.assertEqual(cmd.template_options, {'url': 'foo'}) cmd = QuickNotePluginCommand('quicknote') cmd.parse_options('--option', 'url=foo') self.assertEqual(cmd.template_options, {'url': 'foo'})
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 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 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()