python/importer/import_qq_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_qq_user_dict (fname):
   8     result = []
   9     f = codecs.open (fname, "r", "UTF-16")
  10     for l in f:
  11         if l[0] in ('\n', '\r'):
  12             continue
  13
  14         try:
  15             pystr, utf8str, _ = l.strip().split(",")[0].split(" ")
  16             result.append ((pystr[1:], utf8str))
  17         except:
  18             pass
  19
  20     return result
  21
  22 def main ():
  23     if len (sys.argv) != 2:
  24         print "Please specify the QQ PinYin exported user dict file!"
  25         exit (1)
  26
  27     qq_user_dict = load_qq_user_dict(sys.argv[1])
  28     import_to_sunpinyin_user_dict (qq_user_dict)
  29
  30 if __name__ == "__main__":
  31     main()