python写vim script 之 山寨版 GTD

一直想找个用vim来管理todo列表的script, 没发现特别好用的,
自己写了个,用sqlite来保存数据.
将下面代码存为 SzTodo.vim，放到plugin目录里.
用 :SzTodo  启动.

let  g:sztodo_db_path="/root/.vim/todo"
let  s:list_type="unfinished"
let  s:cur_buf = 0

function! MakeTemplate() 
python << EOF
import vim   

vim.command("call SwitchToDetailView()")
vim.command("call SetSyntax()")
template = "=" * 50 + "\n" \
      + "tag:" +"\n" \
      + "title:" +  "\n" \
      + "=" * 50 + "\n" 
for index,line in enumerate(template.split("\n")):
  if index ==0 :
    vim.current.buffer[0]=line
  else :
    vim.current.buffer.append(line)

EOF
endfunction

function! SaveTodoItem()
python << EOF
import vim   

def loadData(lines):
  item=ToDoItem()
  content=""
  seperates=0
  for line in lines:
    if line.startswith("====="): seperates=seperates+1
    if line.startswith("id:"): item.id=line[3:].strip()
    if line.startswith("tag:"): item.tag=line[4:].strip()
    if line.startswith("title:"): item.title=line[6:].strip()
    if line.startswith("status:"): item.status=line[7:].strip()
    if seperates==2 and not line.startswith("===="):
      content=content+line+"\n"
  item.content=content[:-1]
  return item

def addItem(item):
  insertSql="insert into SzTodo(tag,title,create_date,status,content) values (?,?,?,?,?)"
  con=sqlite.connect(getDbFileName())
  cur=con.cursor()
  values=(item.tag,item.title,item.create_date,item.status,item.content)
  cur.execute(insertSql,values)
  con.commit()
  con.close()

def updateItem(item):
  updateSql="update SzTodo set tag=?,title=?,create_date=?,status=?,content=? where id=?"
  con=sqlite.connect(getDbFileName())
  cur=con.cursor()
  values=(item.tag,item.title,item.create_date,item.status,item.content,item.id)
  cur.execute(updateSql,values)
  con.commit()
  con.close()


data=vim.current.buffer[:]
todoItem=loadData(data)
if not todoItem.title:
  print "title can't be empty"
else:
  if todoItem.id.strip() == "" :
    todoItem.create_date=getCurrentDate()
    todoItem.status="unstarted"
    addItem(todoItem)
  else :
    updateItem(todoItem)
  print "todo has been saved"

EOF
  exec bufwinnr(s:cur_buf) . "wincmd w"  
  call ListItems()

endfunction

function! ListItems()
python << EOF
import vim   

listFinished=vim.eval("s:list_type")

vim.current.buffer[:]=None
vim.command("set nonumber")
if listFinished=="finished":
  selectSql="select id,tag,title,create_date,content from SzTodo where status == 'done' "
else:
  selectSql="select id,tag,title,create_date,content from SzTodo where status != 'done' "

con=sqlite.connect(getDbFileName())
cur=con.cursor()
items=[]
cur.execute(selectSql)


for index,row in enumerate(cur):  
  formatedItem=str(row[0])+". "+str(unicode(row[2]).encode("utf-8"))
  if index==0:
    vim.current.buffer[0]=formatedItem
  else :
    vim.current.buffer.append(formatedItem)

con.commit()
con.close()

EOF
endfunction

function! SwitchToDetailView()  
  let s:cur_buf = bufnr("%")  
  let s:szdb_result_buf=bufnr("SztodoDetail")  
  if bufwinnr(s:szdb_result_buf) > 0  
    exec bufwinnr(s:szdb_result_buf) . "wincmd w"  
    %d  
  else  
    exec 'silent! botright split SztodoDetail'   
    exec "e SztodoDetail"  
    exec "set nowrap"  
    map <silent><buffer>s :call SaveTodoItem()<cr>
  endif  
  
endfunction  

function! ShowItemDetail(preview)
python << EOF
import vim   
(row, col) = vim.current.window.cursor
line = vim.current.buffer[row-1]
id=line[0:line.find(".")]
selectSql="select id,tag,title,create_date,content,status from SzTodo where id=?"
con=sqlite.connect(getDbFileName())
cur=con.cursor()
cur.execute(selectSql,(id,))

