X-Git-Url: https://git.korewanetadesu.com/?p=python-collate.git;a=blobdiff_plain;f=collate%2Ficu%2F__init__.py;h=00a15386ea1cc55f6698901d16976ffddf6c1377;hp=eeee418070483d0797b63b639d1aa7bca219442e;hb=f73c4c6cd3ed326c5735ab33a6896697227d07e3;hpb=08e8f0a8bb8d7276d114087f8ff8dbce4acdb1cf

diff --git a/collate/icu/__init__.py b/collate/icu/__init__.py
index eeee418..00a1538 100644
--- a/collate/icu/__init__.py
+++ b/collate/icu/__init__.py
@@ -11,22 +11,36 @@ Avoid this backend if...
 
 """
 
-import collate.icu._ucol
 import collate._abcollator
 import collate._locale
 import collate.errors
 
+from collate.icu import _icu
+
 class Collator(collate._abcollator.Collator):
     """ICU-based collation."""
 
     def __init__(self, locale, encoding=None):
         locale, encoding = collate._locale.getpair(locale, encoding)
-        self._collator = collate.icu._ucol.Collator(locale)
+        icu_locale = "root" if locale == "C" else locale
+        self._collator = _icu.Collator(icu_locale)
         self.locale = self._collator.locale
         self.encoding = collate._locale.encoding(encoding)
         if self._collator.used_default_information and locale != "C":
             raise collate.errors.InvalidLocaleError(locale)
 
+        try:
+            self._breaker = _icu.WordBreaker(icu_locale)
+        except ValueError:
+            # Thai is the only language with a special break locale,
+            # so this is a harmless error.
+            self._breaker = _icu.WordBreaker("root")
+
+    def words(self, string):
+        if isinstance(string, str):
+            string = string.decode(self.encoding, 'replace')
+        return filter(lambda u: not u.isspace(), self._breaker.words(string))
+
     def key(self, string):
         """Sort key for a string.