languages.py 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352
  1. """
  2. Metadata about languages used by our model training code for our
  3. SingleByteCharSetProbers. Could be used for other things in the future.
  4. This code is based on the language metadata from the uchardet project.
  5. """
  6. from string import ascii_letters
  7. from typing import List, Optional
  8. # TODO: Add Ukrainian (KOI8-U)
  9. class Language:
  10. """Metadata about a language useful for training models
  11. :ivar name: The human name for the language, in English.
  12. :type name: str
  13. :ivar iso_code: 2-letter ISO 639-1 if possible, 3-letter ISO code otherwise,
  14. or use another catalog as a last resort.
  15. :type iso_code: str
  16. :ivar use_ascii: Whether or not ASCII letters should be included in trained
  17. models.
  18. :type use_ascii: bool
  19. :ivar charsets: The charsets we want to support and create data for.
  20. :type charsets: list of str
  21. :ivar alphabet: The characters in the language's alphabet. If `use_ascii` is
  22. `True`, you only need to add those not in the ASCII set.
  23. :type alphabet: str
  24. :ivar wiki_start_pages: The Wikipedia pages to start from if we're crawling
  25. Wikipedia for training data.
  26. :type wiki_start_pages: list of str
  27. """
  28. def __init__(
  29. self,
  30. name: Optional[str] = None,
  31. iso_code: Optional[str] = None,
  32. use_ascii: bool = True,
  33. charsets: Optional[List[str]] = None,
  34. alphabet: Optional[str] = None,
  35. wiki_start_pages: Optional[List[str]] = None,
  36. ) -> None:
  37. super().__init__()
  38. self.name = name
  39. self.iso_code = iso_code
  40. self.use_ascii = use_ascii
  41. self.charsets = charsets
  42. if self.use_ascii:
  43. if alphabet:
  44. alphabet += ascii_letters
  45. else:
  46. alphabet = ascii_letters
  47. elif not alphabet:
  48. raise ValueError("Must supply alphabet if use_ascii is False")
  49. self.alphabet = "".join(sorted(set(alphabet))) if alphabet else None
  50. self.wiki_start_pages = wiki_start_pages
  51. def __repr__(self) -> str:
  52. param_str = ", ".join(
  53. f"{k}={v!r}" for k, v in self.__dict__.items() if not k.startswith("_")
  54. )
  55. return f"{self.__class__.__name__}({param_str})"
  56. LANGUAGES = {
  57. "Arabic": Language(
  58. name="Arabic",
  59. iso_code="ar",
  60. use_ascii=False,
  61. # We only support encodings that use isolated
  62. # forms, because the current recommendation is
  63. # that the rendering system handles presentation
  64. # forms. This means we purposefully skip IBM864.
  65. charsets=["ISO-8859-6", "WINDOWS-1256", "CP720", "CP864"],
  66. alphabet="ءآأؤإئابةتثجحخدذرزسشصضطظعغػؼؽؾؿـفقكلمنهوىيًٌٍَُِّ",
  67. wiki_start_pages=["الصفحة_الرئيسية"],
  68. ),
  69. "Belarusian": Language(
  70. name="Belarusian",
  71. iso_code="be",
  72. use_ascii=False,
  73. charsets=["ISO-8859-5", "WINDOWS-1251", "IBM866", "MacCyrillic"],
  74. alphabet="АБВГДЕЁЖЗІЙКЛМНОПРСТУЎФХЦЧШЫЬЭЮЯабвгдеёжзійклмнопрстуўфхцчшыьэюяʼ",
  75. wiki_start_pages=["Галоўная_старонка"],
  76. ),
  77. "Bulgarian": Language(
  78. name="Bulgarian",
  79. iso_code="bg",
  80. use_ascii=False,
  81. charsets=["ISO-8859-5", "WINDOWS-1251", "IBM855"],
  82. alphabet="АБВГДЕЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЬЮЯабвгдежзийклмнопрстуфхцчшщъьюя",
  83. wiki_start_pages=["Начална_страница"],
  84. ),
  85. "Czech": Language(
  86. name="Czech",
  87. iso_code="cz",
  88. use_ascii=True,
  89. charsets=["ISO-8859-2", "WINDOWS-1250"],
  90. alphabet="áčďéěíňóřšťúůýžÁČĎÉĚÍŇÓŘŠŤÚŮÝŽ",
  91. wiki_start_pages=["Hlavní_strana"],
  92. ),
  93. "Danish": Language(
  94. name="Danish",
  95. iso_code="da",
  96. use_ascii=True,
  97. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  98. alphabet="æøåÆØÅ",
  99. wiki_start_pages=["Forside"],
  100. ),
  101. "German": Language(
  102. name="German",
  103. iso_code="de",
  104. use_ascii=True,
  105. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  106. alphabet="äöüßẞÄÖÜ",
  107. wiki_start_pages=["Wikipedia:Hauptseite"],
  108. ),
  109. "Greek": Language(
  110. name="Greek",
  111. iso_code="el",
  112. use_ascii=False,
  113. charsets=["ISO-8859-7", "WINDOWS-1253"],
  114. alphabet="αβγδεζηθικλμνξοπρσςτυφχψωάέήίόύώΑΒΓΔΕΖΗΘΙΚΛΜΝΞΟΠΡΣΣΤΥΦΧΨΩΆΈΉΊΌΎΏ",
  115. wiki_start_pages=["Πύλη:Κύρια"],
  116. ),
  117. "English": Language(
  118. name="English",
  119. iso_code="en",
  120. use_ascii=True,
  121. charsets=["ISO-8859-1", "WINDOWS-1252", "MacRoman"],
  122. wiki_start_pages=["Main_Page"],
  123. ),
  124. "Esperanto": Language(
  125. name="Esperanto",
  126. iso_code="eo",
  127. # Q, W, X, and Y not used at all
  128. use_ascii=False,
  129. charsets=["ISO-8859-3"],
  130. alphabet="abcĉdefgĝhĥijĵklmnoprsŝtuŭvzABCĈDEFGĜHĤIJĴKLMNOPRSŜTUŬVZ",
  131. wiki_start_pages=["Vikipedio:Ĉefpaĝo"],
  132. ),
  133. "Spanish": Language(
  134. name="Spanish",
  135. iso_code="es",
  136. use_ascii=True,
  137. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  138. alphabet="ñáéíóúüÑÁÉÍÓÚÜ",
  139. wiki_start_pages=["Wikipedia:Portada"],
  140. ),
  141. "Estonian": Language(
  142. name="Estonian",
  143. iso_code="et",
  144. use_ascii=False,
  145. charsets=["ISO-8859-4", "ISO-8859-13", "WINDOWS-1257"],
  146. # C, F, Š, Q, W, X, Y, Z, Ž are only for
  147. # loanwords
  148. alphabet="ABDEGHIJKLMNOPRSTUVÕÄÖÜabdeghijklmnoprstuvõäöü",
  149. wiki_start_pages=["Esileht"],
  150. ),
  151. "Finnish": Language(
  152. name="Finnish",
  153. iso_code="fi",
  154. use_ascii=True,
  155. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  156. alphabet="ÅÄÖŠŽåäöšž",
  157. wiki_start_pages=["Wikipedia:Etusivu"],
  158. ),
  159. "French": Language(
  160. name="French",
  161. iso_code="fr",
  162. use_ascii=True,
  163. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  164. alphabet="œàâçèéîïùûêŒÀÂÇÈÉÎÏÙÛÊ",
  165. wiki_start_pages=["Wikipédia:Accueil_principal", "Bœuf (animal)"],
  166. ),
  167. "Hebrew": Language(
  168. name="Hebrew",
  169. iso_code="he",
  170. use_ascii=False,
  171. charsets=["ISO-8859-8", "WINDOWS-1255"],
  172. alphabet="אבגדהוזחטיךכלםמןנסעףפץצקרשתװױײ",
  173. wiki_start_pages=["עמוד_ראשי"],
  174. ),
  175. "Croatian": Language(
  176. name="Croatian",
  177. iso_code="hr",
  178. # Q, W, X, Y are only used for foreign words.
  179. use_ascii=False,
  180. charsets=["ISO-8859-2", "WINDOWS-1250"],
  181. alphabet="abcčćdđefghijklmnoprsštuvzžABCČĆDĐEFGHIJKLMNOPRSŠTUVZŽ",
  182. wiki_start_pages=["Glavna_stranica"],
  183. ),
  184. "Hungarian": Language(
  185. name="Hungarian",
  186. iso_code="hu",
  187. # Q, W, X, Y are only used for foreign words.
  188. use_ascii=False,
  189. charsets=["ISO-8859-2", "WINDOWS-1250"],
  190. alphabet="abcdefghijklmnoprstuvzáéíóöőúüűABCDEFGHIJKLMNOPRSTUVZÁÉÍÓÖŐÚÜŰ",
  191. wiki_start_pages=["Kezdőlap"],
  192. ),
  193. "Italian": Language(
  194. name="Italian",
  195. iso_code="it",
  196. use_ascii=True,
  197. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  198. alphabet="ÀÈÉÌÒÓÙàèéìòóù",
  199. wiki_start_pages=["Pagina_principale"],
  200. ),
  201. "Lithuanian": Language(
  202. name="Lithuanian",
  203. iso_code="lt",
  204. use_ascii=False,
  205. charsets=["ISO-8859-13", "WINDOWS-1257", "ISO-8859-4"],
  206. # Q, W, and X not used at all
  207. alphabet="AĄBCČDEĘĖFGHIĮYJKLMNOPRSŠTUŲŪVZŽaąbcčdeęėfghiįyjklmnoprsštuųūvzž",
  208. wiki_start_pages=["Pagrindinis_puslapis"],
  209. ),
  210. "Latvian": Language(
  211. name="Latvian",
  212. iso_code="lv",
  213. use_ascii=False,
  214. charsets=["ISO-8859-13", "WINDOWS-1257", "ISO-8859-4"],
  215. # Q, W, X, Y are only for loanwords
  216. alphabet="AĀBCČDEĒFGĢHIĪJKĶLĻMNŅOPRSŠTUŪVZŽaābcčdeēfgģhiījkķlļmnņoprsštuūvzž",
  217. wiki_start_pages=["Sākumlapa"],
  218. ),
  219. "Macedonian": Language(
  220. name="Macedonian",
  221. iso_code="mk",
  222. use_ascii=False,
  223. charsets=["ISO-8859-5", "WINDOWS-1251", "MacCyrillic", "IBM855"],
  224. alphabet="АБВГДЃЕЖЗЅИЈКЛЉМНЊОПРСТЌУФХЦЧЏШабвгдѓежзѕијклљмнњопрстќуфхцчџш",
  225. wiki_start_pages=["Главна_страница"],
  226. ),
  227. "Dutch": Language(
  228. name="Dutch",
  229. iso_code="nl",
  230. use_ascii=True,
  231. charsets=["ISO-8859-1", "WINDOWS-1252", "MacRoman"],
  232. wiki_start_pages=["Hoofdpagina"],
  233. ),
  234. "Polish": Language(
  235. name="Polish",
  236. iso_code="pl",
  237. # Q and X are only used for foreign words.
  238. use_ascii=False,
  239. charsets=["ISO-8859-2", "WINDOWS-1250"],
  240. alphabet="AĄBCĆDEĘFGHIJKLŁMNŃOÓPRSŚTUWYZŹŻaąbcćdeęfghijklłmnńoóprsśtuwyzźż",
  241. wiki_start_pages=["Wikipedia:Strona_główna"],
  242. ),
  243. "Portuguese": Language(
  244. name="Portuguese",
  245. iso_code="pt",
  246. use_ascii=True,
  247. charsets=["ISO-8859-1", "ISO-8859-15", "WINDOWS-1252", "MacRoman"],
  248. alphabet="ÁÂÃÀÇÉÊÍÓÔÕÚáâãàçéêíóôõú",
  249. wiki_start_pages=["Wikipédia:Página_principal"],
  250. ),
  251. "Romanian": Language(
  252. name="Romanian",
  253. iso_code="ro",
  254. use_ascii=True,
  255. charsets=["ISO-8859-2", "WINDOWS-1250"],
  256. alphabet="ăâîșțĂÂÎȘȚ",
  257. wiki_start_pages=["Pagina_principală"],
  258. ),
  259. "Russian": Language(
  260. name="Russian",
  261. iso_code="ru",
  262. use_ascii=False,
  263. charsets=[
  264. "ISO-8859-5",
  265. "WINDOWS-1251",
  266. "KOI8-R",
  267. "MacCyrillic",
  268. "IBM866",
  269. "IBM855",
  270. ],
  271. alphabet="абвгдеёжзийклмнопрстуфхцчшщъыьэюяАБВГДЕЁЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯ",
  272. wiki_start_pages=["Заглавная_страница"],
  273. ),
  274. "Slovak": Language(
  275. name="Slovak",
  276. iso_code="sk",
  277. use_ascii=True,
  278. charsets=["ISO-8859-2", "WINDOWS-1250"],
  279. alphabet="áäčďéíĺľňóôŕšťúýžÁÄČĎÉÍĹĽŇÓÔŔŠŤÚÝŽ",
  280. wiki_start_pages=["Hlavná_stránka"],
  281. ),
  282. "Slovene": Language(
  283. name="Slovene",
  284. iso_code="sl",
  285. # Q, W, X, Y are only used for foreign words.
  286. use_ascii=False,
  287. charsets=["ISO-8859-2", "WINDOWS-1250"],
  288. alphabet="abcčdefghijklmnoprsštuvzžABCČDEFGHIJKLMNOPRSŠTUVZŽ",
  289. wiki_start_pages=["Glavna_stran"],
  290. ),
  291. # Serbian can be written in both Latin and Cyrillic, but there's no
  292. # simple way to get the Latin alphabet pages from Wikipedia through
  293. # the API, so for now we just support Cyrillic.
  294. "Serbian": Language(
  295. name="Serbian",
  296. iso_code="sr",
  297. alphabet="АБВГДЂЕЖЗИЈКЛЉМНЊОПРСТЋУФХЦЧЏШабвгдђежзијклљмнњопрстћуфхцчџш",
  298. charsets=["ISO-8859-5", "WINDOWS-1251", "MacCyrillic", "IBM855"],
  299. wiki_start_pages=["Главна_страна"],
  300. ),
  301. "Thai": Language(
  302. name="Thai",
  303. iso_code="th",
  304. use_ascii=False,
  305. charsets=["ISO-8859-11", "TIS-620", "CP874"],
  306. alphabet="กขฃคฅฆงจฉชซฌญฎฏฐฑฒณดตถทธนบปผฝพฟภมยรฤลฦวศษสหฬอฮฯะัาำิีึืฺุู฿เแโใไๅๆ็่้๊๋์ํ๎๏๐๑๒๓๔๕๖๗๘๙๚๛",
  307. wiki_start_pages=["หน้าหลัก"],
  308. ),
  309. "Turkish": Language(
  310. name="Turkish",
  311. iso_code="tr",
  312. # Q, W, and X are not used by Turkish
  313. use_ascii=False,
  314. charsets=["ISO-8859-3", "ISO-8859-9", "WINDOWS-1254"],
  315. alphabet="abcçdefgğhıijklmnoöprsştuüvyzâîûABCÇDEFGĞHIİJKLMNOÖPRSŞTUÜVYZÂÎÛ",
  316. wiki_start_pages=["Ana_Sayfa"],
  317. ),
  318. "Vietnamese": Language(
  319. name="Vietnamese",
  320. iso_code="vi",
  321. use_ascii=False,
  322. # Windows-1258 is the only common 8-bit
  323. # Vietnamese encoding supported by Python.
  324. # From Wikipedia:
  325. # For systems that lack support for Unicode,
  326. # dozens of 8-bit Vietnamese code pages are
  327. # available.[1] The most common are VISCII
  328. # (TCVN 5712:1993), VPS, and Windows-1258.[3]
  329. # Where ASCII is required, such as when
  330. # ensuring readability in plain text e-mail,
  331. # Vietnamese letters are often encoded
  332. # according to Vietnamese Quoted-Readable
  333. # (VIQR) or VSCII Mnemonic (VSCII-MNEM),[4]
  334. # though usage of either variable-width
  335. # scheme has declined dramatically following
  336. # the adoption of Unicode on the World Wide
  337. # Web.
  338. charsets=["WINDOWS-1258"],
  339. alphabet="aăâbcdđeêghiklmnoôơpqrstuưvxyAĂÂBCDĐEÊGHIKLMNOÔƠPQRSTUƯVXY",
  340. wiki_start_pages=["Chữ_Quốc_ngữ"],
  341. ),
  342. }