todoItem=ToDoItem()
for row in cur:  
  todoItem.id=row[0]
  todoItem.tag=unicode(row[1]).encode("utf-8")
  todoItem.title=unicode(row[2]).encode("utf-8")
  todoItem.create_date=unicode(row[3]).encode("utf-8")
  todoItem.content=unicode(row[4]).encode("utf-8")
  todoItem.status=unicode(row[5]).encode("utf-8")

vim.command("call SwitchToDetailView()")  
vim.command("call SetSyntax()")
for index,line in enumerate(str(todoItem).split("\n")):
  if index==0:
    vim.current.buffer[0]=line
  else:
    vim.current.buffer.append(line)
EOF
  if a:preview=="true"
    exec bufwinnr(s:cur_buf) . "wincmd w"  
  endif

endfunction

function! UpdateItemStatus(status)
  let choice=input('you really want to update the todo item status to '.a:status."?[y/n]")
  if choice=="n"
    return
  endif
python << EOF
import vim
(row, col) = vim.current.window.cursor
line = vim.current.buffer[row-1]
id=line[0:line.find(".")]
updateSql="update SzTodo set status = ? where id=?"
status=vim.eval("a:status")
con=sqlite.connect(getDbFileName())
cur=con.cursor()
cur.execute(updateSql,(status,id))
con.commit()
con.close()
EOF
endfunction

function! InitDb()
python << EOF
import vim   
import os
createSql="create table SzTodo (id integer primary key , tag char(20), title varchar(200), \
  create_date varchar(10),status char(1),content varchar(5000))"
path=os.path.dirname(getDbFileName())
if not os.path.exists(path):
  os.makedirs(path)
con=sqlite.connect(getDbFileName())
cur=con.cursor()
cur.execute(createSql)
con.commit()
con.close()
print "db has been created"

EOF
endfunction

function! s:DefSzTodoGlobal() 
python << EOF
import vim   
from pysqlite2 import dbapi2 as sqlite

statusDict=dict(done="done",postpone="postpone",doing="doing",unstarted="unstarted")

class ToDoItem(object):
  def __init__(self,id="",tag="",title="",content="",create_date="",status=""):
    self.id=id
    self.tag=tag
    self.title=title
    self.create_date=create_date
    self.status=status
    self.content=content

  def __str__(self):
    return "=" * 50 + "\n" \
        + "id:" + str(self.id) +"\n" \
        + "tag:" + self.tag +"\n" \
        + "title:" + self.title + "\n" \
        + "status:" + self.status + "\n" \
        + "=" * 50 + "\n" \
        + self.content 

def getCurrentDate():
  from datetime import datetime
  t=datetime.now()
  return t.strftime("%Y-%m-%d %H:%M")

def getDbFileName():
  dbpath=vim.eval("g:sztodo_db_path")
  path=os.path.join(dbpath,"todo.dat")
  return path

EOF
endfunction

function! StartApp()
python << EOF
import vim   
import os
if not os.path.exists(getDbFileName()):
  vim.command("call InitDb()")
vim.command("call ListItems()")
vim.command("call SetMapping()")
EOF
endfunction


function! SetMapping()
  map <silent><buffer> o  :call ShowItemDetail("false")<cr> 
  map <silent><buffer> s  :call ShowItemDetail("true")<cr> 
  map <silent><buffer> i  :call MakeTemplate()<cr> 
  map <silent><buffer> r  :call ListItems()<cr> 

  map <silent><buffer> p  :call UpdateItemStatus("postpone")<cr> 
  map <silent><buffer> d  :call UpdateItemStatus("done")<cr> 
  command! -nargs=0 FinishedItem :call FinishedItem()
  command! -nargs=0 UnfinishedItem :call UnfinishedItem()
endfunction

function! FinishedItem() 
  let s:list_type="finished"
  call ListItems()
endfunction

function! UnfinishedItem() 
  let s:list_type="unfinished"
  call ListItems()
endfunction

function! SetSyntax()

  syn keyword sztodoKeyword tag title id status 
  syn keyword sztodoStatus unstarted done doing postpone

  syn match tag "^tag:.*" 
  syn match title "^title:.*"
  syn match id  "^id:.*"
  syn match status  "^status:.*"

  hi def link sztodoKeyword Keyword
  hi def link sztodoStatus Identifier
  hi def link tag String
  hi def link id String
  hi def link title String
  hi def link status String

endfunction

call s:DefSzTodoGlobal()
command! -nargs=0 SzTodo :call StartApp()