python/importer/import_sogou_userdict.py

   1 #!/usr/bin/python
   2
   3 import os, sys
   4 import codecs
   5 from importer import import_to_sunpinyin_user_dict
   6
   7 def load_sogou_user_dict (fname):
   8     result = []
   9     f = codecs.open (fname, "r", "UTF-16")
  10
  11     for l in f:
  12         if l[0] in (';', '\n', '\r'):
  13             continue
  14
  15         pystr, utf8str = l.strip().split(" ")
  16         result.append ((pystr[1:], utf8str))
  17
  18     return result
  19
  20 def main ():
  21     if len (sys.argv) != 2:
  22         print "Please specify the Sogou PinYin exported user dict file!"
  23         exit (1)
  24
  25     sogou_user_dict = load_sogou_user_dict(sys.argv[1])
  26     import_to_sunpinyin_user_dict (sogou_user_dict)
  27
  28 if __name__ == "__main__":
  29     main()