audrius/account-reconcile
1
0
Fork 1
mirror of https://github.com/OCA/account-reconcile.git synced 2025-01-20 12:27:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

[IMP] Allow to import xlsx files

Browse Source
This commit is contained in:
Florian da Costa
2014-02-12 15:50:40 +01:00
parent 59116872ea
commit 3278d9990e
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
account_statement_base_import/parser/file_parser.py
View File

@@ -35,24 +35,24 @@ def float_or_zero(val):
class FileParser(BankStatementImportParser): class FileParser(BankStatementImportParser):
""" """
Generic abstract class for defining parser for .csv or .xls file format. Generic abstract class for defining parser for .csv, .xls or .xlsx file format.
""" """
def __init__(self, parse_name, ftype='csv', extra_fields=None, header=None, **kwargs): def __init__(self, parse_name, ftype='csv', extra_fields=None, header=None, **kwargs):
""" """
:param char: parse_name: The name of the parser :param char: parse_name: The name of the parser
:param char: ftype: extension of the file (could be csv or xls) :param char: ftype: extension of the file (could be csv, xls or xlsx)
:param dict: extra_fields: extra fields to add to the conversion dict. In the format :param dict: extra_fields: extra fields to add to the conversion dict. In the format
{fieldname: fieldtype} {fieldname: fieldtype}
:param list: header : specify header fields if the csv file has no header :param list: header : specify header fields if the csv file has no header
""" """
super(FileParser, self).__init__(parse_name, **kwargs) super(FileParser, self).__init__(parse_name, **kwargs)
if ftype in ('csv', 'xls'): if ftype in ('csv', 'xls' ,'xlsx'):
self.ftype = ftype self.ftype = ftype[0:3]
else: else:
raise except_osv(_('User Error'), raise except_osv(_('User Error'),
_('Invalid file type %s. Please use csv or xls') % ftype) _('Invalid file type %s. Please use csv, xls or xlsx') % ftype)
self.conversion_dict = { self.conversion_dict = {
'ref': unicode, 'ref': unicode,
'label': unicode, 'label': unicode,
@@ -81,7 +81,7 @@ class FileParser(BankStatementImportParser):
def _parse(self, *args, **kwargs): def _parse(self, *args, **kwargs):
""" """
Launch the parsing through .csv or .xls depending on the Launch the parsing through .csv, .xls or .xlsx depending on the
given ftype given ftype
""" """
@@ -128,7 +128,7 @@ class FileParser(BankStatementImportParser):
def _parse_xls(self): def _parse_xls(self):
""" """
:return: dict of dict from xls file (line/rows) :return: dict of dict from xls/xlsx file (line/rows)
""" """
wb_file = tempfile.NamedTemporaryFile() wb_file = tempfile.NamedTemporaryFile()
wb_file.write(self.filebuffer) wb_file.write(self.filebuffer)
@@ -180,7 +180,7 @@ class FileParser(BankStatementImportParser):
def _from_xls(self, result_set, conversion_rules): def _from_xls(self, result_set, conversion_rules):
""" """
Handle the converstion from the dict and handle date format from Handle the converstion from the dict and handle date format from
an .xls file. an .csv, .xls or .xlsx file.
""" """
for line in result_set: for line in result_set:
for rule in conversion_rules: for rule in conversion_rules: