Blame view

pma/libraries/import/ldi.php 3.71 KB
a1684257   Administrator   first commit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
  <?php
  /* vim: set expandtab sw=4 ts=4 sts=4: */
  /**
   * CSV import plugin for phpMyAdmin using LOAD DATA
   *
   * @package PhpMyAdmin-Import
   * @subpackage LDI
   */
  if (! defined('PHPMYADMIN')) {
      exit;
  }
  
  /**
   *
   */
  if ($plugin_param !== 'table') {
      return;
  }
  
  if (isset($plugin_list)) {
      if ($GLOBALS['cfg']['Import']['ldi_local_option'] == 'auto') {
          $GLOBALS['cfg']['Import']['ldi_local_option'] = false;
  
          $result = PMA_DBI_try_query('SHOW VARIABLES LIKE \'local\\_infile\';');
          if ($result != false && PMA_DBI_num_rows($result) > 0) {
              $tmp = PMA_DBI_fetch_row($result);
              if ($tmp[1] == 'ON') {
                  $GLOBALS['cfg']['Import']['ldi_local_option'] = true;
              }
          }
          PMA_DBI_free_result($result);
          unset($result);
      }
      $plugin_list['ldi'] = array(
          'text' => __('CSV using LOAD DATA'),
          'extension' => 'ldi', // This is nonsense, however we want to default to our parser for csv
          'options' => array(
              array('type' => 'begin_group', 'name' => 'general_opts'),
              array('type' => 'bool', 'name' => 'replace', 'text' => __('Replace table data with file')),
              array('type' => 'bool', 'name' => 'ignore', 'text' => __('Do not abort on INSERT error')),
              array('type' => 'text', 'name' => 'terminated', 'text' => __('Columns terminated by'), 'size' => 2, 'len' => 2),
              array('type' => 'text', 'name' => 'enclosed', 'text' => __('Columns enclosed by'), 'size' => 2, 'len' => 2),
              array('type' => 'text', 'name' => 'escaped', 'text' => __('Columns escaped by'), 'size' => 2, 'len' => 2),
              array('type' => 'text', 'name' => 'new_line', 'text' => __('Lines terminated by'), 'size' => 2),
              array('type' => 'text', 'name' => 'columns', 'text' => __('Column names')),
              array('type' => 'bool', 'name' => 'local_option', 'text' => __('Use LOCAL keyword')),
              array('type' => 'end_group')
              ),
          'options_text' => __('Options'),
          );
      /* We do not define function when plugin is just queried for information above */
      return;
  }
  
  if ($import_file == 'none' || $compression != 'none' || $charset_conversion) {
      // We handle only some kind of data!
      $message = PMA_Message::error(__('This plugin does not support compressed imports!'));
      $error = true;
      return;
  }
  
  $sql = 'LOAD DATA';
  if (isset($ldi_local_option)) {
      $sql .= ' LOCAL';
  }
  $sql .= ' INFILE \'' . PMA_sqlAddSlashes($import_file) . '\'';
  if (isset($ldi_replace)) {
      $sql .= ' REPLACE';
  } elseif (isset($ldi_ignore)) {
      $sql .= ' IGNORE';
  }
  $sql .= ' INTO TABLE ' . PMA_backquote($table);
  
  if (strlen($ldi_terminated) > 0) {
      $sql .= ' FIELDS TERMINATED BY \'' . $ldi_terminated . '\'';
  }
  if (strlen($ldi_enclosed) > 0) {
      $sql .= ' ENCLOSED BY \'' . PMA_sqlAddSlashes($ldi_enclosed) . '\'';
  }
  if (strlen($ldi_escaped) > 0) {
      $sql .= ' ESCAPED BY \'' . PMA_sqlAddSlashes($ldi_escaped) . '\'';
  }
  if (strlen($ldi_new_line) > 0) {
      if ($ldi_new_line == 'auto') {
          $ldi_new_line = PMA_whichCrlf() == "\n" ? '\n' : '\r\n';
      }
      $sql .= ' LINES TERMINATED BY \'' . $ldi_new_line . '\'';
  }
  if ($skip_queries > 0) {
      $sql .= ' IGNORE ' . $skip_queries . ' LINES';
      $skip_queries = 0;
  }
  if (strlen($ldi_columns) > 0) {
      $sql .= ' (';
      $tmp   = preg_split('/,( ?)/', $ldi_columns);
      $cnt_tmp = count($tmp);
      for ($i = 0; $i < $cnt_tmp; $i++) {
          if ($i > 0) {
              $sql .= ', ';
          }
          /* Trim also `, if user already included backquoted fields */
          $sql     .= PMA_backquote(trim($tmp[$i], " \t\r\n\0\x0B`"));
      } // end for
      $sql .= ')';
  }
  
  PMA_importRunQuery($sql, $sql);
  PMA_importRunQuery();
  $finished = true;
  ?>