From ec49e68ee316d65d5bae8c40d60093725fc1b0ca Mon Sep 17 00:00:00 2001
From: Matthieu Dietrich <matthieu.dietrich@camptocamp.com>
Date: Wed, 17 Dec 2014 11:05:20 +0100
Subject: [PATCH] Add account filter for CSV export

---
 .../wizard/account_export_csv.py              | 23 +++++++++++++++++--
 .../wizard/account_export_csv_view.xml        |  1 +
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/account_export_csv/wizard/account_export_csv.py b/account_export_csv/wizard/account_export_csv.py
index 197d1f6a..eb031fb0 100644
--- a/account_export_csv/wizard/account_export_csv.py
+++ b/account_export_csv/wizard/account_export_csv.py
@@ -93,6 +93,13 @@ class AccountCSVExport(orm.TransientModel):
             'journal_id',
             'Journals',
             help='If empty, use all journals, only used for journal entries'),
+        'account_ids': fields.many2many(
+            'account.account',
+            'rel_wizard_account',
+            'wizard_id',
+            'account_id',
+            'Accounts',
+            help='If empty, use all accounts, only used for journal entries'),
         'export_filename': fields.char('Export CSV Filename', size=128),
     }
 
@@ -146,6 +153,7 @@ class AccountCSVExport(orm.TransientModel):
                           fiscalyear_id,
                           period_range_ids,
                           journal_ids,
+                          account_ids,
                           context=None):
         """
         Return list to generate rows of the CSV file
@@ -208,6 +216,7 @@ class AccountCSVExport(orm.TransientModel):
                            fiscalyear_id,
                            period_range_ids,
                            journal_ids,
+                           account_ids,
                            context=None):
         """
         Return list to generate rows of the CSV file
@@ -314,6 +323,7 @@ class AccountCSVExport(orm.TransientModel):
                                   fiscalyear_id,
                                   period_range_ids,
                                   journal_ids,
+                                  account_ids,
                                   context=None):
         """
         Create a generator of rows of the CSV file
@@ -372,10 +382,12 @@ class AccountCSVExport(orm.TransientModel):
             (account_bank_statement.id=account_move_line.statement_id)
         WHERE account_period.id IN %(period_ids)s
         AND account_journal.id IN %(journal_ids)s
+        AND account_account.id IN %(account_ids)s
         ORDER BY account_move_line.date
         """,
-                   {'period_ids': tuple(
-                       period_range_ids), 'journal_ids': tuple(journal_ids)}
+                   {'period_ids': tuple(period_range_ids),
+                    'journal_ids': tuple(journal_ids),
+                    'account_ids': tuple(account_ids)}
                    )
         while 1:
             # http://initd.org/psycopg/docs/cursor.html#cursor.fetchmany
@@ -407,12 +419,19 @@ class AccountCSVExport(orm.TransientModel):
         else:
             j_obj = self.pool.get("account.journal")
             journal_ids = j_obj.search(cr, uid, [], context=context)
+        account_ids = None
+        if form.account_ids:
+            account_ids = [x.id for x in form.account_ids]
+        else:
+            aa_obj = self.pool.get("account.account")
+            account_ids = aa_obj.search(cr, uid, [], context=context)
         rows = itertools.chain((get_header_func(cr, uid, ids,
                                                 context=context),),
                                get_rows_func(cr, uid, ids,
                                              fiscalyear_id,
                                              period_range_ids,
                                              journal_ids,
+                                             account_ids,
                                              context=context)
                                )
         return rows
diff --git a/account_export_csv/wizard/account_export_csv_view.xml b/account_export_csv/wizard/account_export_csv_view.xml
index f447138a..98faa9b7 100644
--- a/account_export_csv/wizard/account_export_csv_view.xml
+++ b/account_export_csv/wizard/account_export_csv_view.xml
@@ -15,6 +15,7 @@
                     <group colspan="4" col="2">
                         <field name="periods" domain="[('fiscalyear_id','=',fiscalyear_id)]"/>
                         <field name="journal_ids"/>
+                        <field name="account_ids"/>
                     </group>
                    <separator string ="Report" colspan="4"/>
                     <group colspan="4">