@@ 40-70 (lines=31) @@ | ||
37 | /** Load WordPress Bootstrap */ |
|
38 | require_once(dirname(dirname(__FILE__)).'/wp-load.php'); |
|
39 | ||
40 | if ( ! function_exists('maybe_create_table') ) : |
|
41 | /** |
|
42 | * Create database table, if it doesn't already exist. |
|
43 | * |
|
44 | * @since 1.0.0 |
|
45 | * |
|
46 | * @global wpdb $wpdb WordPress database abstraction object. |
|
47 | * |
|
48 | * @param string $table_name Database table name. |
|
49 | * @param string $create_ddl Create database table SQL. |
|
50 | * @return bool False on error, true if already exists or success. |
|
51 | */ |
|
52 | function maybe_create_table($table_name, $create_ddl) { |
|
53 | global $wpdb; |
|
54 | foreach ($wpdb->get_col("SHOW TABLES",0) as $table ) { |
|
55 | if ($table == $table_name) { |
|
56 | return true; |
|
57 | } |
|
58 | } |
|
59 | // Didn't find it, so try to create it. |
|
60 | $wpdb->query($create_ddl); |
|
61 | ||
62 | // We cannot directly tell that whether this succeeded! |
|
63 | foreach ($wpdb->get_col("SHOW TABLES",0) as $table ) { |
|
64 | if ($table == $table_name) { |
|
65 | return true; |
|
66 | } |
|
67 | } |
|
68 | return false; |
|
69 | } |
|
70 | endif; |
|
71 | ||
72 | if ( ! function_exists('maybe_add_column') ) : |
|
73 | /** |
|
@@ 72-105 (lines=34) @@ | ||
69 | } |
|
70 | endif; |
|
71 | ||
72 | if ( ! function_exists('maybe_add_column') ) : |
|
73 | /** |
|
74 | * Add column to database table, if column doesn't already exist in table. |
|
75 | * |
|
76 | * @since 1.0.0 |
|
77 | * |
|
78 | * @global wpdb $wpdb WordPress database abstraction object. |
|
79 | * |
|
80 | * @param string $table_name Database table name |
|
81 | * @param string $column_name Table column name |
|
82 | * @param string $create_ddl SQL to add column to table. |
|
83 | * @return bool False on failure. True, if already exists or was successful. |
|
84 | */ |
|
85 | function maybe_add_column($table_name, $column_name, $create_ddl) { |
|
86 | global $wpdb; |
|
87 | foreach ($wpdb->get_col("DESC $table_name",0) as $column ) { |
|
88 | ||
89 | if ($column == $column_name) { |
|
90 | return true; |
|
91 | } |
|
92 | } |
|
93 | ||
94 | // Didn't find it, so try to create it. |
|
95 | $wpdb->query($create_ddl); |
|
96 | ||
97 | // We cannot directly tell that whether this succeeded! |
|
98 | foreach ($wpdb->get_col("DESC $table_name",0) as $column ) { |
|
99 | if ($column == $column_name) { |
|
100 | return true; |
|
101 | } |
|
102 | } |
|
103 | return false; |
|
104 | } |
|
105 | endif; |
|
106 | ||
107 | /** |
|
108 | * Drop column from database table, if it exists. |