wizard/account_export_csv.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    etudesetchantiersidf module for OpenERP, Custom module for Étude et
   5 #    Chantiers île-de-France
   6 #    Copyright (C) 2014-2022 etudesetchantiersidf
   7 #    (<http://etudesetchantiersiledefrance.unarec.org/>)
   8 #
   9 #    This file is a part of etudesetchantiersidf
  10 #
  11 #    etudesetchantiersidf is free software: you can redistribute it and/or
  12 #    modify it under the terms of the GNU General Public License as published
  13 #    by the Free Software Foundation, either version 3 of the License, or (at
  14 #    your option) any later version.
  15 #
  16 #    etudesetchantiersidf is distributed in the hope that it will be useful,
  17 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 #    GNU General Public License for more details.
  20 #
  21 #    You should have received a copy of the GNU General Public License
  22 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  23 #
  24 ##############################################################################
  25
  26 import StringIO
  27 import cStringIO
  28 import base64
  29 import csv
  30 import codecs
  31 import calendar
  32 import datetime
  33 import string
  34
  35 import openerp.exceptions
  36 from openerp.osv import osv
  37 from openerp.tools.translate import _
  38 from openerp.addons.account_export_csv.wizard.account_export_csv import AccountUnicodeWriter
  39
  40
  41 class AccountUnicodeWriter(AccountUnicodeWriter):
  42
  43     def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
  44         # Redirect output to a queue
  45         self.queue = cStringIO.StringIO()
  46         # created a writer with Excel formating settings
  47
  48         self.writer = csv.writer(self.queue, dialect=dialect,
  49                                  quoting=csv.QUOTE_NONE, **kwds)
  50         self.stream = f
  51         self.encoder = codecs.getincrementalencoder(encoding)()
  52
  53
  54 class AccountCSVExport(osv.osv_memory):
  55     _inherit = "account.csv.export"
  56
  57     def action_manual_export_sage(self, cr, uid, ids, context=None):
  58         this = self.browse(cr, uid, ids)[0]
  59         rows = self.get_data(cr, uid, ids, "sage", context)
  60         file_data = StringIO.StringIO()
  61         try:
  62             writer = AccountUnicodeWriter(file_data)
  63             writer.writerows(rows)
  64             file_value = file_data.getvalue()
  65             self.write(cr, uid, ids,
  66                        {'data': base64.encodestring(file_value)},
  67                        context=context)
  68         finally:
  69             file_data.close()
  70         return {
  71             'type': 'ir.actions.act_window',
  72             'res_model': 'account.csv.export',
  73             'view_mode': 'form',
  74             'view_type': 'form',
  75             'res_id': this.id,
  76             'views': [(False, 'form')],
  77             'target': 'new',
  78         }
  79
  80     def _get_header_sage(self, cr, uid, ids, context=None):
  81         """Return header for SAGE export"""
  82         return [_(u'Code du journal'),
  83                 _(u'N° de pièce'),
  84                 _(u'Date'),
  85                 _(u'Compte Général'),
  86                 _(u'Compte Tiers'),
  87                 _(u'Général/analytique'),
  88                 _(u'Compte Analytique'),
  89                 _(u'Nom du client'),
  90                 _(u'Débit'),
  91                 _(u'Crédit'),
  92                 ]
  93
  94     def format_row_sage(self, dico, numero_piece, side):
  95         """Return a formatted row for SAGE export"""
  96         if side == "D":
  97             debit = dico['debit']
  98             credit = 0
  99         if side == "C":
 100             debit = 0
 101             credit = dico['credit']
 102         return [
 103             dico['journal_code'],
 104             numero_piece,
 105             dico['date'],
 106             dico['account_code'],
 107             dico['compte_tiers'],
 108             dico['general_analytic'],
 109             dico['analytic_account'],
 110             dico['client_name'],
 111             debit,
 112             credit,
 113             ]
 114
 115     def _get_rows_sage(self, cr, uid, ids,
 116                        fiscalyear_id,
 117                        period_range_ids,
 118                        journal_ids,
 119                        account_ids=None,
 120                        context=None):
 121         fiscalyear_obj = self.pool.get('account.fiscalyear')
 122         fiscalyear_code = fiscalyear_obj.browse(cr, uid, fiscalyear_id, context=context)
 123         fiscalyear_code = fiscalyear_obj.browse(cr, uid, fiscalyear_id, context=context).code
 124
 125         period_obj = self.pool.get('account.period')
 126         journal_obj = self.pool.get('account.journal')
 127         account_obj = self.pool.get('account.account')
 128         numero_piece = 0
 129         prev_period = 0
 130         prev_journal = 0
 131         cr.execute("""
 132             select
 133                 aml.period_id,
 134                 aml.journal_id as journal_id,
 135                 aml.account_id as account_id,
 136                 sum(aml.debit) as sum_debit,
 137                 sum(aml.credit) as sum_credit
 138             from
 139                 account_move_line as aml
 140             where
 141                 aml.period_id in %(period_ids)s and
 142                 aml.journal_id in %(journal_ids)s and
 143                 aml.account_id in %(account_ids)s
 144             group by
 145                 period_id,
 146                 journal_id,
 147                 account_id
 148             order by
 149                 period_id,
 150                 journal_id,
 151                 account_id
 152            """,
 153                    {
 154                        'period_ids': tuple(period_range_ids),
 155                        'journal_ids': tuple(journal_ids),
 156                        'account_ids': tuple(account_ids),
 157                        }
 158         )
 159         res = cr.fetchall()
 160         rows = []
 161         for line in res:
 162             dico = {}
 163             journal = journal_obj.browse(cr, uid, line[1], context=context)
 164             dico['journal_code'] = journal.code[0:2]
 165             period = period_obj.browse(cr, uid, line[0], context=context)
 166             if prev_journal != journal or prev_period != period:
 167                 numero_piece += 1
 168             year = int(period.code[3:7])
 169             month = int(period.code[0:2])
 170             day = calendar.monthrange(year, month)[1]
 171             dico['date'] = datetime.datetime(year, month, day).strftime("%d/%m/%Y")
 172             account = account_obj.browse(cr, uid, line[2], context=context)
 173             dico['account_code'] = account.code
 174             preformatted_client_name = journal.name.split(" - ", 1)[1].upper().replace(' ','')
 175             for char in string.punctuation:
 176                 preformatted_client_name = preformatted_client_name.replace(char,'')
 177             dico['client_name'] = preformatted_client_name
 178             if account.type in ("receivable", "payable"):
 179                 dico['compte_tiers'] = dico['client_name']
 180             else:
 181                 dico['compte_tiers'] = ""
 182             dico['general_analytic'] = "G"
 183             dico['analytic_account'] = ""
 184             dico['debit'] = line[3]
 185             dico['credit'] = line[4]
 186             if dico['debit'] == dico['credit']:
 187                 pass
 188             else:
 189                 if dico['debit']:
 190                     row = self.format_row_sage(dico, numero_piece, "D")
 191                     rows.append(row)
 192                 if dico['credit']:
 193                     row = self.format_row_sage(dico, numero_piece, "C")
 194                     rows.append(row)
 195                 if account.type == "other":
 196                     dico['general_analytic'] = "A"
 197                     dico['analytic_account'] = journal.code[3:]
 198                     if dico['debit']:
 199                         row = self.format_row_sage(dico, numero_piece, "D")
 200                         rows.append(row)
 201                     if dico['credit']:
 202                         row = self.format_row_sage(dico, numero_piece, "C")
 203                         rows.append(row)
 204
 205             prev_journal = journal
 206             prev_period = period
 207         return rows
 208
 209 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: