[Sommaire] [Imprimer]

   1  <?php
   2  require_once('../wp-config.php');
   3  $debug = 0;
   4  
   5  /**
   6   ** maybe_create_table()
   7   ** Create db table if it doesn't exist.
   8   ** Returns:  true if already exists or on successful completion
   9   **           false on error
  10   */
  11  function maybe_create_table($table_name, $create_ddl) {
  12      global $wpdb;
  13      foreach ($wpdb->get_col("SHOW TABLES",0) as $table ) {
  14          if ($table == $table_name) {
  15              return true;
  16          }
  17      }
  18      //didn't find it try to create it.
  19      $q = $wpdb->query($create_ddl);
  20      // we cannot directly tell that whether this succeeded!
  21      foreach ($wpdb->get_col("SHOW TABLES",0) as $table ) {
  22          if ($table == $table_name) {
  23              return true;
  24          }
  25      }
  26      return false;
  27  }
  28  
  29  /**
  30   ** maybe_add_column()
  31   ** Add column to db table if it doesn't exist.
  32   ** Returns:  true if already exists or on successful completion
  33   **           false on error
  34   */
  35  function maybe_add_column($table_name, $column_name, $create_ddl) {
  36      global $wpdb, $debug;
  37      foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  38          if ($debug) echo("checking $column == $column_name<br />");
  39              if ($column == $column_name) {
  40                  return true;
  41              }
  42      }
  43      //didn't find it try to create it.
  44      $q = $wpdb->query($create_ddl);
  45      // we cannot directly tell that whether this succeeded!
  46      foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  47          if ($column == $column_name) {
  48              return true;
  49          }
  50      }
  51      return false;
  52  }
  53  
  54  
  55  /**
  56   ** maybe_drop_column()
  57   ** Drop column from db table if it exists.
  58   ** Returns:  true if it doesn't already exist or on successful drop
  59   **           false on error
  60   */
  61  function maybe_drop_column($table_name, $column_name, $drop_ddl) {
  62      global $wpdb;
  63      foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  64          if ($column == $column_name) {
  65              //found it try to drop it.
  66              $q = $wpdb->query($drop_ddl);
  67              // we cannot directly tell that whether this succeeded!
  68              foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  69                  if ($column == $column_name) {
  70                      return false;
  71                  }
  72              }
  73          }
  74      }
  75      // else didn't find it
  76      return true;
  77  }
  78  
  79  
  80  /**
  81   ** check_column()
  82   ** Check column matches passed in criteria.
  83   ** Pass in null to skip checking that criteria
  84   ** Returns:  true if it matches
  85   **           false otherwise
  86   ** (case sensitive) Column names returned from DESC table are:
  87   **      Field
  88   **      Type
  89   **      Null
  90   **      Key
  91   **      Default
  92   **      Extra
  93   */
  94  function check_column($table_name, $col_name, $col_type, $is_null = null, $key = null, $default = null, $extra = null) {
  95      global $wpdb, $debug;
  96      $diffs = 0;
  97      $results = $wpdb->get_results("DESC $table_name");
  98  
  99      foreach ($results as $row ) {
 100          if ($debug > 1) print_r($row);
 101              if ($row->Field == $col_name) {
 102                  // got our column, check the params
 103                  if ($debug) echo ("checking $row->Type against $col_type\n");
 104                  if (($col_type != null) && ($row->Type != $col_type)) {
 105                      ++$diffs;
 106                  }
 107                  if (($is_null != null) && ($row->Null != $is_null)) {
 108                      ++$diffs;
 109                  }
 110                  if (($key != null) && ($row->Key  != $key)) {
 111                      ++$diffs;
 112                  }
 113                  if (($default != null) && ($row->Default != $default)) {
 114                      ++$diffs;
 115                  }
 116                  if (($extra != null) && ($row->Extra != $extra)) {
 117                      ++$diffs;
 118                  }
 119                  if ($diffs > 0) {
 120                      if ($debug) echo ("diffs = $diffs returning false\n");
 121                      return false;
 122                  }
 123                  return true;
 124              } // end if found our column
 125      }
 126      return false;
 127  }
 128  
 129  /*
 130  echo "<p>testing</p>";
 131  echo "<pre>";
 132  
 133  //check_column('wp_links', 'link_description', 'mediumtext');
 134  //if (check_column($wpdb->comments, 'comment_author', 'tinytext'))
 135  //    echo "ok\n";
 136  $error_count = 0;
 137  $tablename = $wpdb->links;
 138  // check the column
 139  if (!check_column($wpdb->links, 'link_description', 'varchar(255)'))
 140  {
 141      $ddl = "ALTER TABLE $wpdb->links MODIFY COLUMN link_description varchar(255) NOT NULL DEFAULT '' ";
 142      $q = $wpdb->query($ddl);
 143  }
 144  if (check_column($wpdb->links, 'link_description', 'varchar(255)')) {
 145      $res .= $tablename . ' - ok <br />';
 146  } else {
 147      $res .= 'There was a problem with ' . $tablename . '<br />';
 148      ++$error_count;
 149  }
 150  echo "</pre>";
 151  */
 152  ?>