sql.php 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206
  1. <?php
  2. require_once __DIR__."/../config.php";
  3. /*
  4. * 该脚本用于生成 SQL 语句, 将三藏语料 CSV 数据 (如:abh01a.att.csv)
  5. * 转换为 SQL 语句插入到 PostgreSQL 内,数据表结构参见 fts.sql
  6. * 由于懒惰,没有优化脚本,占用了较多内存,所以执行时请多给 PHP 一些内存:
  7. * php -d memory_limit=1024M sql.php
  8. *
  9. */
  10. set_exception_handler(function($e){
  11. fwrite(STDERR,"error-msg:".$e->getMessage().PHP_EOL);
  12. fwrite(STDERR,"error-file:".$e->getFile().PHP_EOL);
  13. fwrite(STDERR,"error-line:".$e->getLine().PHP_EOL);
  14. exit;
  15. });
  16. function is_pali_word ($str) {
  17. $pali_word_exp = "/^[āīūṅñṭḍṇḷṃṁŋĀĪŪṄÑṬḌṆḶṂṀŊabcdefghijklmnoprstuvyABCDEFGHIJKLMNOPRSTUVY-]+$/";
  18. return preg_match($pali_word_exp, $str) === 1;
  19. }
  20. /*
  21. *
  22. * 通过黑体字数组来计算黑体字连续出现的次数
  23. * 参数样例: ['a', '', '', 'b', 'c', 'd', '','','e','f', '', 'g','h']
  24. * 函数返回值样例:
  25. *
  26. * Array
  27. * (
  28. * [bold_single] => a
  29. * [bold_double] => e f , g h
  30. * [bold_multiple] => b c d
  31. * )
  32. *
  33. * */
  34. function count_bld ($bld_array) {
  35. $prev = '';
  36. $bag = [];
  37. $result = [];
  38. // 添加最后一个空白结束占位符
  39. array_push($bld_array, '');
  40. foreach($bld_array as $v) {
  41. if (empty($v)) {
  42. $prev = $v;
  43. if (!empty($bag)) {
  44. array_push($result, $bag);
  45. $bag = [];
  46. }
  47. continue;
  48. } else {
  49. array_push($bag, $v);
  50. }
  51. }
  52. $final_result = [];
  53. foreach($result as $v) {
  54. $cnt = count($v);
  55. $content = join(' ', $v);
  56. if ($cnt == 1) {
  57. $key = 'bold_single';
  58. } else if ($cnt == 2) {
  59. $key = 'bold_double';
  60. } else if ($cnt > 2) {
  61. $key = 'bold_multiple';
  62. }
  63. if (empty($final_result[$key])) {
  64. $final_result[$key] = $content;
  65. } else {
  66. $final_result[$key] .= (' , ' . $content);
  67. }
  68. }
  69. return $final_result;
  70. }
  71. $dns = _DB_ENGIN_.":host="._DB_HOST_.";port="._DB_PORT_.";dbname="._DB_NAME_.";user="._DB_USERNAME_.";password="._DB_PASSWORD_.";";
  72. $dbh_fts = new PDO($dns, _DB_USERNAME_, _DB_PASSWORD_, array(PDO::ATTR_PERSISTENT => true));
  73. $dbh_fts->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
  74. // 查找 tmp/palicsv/ 目录下的语料数据
  75. $palicsv_path = __DIR__.'/../../tmp/palicsv';
  76. $scan = scandir($palicsv_path);
  77. $fileCounter = 0;
  78. foreach($scan as $foldername) {
  79. if (is_dir("$palicsv_path/$foldername")) {
  80. $csv_file = "$palicsv_path/$foldername/$foldername.csv";
  81. // DEBUG
  82. // if ($foldername != 'abh01m.mul') continue;
  83. if (is_file($csv_file)) {
  84. $fileCounter++;
  85. fwrite(STDOUT,"runing : $fileCounter" . PHP_EOL . $csv_file . PHP_EOL);
  86. // 初始化段落为 0 (没有这种段落)
  87. $paragraph = 0;
  88. // 初始化当前段落的黑体字数组
  89. $bold_text = [];
  90. if (($handle = fopen($csv_file, "r")) !== FALSE) {
  91. # 获取book id
  92. $data = fgetcsv($handle, 1000, ",");
  93. $data = fgetcsv($handle, 1000, ",");
  94. $bookId = (int)mb_substr($data[2],1);
  95. #删除旧数据
  96. $query = "DELETE FROM "._TABLE_FTS_." WHERE book=?";
  97. $stmt = $dbh_fts->prepare($query);
  98. $stmt->execute(array($bookId));
  99. // 开始一个事务,关闭自动提交
  100. $dbh_fts->beginTransaction();
  101. $query = "INSERT INTO "._TABLE_FTS_." (book , paragraph , wid,bold_single,bold_double,bold_multiple,content) VALUES ( ? , ? , ? , ? , ? , ? , ? )";
  102. $stmt = $dbh_fts->prepare($query);
  103. rewind($handle);
  104. $row=0;
  105. while (($data = fgetcsv($handle, 1000, ",")) !== FALSE) {
  106. #忽略第一行
  107. if($row > 0){
  108. $current_word = $data[5];
  109. $style = $data[15];
  110. if ($style == 'paranum') {
  111. // 如果是段落编号,则保留数字
  112. $current_word = $data[4];
  113. } else if (!is_pali_word($current_word)) {
  114. /*
  115. * 如果当前单词不是巴利语单词,则忽略,当作它不存在
  116. * TODO 这样的处理方式,可能不合适,如下面场景:
  117. * bld1 - bld2
  118. * bld1 和 bld2 是否应该分开对待呢?
  119. */
  120. continue;
  121. }
  122. if ($paragraph == $data[3]) {
  123. // 如果是同一段落,那么合并段落中的内容,中间加入空格
  124. $content .= ' ' . $current_word;
  125. // wid 取最后一个不为空的值 TODO (不一定合适)
  126. $wid = empty($data[1]) ? $wid : $data[1];
  127. array_push($bold_text, $style == 'bld' ? $current_word : '');
  128. } else {
  129. // 如果是不同段落
  130. if ($paragraph !== 0) {
  131. // 如果刚才已经记录有数据,则转换为 SQL
  132. $bold_result = count_bld($bold_text);
  133. if(isset($bold_result['bold_single'])){
  134. $bold_single = $bold_result['bold_single'];
  135. }else{
  136. $bold_single = "";
  137. }
  138. if(isset($bold_result['bold_double'])){
  139. $bold_double = $bold_result['bold_double'];
  140. }else{
  141. $bold_double = "";
  142. }
  143. if(isset($bold_result['bold_multiple'])){
  144. $bold_multiple = $bold_result['bold_multiple'];
  145. }else{
  146. $bold_multiple = "";
  147. }
  148. $stmt->execute(array($book, $paragraph, $wid,$bold_single,$bold_double,$bold_multiple,$content));
  149. // 转换后,重置黑体字数据
  150. $bold_text = [];
  151. }
  152. // 如果是不同段落,则赋新的值
  153. $content = $current_word;
  154. $paragraph = (int)$data[3];
  155. $book = (int)mb_substr($data[2],1);
  156. $wid = $data[1];
  157. array_push($bold_text, $style == 'bld' ? $current_word : '');
  158. }
  159. }
  160. $row++;
  161. }
  162. fclose($handle);
  163. // 提交更改
  164. $dbh_fts->commit();
  165. if (!$stmt || ($stmt && $stmt->errorCode() != 0)) {
  166. $error = $dbh_fts->errorInfo();
  167. fwrite(STDERR, "error - $error[2]".PHP_EOL);
  168. } else {
  169. fwrite(STDOUT, "updata $row recorders.".PHP_EOL);
  170. }
  171. }
  172. // DEBUG 仅生成一个文件,测试用
  173. // exit;
  174. }
  175. }
  176. }
  177. fwrite(STDOUT, "Done. Amitābha".PHP_EOL);
  178. ?>