Added some preliminary code for convert CSV to Ledger

1 files changed, 136 insertions, 0 deletions

diff --git a/src/convert.cc b/src/convert.cc
new file mode 100644
index 00000000..e47fb6c0
--- /dev/null
+++ b/src/convert.cc
@@ -0,0 +1,136 @@
+/*
+ * Copyright (c) 2003-2010, John Wiegley.  All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ *   notice, this list of conditions and the following disclaimer.
+ *
+ * - Redistributions in binary form must reproduce the above copyright
+ *   notice, this list of conditions and the following disclaimer in the
+ *   documentation and/or other materials provided with the distribution.
+ *
+ * - Neither the name of New Artisans LLC nor the names of its
+ *   contributors may be used to endorse or promote products derived from
+ *   this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <system.hh>
+
+#include "convert.h"
+#include "csv.h"
+#include "scope.h"
+#include "interactive.h"
+#include "iterators.h"
+#include "report.h"
+#include "xact.h"
+#include "output.h"
+
+namespace ledger {
+
+value_t convert_command(call_scope_t& scope)
+{
+  interactive_t args(scope, "s");
+  report_t&	report(find_scope<report_t>(scope));
+  journal_t&	journal(*report.session.journal.get());
+
+  string bucket_name;
+  if (report.HANDLED(account_))
+    bucket_name = report.HANDLER(account_).str();
+  else
+    bucket_name = "Equity:Unknown";
+
+  account_t * bucket = journal.master->find_account(bucket_name);
+
+  // Make an amounts mapping for the account under consideration
+
+  typedef std::map<value_t, std::list<post_t *> > post_map_t;
+  post_map_t post_map;
+
+  xacts_iterator journal_iter(journal);
+  while (xact_t * xact = journal_iter()) {
+    post_t * post = NULL;
+    xact_posts_iterator xact_iter(*xact);
+    while ((post = xact_iter()) != NULL) {
+      if (post->account == bucket)
+	break;
+    }
+    if (post) {
+      post_map_t::iterator i = post_map.find(post->amount);
+      if (i == post_map.end()) {
+	std::list<post_t *> post_list;
+	post_list.push_back(post);
+	post_map.insert(post_map_t::value_type(post->amount, post_list));
+      } else {
+	(*i).second.push_back(post);
+      }
+    }
+  }
+
+  // Read in the series of transactions from the CSV file
+
+  format_posts formatter(report,
+			 report.report_format(report.HANDLER(print_format_)),
+			 false);
+
+  ifstream   data(path(args.get<string>(0)));
+  csv_reader reader(data);
+  while (xact_t * xact = reader.read_xact(journal, bucket)) {
+    bool matched = false;
+    post_map_t::iterator i = post_map.find(- xact->posts.front()->amount);
+    if (i != post_map.end()) {
+      std::list<post_t *>& post_list((*i).second);
+      foreach (post_t * post, post_list) {
+	if (xact->code && post->xact->code &&
+	    *xact->code == *post->xact->code) {
+	  matched = true;
+	  break;
+	}
+	else if (xact->actual_date() == post->actual_date()) {
+	  matched = true;
+	  break;
+	}
+      }
+    }
+
+    if (matched) {
+      DEBUG("convert.csv", "Ignored xact with code: " << *xact->code);
+      delete xact;		// ignore it
+    }
+    else if (! journal.add_xact(xact)) {
+      delete xact;
+      throw_(std::runtime_error,
+	     _("Failed to finalize derived transaction (check commodities)"));
+    }
+    else {
+      xact_posts_iterator xact_iter(*xact);
+      while (post_t * post = xact_iter())
+	formatter(*post);
+    }
+  }
+
+  // If not, transform the payee according to regexps
+
+  // Set the account to a default vaule, then transform the account according
+  // to the payee
+
+  // Print out the final form of the transaction
+
+  return true;
+}
+
+} // namespace ledger