def testNote2buffer(self): # {{{
editor = EvervimEditor.getInstance()
note = Types.Note()
note.title = u'タイトルテスト'.encode('utf-8')
note.tagNames = [u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')]
note.content = u"""<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE en-note SYSTEM "http://xml.evernote.com/pub/enml2.dtd"><en-note>this is content
本文テスト
<h3>たぐ３</h3>
</en-note>
""".encode('utf-8')
EvervimPref.getInstance().usemarkdown = '0' # dont use markdown
EvervimPref.getInstance().xmlindent = ' ' # default ts=4
xmlStrings = editor.note2buffer(note)
self.assertEqual(u'タイトルテスト'.encode('utf-8'), xmlStrings[0])
self.assertEqual(u'Tags:タグ１,*タグ２'.encode('utf-8'), xmlStrings[1])
self.assertEqual('this is content'.encode('utf-8'), xmlStrings[2]) # this is content
self.assertEqual('本文テスト'.encode('utf-8'), xmlStrings[3]) # 本文テスト
self.assertEqual('<h3>'.encode('utf-8'), xmlStrings[4]) # <h3>
self.assertEqual(' たぐ３'.encode('utf-8'), xmlStrings[5]) # たぐ３
self.assertEqual('</h3>'.encode('utf-8'), xmlStrings[6]) # </h3>
EvervimPref.getInstance().usemarkdown = '1' # dont use markdown
mkdStrings = editor.note2buffer(note)
self.assertEqual(u'# タイトルテスト'.encode('utf-8'), mkdStrings[0])
self.assertEqual(u'Tags:タグ１,*タグ２'.encode('utf-8'), xmlStrings[1])
self.assertEqual('this is content'.encode('utf-8'), mkdStrings[2])
self.assertEqual('本文テスト'.encode('utf-8'), mkdStrings[3])
self.assertEqual('### たぐ３'.encode('utf-8'), mkdStrings[4])
def testNote2buffer(self): # {{{
editor = EvervimEditor.getInstance()
note = Types.Note()
note.title = u'タイトルテスト'.encode('utf-8')
note.tagNames = [u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')]
note.content = u"""<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE en-note SYSTEM "http://xml.evernote.com/pub/enml2.dtd"><en-note>this is content
本文テスト
<h3>たぐ３</h3>
</en-note>
""".encode('utf-8')
EvervimPref.getInstance().usemarkdown = '0' # dont use markdown
EvervimPref.getInstance().xmlindent = ' ' # default ts=4
xmlStrings = editor.note2buffer(note)
self.assertEqual(u'タイトルテスト'.encode('utf-8'), xmlStrings[0])
self.assertEqual(u'Tags:タグ１,*タグ２'.encode('utf-8'), xmlStrings[2])
self.assertEqual('this is content'.encode('utf-8'),
xmlStrings[3]) # this is content
self.assertEqual('本文テスト'.encode('utf-8'), xmlStrings[4]) # 本文テスト
self.assertEqual('<h3>'.encode('utf-8'), xmlStrings[5]) # <h3>
self.assertEqual(' たぐ３'.encode('utf-8'), xmlStrings[6]) # たぐ３
self.assertEqual('</h3>'.encode('utf-8'), xmlStrings[7]) # </h3>
EvervimPref.getInstance().usemarkdown = '1' # dont use markdown
mkdStrings = editor.note2buffer(note)
self.assertEqual(u'# タイトルテスト'.encode('utf-8'), mkdStrings[0])
self.assertEqual(u'Tags:タグ１,*タグ２'.encode('utf-8'), xmlStrings[2])
self.assertEqual('this is content'.encode('utf-8'), mkdStrings[3])
self.assertEqual('本文テスト'.encode('utf-8'), mkdStrings[4])
self.assertEqual('### たぐ３'.encode('utf-8'), mkdStrings[5])
def testPref(self): # {{{
pref = EvervimPref._instance = None
pref = EvervimPref.getInstance()
self.assertIsNone(pref.workdir)
self.assertIsNone(pref.devtoken)
self.assertIsNone(pref.sortnotebooks)
self.assertIsNone(pref.sorttags)
self.assertIsNone(pref.xmlindent)
self.assertIsNone(pref.usemarkdown)
self.assertRaises(AttributeError, lambda: pref.zzzzzzzzzzzz)
self.assertRaises(RuntimeError, lambda: EvervimPref())
def testBuffer2note(self): # {{{
editor = EvervimEditor.getInstance()
pref = EvervimPref.getInstance()
pref.username = USERNAME
pref.password = PASSWORD
editor.setAPI()
note = Types.Note()
xmlBufferHead = u"""タイトルテスト
タグ１,*タグ２
""".encode('utf-8')
xmlBufferContent = u"""this is content
本文テスト
<h3>たぐ３</h3>""".encode('utf-8')
EvervimPref.getInstance().usemarkdown = '0' # dont use markdown
editednote = editor.buffer2note(note, (xmlBufferHead + xmlBufferContent).splitlines())
self.assertEqual(u'タイトルテスト'.encode('utf-8'), editednote.title)
self.assertEqual([u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')], editednote.tagNames)
self.assertEqual(EvernoteAPI.NOTECONTENT_HEADER + xmlBufferContent + EvernoteAPI.NOTECONTENT_FOOTER, editednote.content)
EvervimPref.getInstance().usemarkdown = '1'
note = Types.Note()
mkdBuffer = u"""# タイトルテスト
タグ１,*タグ２
this is content
本文テスト
### たぐ３""".encode('utf-8')
mkdConverted = u"""<p>this is content
本文テスト</p>
<h3>たぐ３</h3>""".encode('utf-8')
mkdeditednote = editor.buffer2note(note, mkdBuffer.splitlines())
self.assertEqual(u'タイトルテスト'.encode('utf-8'), mkdeditednote.title)
self.assertEqual([u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')], mkdeditednote.tagNames)
self.assertEqual(EvernoteAPI.NOTECONTENT_HEADER + mkdConverted + EvernoteAPI.NOTECONTENT_FOOTER, mkdeditednote.content)
# 1行目の先頭が #で始まっていない場合も、テストが通ること
note = Types.Note()
mkdBuffer = u"""タイトルテスト ### 途中#
タグ１,*タグ２
this is content
本文テスト
### たぐ３""".encode('utf-8')
mkdConverted = u"""<p>this is content
本文テスト</p>
<h3>たぐ３</h3>""".encode('utf-8')
mkdeditednote = editor.buffer2note(note, mkdBuffer.splitlines())
self.assertEqual(u'タイトルテスト ### 途中#'.encode('utf-8'), mkdeditednote.title)
self.assertEqual([u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')], mkdeditednote.tagNames)
self.assertEqual(EvernoteAPI.NOTECONTENT_HEADER + mkdConverted + EvernoteAPI.NOTECONTENT_FOOTER, mkdeditednote.content)
def testBuffer2note(self): # {{{
editor = EvervimEditor.getInstance()
pref = EvervimPref.getInstance()
pref.devtoken = DEVELOPER_TOKEN
editor.setAPI()
note = Types.Note()
xmlBufferHead = u"""タイトルテスト
タグ１,*タグ２
""".encode('utf-8')
xmlBufferContent = u"""this is content
本文テスト
<h3>たぐ３</h3>""".encode('utf-8')
EvervimPref.getInstance().usemarkdown = '0' # dont use markdown
editednote = editor.buffer2note(note, (xmlBufferHead + xmlBufferContent).splitlines())
self.assertEqual(u'タイトルテスト'.encode('utf-8'), editednote.title)
self.assertEqual([u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')], editednote.tagNames)
self.assertEqual(EvernoteAPI.NOTECONTENT_HEADER + xmlBufferContent + EvernoteAPI.NOTECONTENT_FOOTER, editednote.content)
EvervimPref.getInstance().usemarkdown = '1'
note = Types.Note()
mkdBuffer = u"""# タイトルテスト
タグ１,*タグ２
this is content
本文テスト
### たぐ３""".encode('utf-8')
mkdConverted = u"""<p>this is content
本文テスト</p>
<h3>たぐ３</h3>""".encode('utf-8')
mkdeditednote = editor.buffer2note(note, mkdBuffer.splitlines())
self.assertEqual(u'タイトルテスト'.encode('utf-8'), mkdeditednote.title)
self.assertEqual([u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')], mkdeditednote.tagNames)
self.assertEqual(EvernoteAPI.NOTECONTENT_HEADER + mkdConverted + EvernoteAPI.NOTECONTENT_FOOTER, mkdeditednote.content)
# 1行目の先頭が #で始まっていない場合も、テストが通ること
note = Types.Note()
mkdBuffer = u"""タイトルテスト ### 途中#
タグ１,*タグ２
this is content
本文テスト
### たぐ３""".encode('utf-8')
mkdConverted = u"""<p>this is content
本文テスト</p>
<h3>たぐ３</h3>""".encode('utf-8')
mkdeditednote = editor.buffer2note(note, mkdBuffer.splitlines())
self.assertEqual(u'タイトルテスト ### 途中#'.encode('utf-8'), mkdeditednote.title)
self.assertEqual([u'タグ１'.encode('utf-8'), u'*タグ２'.encode('utf-8')], mkdeditednote.tagNames)
self.assertEqual(EvernoteAPI.NOTECONTENT_HEADER + mkdConverted + EvernoteAPI.NOTECONTENT_FOOTER, mkdeditednote.content)
def testSetAPI(self): # {{{
pref = EvervimPref.getInstance()
pref.username = None
pref.password = None
self.assertRaises(AttributeError, lambda: self.editor.setAPI())
self.setPrefUserName()
self.editor.setAPI()
self.assertTrue(True)
def testSetAPI(self): # {{{
pref = EvervimPref.getInstance()
pref.username = None
pref.password = None
self.assertRaises(AttributeError, lambda: self.editor.setAPI())
self.setPrefUserName()
self.editor.setAPI()
self.assertTrue(True)
def testPref(self): # {{{
pref = EvervimPref._instance = None
pref = EvervimPref.getInstance()
self.assertIsNone(pref.workdir)
self.assertIsNone(pref.devtoken)
self.assertIsNone(pref.sortnotebooks)
self.assertIsNone(pref.sorttags)
self.assertIsNone(pref.xmlindent)
self.assertIsNone(pref.usemarkdown)
self.assertRaises(AttributeError, lambda: pref.zzzzzzzzzzzz)
self.assertRaises(RuntimeError, lambda: EvervimPref())
def setPrefdevtoken(self): # {{{
pref = EvervimPref.getInstance()
pref.devtoken = DEVELOPER_TOKEN
def testSetAPI(self): # {{{
pref = EvervimPref.getInstance()
self.assertRaises(AttributeError, lambda: self.editor.setAPI())
self.setPrefdevtoken()
self.editor.setAPI()
self.assertTrue(True)
class Evervimmer(object):
""" interface to vim """
_instance = None
def __init__(self):
if Evervimmer._instance is not None:
raise RuntimeError("EvervimPref must be one object!!!")
@classmethod
def getInstance(self):
if Evervimmer._instance is None:
Evervimmer._instance = Evervimmer()
Evervimmer._instance.setPref()
Evervimmer._instance.setAPI()
return Evervimmer._instance
# recentry loaded
currentnote = None
notes = []
notebooks = []
tags = []
maxpages = 0
currentpage = 0
editor = EvervimEditor.getInstance()
pref = EvervimPref.getInstance()
""" prefs from vim option """
def setPref(self): # {{{
self.pref.workdir = vim.eval("g:evervim_workdir")
self.pref.devtoken = vim.eval("g:evervim_devtoken")
self.pref.sortnotebooks = vim.eval("g:evervim_sortnotebooks")
self.pref.sorttags = vim.eval("g:evervim_sorttags")
self.pref.xmlindent = vim.eval("g:evervim_xmlindent")
self.pref.usemarkdown = vim.eval("g:evervim_usemarkdown")
self.pref.asyncupdate = vim.eval("g:evervim_asyncupdate")
self.pref.encoding = vim.eval('&enc')
self.pref.enscriptpath = None
# }}}
def setAPI(self): # {{{
"""
setup API
"""
Evervimmer.editor.setAPI()
#}}}
def auth(self): # {{{
""" auth """
Evervimmer.editor.api.auth()
#}}}
def notesByNotebook(self, notebook=None): # {{{
""" get notelist by notebook """
if notebook == None: # notebook is selected on list.( else then use currentNotebook for paginate.)
self.currentNotebook = Evervimmer.notebooks[self.__getArrayIndexByCurrentLine()]
self.currentpage = 0
noteList = Evervimmer.editor.api.notesByNotebook(self.currentNotebook, self.currentpage)
self.__setNoteListPrameter(noteList)
notetitles = [self.__changeEncodeToBuffer(note.title) for note in Evervimmer.notes]
self.__setBufferList(notetitles,
" [notebook:%s(%s)] page:%s/%s [\">\":next, \"<\":prev]" % (self.__changeEncodeToBuffer(self.currentNotebook.name) ,
Evervimmer.maxcount,
Evervimmer.currentpage + 1,
Evervimmer.maxpages + 1
))
#}}}
def notesByNotebookNextpage(self): # {{{
if self.currentpage == self.maxpages:
return
self.currentpage += 1
self.notesByNotebook(self.currentNotebook)
#}}}
def notesByNotebookPrevpage(self): # {{{
if self.currentpage == 0:
return
self.currentpage -= 1
self.notesByNotebook(self.currentNotebook)
#}}}
def notesByTag(self, tag=None): # {{{
if tag == None:
self.currentTag = Evervimmer.tags[self.__getArrayIndexByCurrentLine()]
self.currentpage = 0
noteList = Evervimmer.editor.api.notesByTag(self.currentTag, self.currentpage)
self.__setNoteListPrameter(noteList)
notetitles = [self.__changeEncodeToBuffer(note.title) for note in Evervimmer.notes]
self.__setBufferList(notetitles,
" [tag:%s(%s)] page:%s/%s [\">\":next, \"<\":prev]" % (self.__changeEncodeToBuffer(self.currentTag.name) ,
Evervimmer.maxcount,
Evervimmer.currentpage + 1,
Evervimmer.maxpages + 1
))
#}}}
def notesByTagNextpage(self): # {{{
if self.currentpage == self.maxpages:
return
self.currentpage += 1
self.notesByTag(self.currentTag)
#}}}
def notesByTagPrevpage(self): # {{{
if self.currentpage == 0:
return
self.currentpage -= 1
self.notesByTag(self.currentTag)
#}}}
def listNotebooks(self): # {{{
Evervimmer.notebooks = Evervimmer.editor.api.listNotebooks()
sortOpt = vim.eval('g:evervim_sortnotebooks').split()
if sortOpt[1] == 'asc':
Evervimmer.notebooks.sort(lambda a, b: cmp(getattr(a, sortOpt[0]),
getattr(b, sortOpt[0])))
else:
Evervimmer.notebooks.sort(lambda a, b: cmp(getattr(b, sortOpt[0]),
getattr(a, sortOpt[0])))
strs = [self.__changeEncodeToBuffer(notebook.name) for notebook in Evervimmer.notebooks]
self.__setBufferList(strs, " [all notebooks]")
#}}}
def listTags(self): # {{{
Evervimmer.tags = Evervimmer.editor.api.listTags()
sortOpt = vim.eval('g:evervim_sorttags').split()
if sortOpt[1] == 'asc':
Evervimmer.tags.sort(lambda a, b: cmp(getattr(a, sortOpt[0]),
getattr(b, sortOpt[0])))
else:
Evervimmer.tags.sort(lambda a, b: cmp(getattr(b, sortOpt[0]),
getattr(a, sortOpt[0])))
strs = [self.__changeEncodeToBuffer(tag.name) for tag in Evervimmer.tags]
self.__setBufferList(strs, " [all tags]")
#}}}
def checkNote(self): # {{{
""" check note format """
bufstrs = [self.__changeEncodeFromBuffer(line) for line in vim.current.buffer[:]]
note = self.editor.buffer2note(Evervimmer.currentnote, bufstrs)
minidom.parseString(note.content)
Evervimmer.currentnote = note
if len(note.title) == 0:
raise StandardError("*** must set title! ***")
#}}}
def updateNoteInthread(self, note): # {{{
Evervimmer.editor.api.updateNote(note)
vim.command("echo 'evernote update successful.'")
#}}}
def updateNote(self): # {{{
self.checkNote()
if self.pref.asyncupdate == '1':
note = copy.deepcopy(Evervimmer.currentnote)
if hasattr(self, 'updatethread') and self.updatethread.isAlive():
vim.command("echohl WarningMsg | echomsg 'now updating... so not update this save. do save buffer later.' | echohl None")
return
self.updatethread = threading.Thread(target=self.updateNoteInthread, args=(note,))
self.updatethread.start()
else:
Evervimmer.editor.api.updateNote(Evervimmer.currentnote)
#}}}
def searchByQuery(self, query=None): # {{{
if query == None:
self.query = vim.eval("a:word")
self.currentpage = 0
noteList = Evervimmer.editor.api.notesByQuery(self.query, self.currentpage)
self.__setNoteListPrameter(noteList)
notetitles = [self.__changeEncodeToBuffer(note.title) for note in Evervimmer.notes]
self.__setBufferList(notetitles,
" [query:%s(%s)] page:%s/%s [\">\":next, \"<\":prev]" % (self.__changeEncodeToBuffer(self.query) ,
Evervimmer.maxcount,
Evervimmer.currentpage + 1,
Evervimmer.maxpages + 1
))
#}}}
def searchByQueryNextpage(self): # {{{
if self.currentpage == self.maxpages:
return
self.currentpage += 1
self.searchByQuery(self.query)
#}}}
def searchByQueryPrevpage(self): # {{{
if self.currentpage == 0:
return
self.currentpage -= 1
self.searchByQuery(self.query)
#}}}
def createNote(self): # {{{
try:
Evervimmer.currentnote = Evervimmer.editor.api.newNote()
self.checkNote()
createdNote = Evervimmer.editor.api.createNote(Evervimmer.currentnote)
Evervimmer.currentnote = createdNote
except:
print traceback.format_exc(sys.exc_info()[2])
raise StandardError("createNote error! aborted.")
#}}}
def getNote(self): # {{{
currentline = int(vim.eval('l:pointer'))
selectedNote = Evervimmer.notes[currentline - 2]
note = Evervimmer.editor.api.getNote(selectedNote)
Evervimmer.currentnote = note
vim.current.buffer[:] = None # clear buffer
lines = [self.__changeEncodeToBuffer(line) for line in self.editor.note2buffer(note)]
vim.current.buffer[0] = lines[0]
for line in lines[1:]:
vim.current.buffer.append(line)
#}}}
def currentNoteOpenBrowser(self): # {{{
if None == Evervimmer.currentnote or None == Evervimmer.currentnote.guid:
return
self.__openBrowser(Evervimmer.currentnote.guid)
#}}}
def cursorNoteOpenBrowser(self): # {{{
currentline = int(vim.eval("line('.')"))
if currentline < 2:
return
selectedNote = Evervimmer.notes[currentline - 2]
self.__openBrowser(selectedNote.guid)
#}}}
def currentNoteOpenClient(self): # {{{
if None == Evervimmer.currentnote or None == Evervimmer.currentnote.guid:
return
self.__openClient(Evervimmer.currentnote.title)
#}}}
def cursorNoteOpenClient(self): # {{{
currentline = int(vim.eval("line('.')"))
if currentline < 2:
return
selectedNote = Evervimmer.notes[currentline - 2]
self.__openClient(selectedNote.title)
#}}}
# ----- private methods
def __openBrowser(self, guid): # {{{
uri = "https://www.evernote.com/view/" + guid
vim.command(":OpenBrowser " + uri)
#}}}
def __openClient(self, title): # {{{ NOTE:this is beta.
if self.pref.enscriptpath is None:
try:
import _winreg
reg = _winreg.OpenKey(_winreg.HKEY_LOCAL_MACHINE, 'SOFTWARE\Microsoft\Windows\CurrentVersion\App Paths\ENScript.exe')
self.pref.enscriptpath = _winreg.EnumValue(reg, 0)[1].encode('shift_jis')
except:
print '_winreg error!'
pass
title_sjis = unicode(title, 'utf-8', 'ignore').encode('shift_jis')
subprocess.Popen(self.pref.enscriptpath + " showNotes /q intitle:\"%s\"" % title_sjis)
#}}}
def __setBufferList(self, buffertitlelist, title): # {{{
vim.current.buffer[:] = None # clear buffer
vim.current.buffer[0] = title # remove empty line(line 1)
for bufline in buffertitlelist: # 一括だとMemoryError
vim.current.buffer.append(bufline)
# }}}
def __getArrayIndexByCurrentLine(self): # {{{
""" get index ** 1st line is title """
try:
index = int(vim.eval('line(".")')) - 2
return index if index >= 0 else 0
except:
return 0
# }}}
def __changeEncodeFromBuffer(self, string): # {{{
""" change &enc to utf-8 """
if(self.pref.encoding == 'utf-8'):
return string
else:
try:
return unicode(string, self.pref.encoding).encode('utf-8')
except:
return string
# }}}
def __changeEncodeToBuffer(self, string): # {{{
""" change utf-8 to &enc"""
if(self.pref.encoding == 'utf-8'):
return string
else:
try:
return unicode(string, 'utf-8').encode(self.pref.encoding)
except:
return string
# }}}
def __setNoteListPrameter(self, noteList): # {{{
""" set host variable from noteList """
Evervimmer.notes = noteList.elem
Evervimmer.maxpages = noteList.maxpages
Evervimmer.currentpage = noteList.currentpage
Evervimmer.maxcount = noteList.maxcount
def setPrefUserName(self): # {{{
pref = EvervimPref.getInstance()
pref.username = USERNAME
pref.password = PASSWORD
def setPrefdevtoken(self): # {{{
pref = EvervimPref.getInstance()
pref.devtoken = DEVELOPER_TOKEN
def testSetAPI(self): # {{{
pref = EvervimPref.getInstance()
self.assertRaises(AttributeError, lambda: self.editor.setAPI())
self.setPrefdevtoken()
self.editor.setAPI()
self.assertTrue(True)
def setPrefUserName(self): # {{{
pref = EvervimPref.getInstance()
pref.username = USERNAME
pref.password = PASSWORD
