X-Git-Url: http://git.polytechnique.org/?a=blobdiff_plain;f=bin%2Fcsv2sql.php;h=4dcea437a74c9b0210d1cc13cf190fb342f99c7d;hb=c76545c351fae4e2298624ff9ee5bf854dc5a5b6;hp=06b1dbbf178630b54eb9a8bdd615b764322d273c;hpb=5ddeb07cc787dd1dc3630a31f1528f5cc7c4d9b9;p=platal.git

diff --git a/bin/csv2sql.php b/bin/csv2sql.php
index 06b1dbb..4dcea43 100755
--- a/bin/csv2sql.php
+++ b/bin/csv2sql.php
@@ -1,7 +1,7 @@
 #!/usr/bin/php5
 <?php
 /***************************************************************************
- *  Copyright (C) 2003-2007 Polytechnique.org                              *
+ *  Copyright (C) 2003-2010 Polytechnique.org                              *
  *  http://opensource.polytechnique.org/                                   *
  *                                                                         *
  *  This program is free software; you can redistribute it and/or modify   *
@@ -20,15 +20,22 @@
  *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA                *
  ***************************************************************************/
 
+require_once dirname(__FILE__) . '/../core/classes/csvimporter.php';
+
 // {{{ function showHelp()
 
-function showHelp($error = null) {
+function showHelp($error = null)
+{
     if ($error) {
         echo 'Ligne de commande non-valide : ' . $error, "\n\n";
     }
-    echo 'csv2sql.php -t table [-i source] [-r phpfile]', "\n\n";
+    echo 'csv2sql.php -t table [-du] [-k keys] [-i source] [-r phpfile]', "\n\n";
     echo 'options:', "\n";
     echo ' -t table: table in which insertion is to be done', "\n";
+    echo ' -d: switch to debug mode', "\n";
+    echo ' -u: switch to update mode', "\n";
+    echo ' -o: switch to update-only mode', "\n";
+    echo ' -k keys: comma-separated list of keys', "\n";
     echo ' -i source: CSV source file (stdin if not defined or if source is \'-\'', "\n";
     echo ' -r phpfile: PHP file which define relations', "\n";
 }
@@ -38,20 +45,34 @@ function showHelp($error = null) {
 
 function processArgs()
 {
-    global $sourceName, $table, $includedFile;
-    $opts = getopt('i:t:r:d:');
+    global $sourceName, $table, $includedFile, $debug, $action, $keys;
+    $opts = getopt('oui:t:r:dk:');
     if ($opts['i'] == '-' || empty($opts['i'])) {
         $sourceName = 'php://stdin';
     } else {
         $sourceName = $opts['i'];
     }
 
+    if ($opts['k'] && !empty($opts['k'])) {
+        $keys = $opts['k'];
+    }
+
+    if (isset($opts['u'])) {
+        $action = CSV_UPDATE ;
+    } elseif (isset($opts['o'])) {
+        $action = CSV_UPDATEONLY;
+    }
+
+    if (isset($opts['d'])) {
+        $debug = true;
+    }
+
     if ($opts['r'] && !empty($opts['r'])) {
         $includedFile = $opts['r'];
     }
 
     if (!$opts['t'] || empty($opts['t'])) {
-        showHelp('Table non définie');
+        showHelp('Table non définie');
         exit;
     }
     $table = $opts['t'];
@@ -59,21 +80,24 @@ function processArgs()
 
 // }}}
 
+global $debug, $action, $keys;
+$debug           = false;
+$action          = CSV_INSERT;
+$keys            = 'id';
+
 processArgs();
-require_once(dirname(__FILE__) . '/../classes/csvimporter.php');
-require_once(dirname(__FILE__) . '/../classes/xdb.php');
+require_once dirname(__FILE__) . '/../core/classes/xdb.php';
 
-$source    = file_get_contents($sourceName);
+$source          = file_get_contents($sourceName);
 $insert_relation = null;
 $update_relation = null;
-$debug           = false;
-$action          = CSV_INSERT;
 if (isset($includedFile)) {
-    require_once($includedFile);
+    require_once $includedFile;
 }
 
-$translater = new CSVImporter($table, $key, !$debug);
+$translater = new CSVImporter($table, $keys, !$debug);
 $translater->setCSV($source);
 $translater->run($action, $insert_relation, $update_relation);
 
+// vim:set et sw=4 sts=4 sws=4 foldmethod=marker enc=utf-8:
 ?>