trytond-csv_import/tests/scenario_csv_import.rst

===================
CSV Import Scenario
===================

Imports::

    >>> import datetime
    >>> import os
    >>> import shutil
    >>> import sys
    >>> import re
    >>> from dateutil.relativedelta import relativedelta
    >>> from decimal import Decimal
    >>> from operator import attrgetter
    >>> from proteus import config, Model, Wizard
    >>> from trytond.tests.tools import activate_modules
    >>> from trytond.config import config
    >>> from trytond.tests.test_tryton import DB_NAME as db_name
    >>> today = datetime.date.today()

    >>> module_path = os.path.dirname(__file__) + '/'

    >>> data_path = config.get('database', 'path') + '/'
    >>> db_name += '/'
    >>> module_name = 'csv_import' + '/'
    >>> if not os.path.exists(data_path):
    ...     os.makedirs(data_path)
    >>> if not os.path.exists(data_path + db_name):
    ...     os.makedirs(data_path + db_name)
    >>> if not os.path.exists(data_path + db_name + module_name):
    ...     os.makedirs(data_path + db_name + module_name)

Install csv_import::

    >>> config = activate_modules(['csv_import', 'party'])

Init models::

    >>> Model = Model.get('ir.model')
    >>> Field = Model.get('ir.model.field')
    >>> BaseExternalMapping = Model.get('base.external.mapping')
    >>> BaseExternalMappingLine = Model.get('base.external.mapping.line')
    >>> CSVProfile = Model.get('csv.profile')

Create party mapping::

    >>> model_party = Model.find([('model', '=', 'party.party')])[0]
    >>> mapping = BaseExternalMapping()
    >>> mapping.name = 'party.csv'
    >>> mapping.model = model_party
    >>> mapping.state = 'done'
    >>> mapping_line = BaseExternalMappingLine()
    >>> mapping.mapping_lines.append(mapping_line)
    >>> mapping_line.sequence = 0
    >>> mapping_line.field = Field.find([
    ...     ('name', '=', 'name'),
    ...     ('model', '=', model_party.id),
    ...     ])[0]
    >>> mapping_line.external_field = 'name'
    >>> mapping_line.mapping_type = 'in_out'
    >>> mapping_line.external_type = 'str'
    >>> mapping_line = BaseExternalMappingLine()
    >>> mapping.mapping_lines.append(mapping_line)
    >>> mapping_line.sequence = 0
    >>> mapping_line.field = Field.find([
    ...     ('name', '=', 'addresses'),
    ...     ('model', '=', model_party.id),
    ...     ])[0]
    >>> mapping_line.external_field = 'name'
    >>> mapping_line.mapping_type = 'in_out'
    >>> mapping_line.external_type = 'str'
    >>> mapping_line.in_function = 'result = []'
    >>> mapping.save()

Create address mapping::

    >>> model_address = Model.find([('model', '=', 'party.address')])[0]
    >>> mapping2 = BaseExternalMapping()
    >>> mapping2.name = 'address.csv'
    >>> mapping2.model = model_address
    >>> mapping2.state = 'done'
    >>> mapping2.csv_mapping = mapping
    >>> mapping2.csv_rel_field = Field.find([
    ...     ('name', '=', 'addresses'),
    ...     ('relation', '=', 'party.address')])[0]
    >>> mapping_line = BaseExternalMappingLine()
    >>> mapping2.mapping_lines.append(mapping_line)
    >>> mapping_line.sequence = 1
    >>> mapping_line.field = Field.find([
    ...     ('name', '=', 'street'),
    ...     ('model', '=', model_address.id),
    ...     ])[0]
    >>> mapping_line.external_field = 'street'
    >>> mapping_line.mapping_type = 'in_out'
    >>> mapping_line.external_type = 'str'
    >>> mapping_line = BaseExternalMappingLine()
    >>> mapping2.mapping_lines.append(mapping_line)
    >>> mapping_line.sequence = 2
    >>> mapping_line.field = Field.find([
    ...     ('name', '=', 'city'),
    ...     ('model', '=', model_address.id),
    ...     ])[0]
    >>> mapping_line.external_field = 'city'
    >>> mapping_line.mapping_type = 'in_out'
    >>> mapping_line.external_type = 'str'
    >>> mapping2.save()

Create profile::

    >>> CSVProfile = Model.get('csv.profile')
    >>> profile = CSVProfile()
    >>> profile.name = 'Parties'
    >>> profile.model = Model.find([('model', '=', 'party.party')])[0]
    >>> profile.create_record = True
    >>> profile.csv_header = True
    >>> profile.csv_archive_separator = ','
    >>> profile.csv_quote = '"'
    >>> profile.mappings.append(mapping)
    >>> profile.mappings.append(mapping2)
    >>> profile.save()

Create CSV archive::

    >>> srcfile = '%s/%s' % (module_path, 'import_party.csv')
    >>> dstfile = '%s/%s/%s/%s' % (data_path, db_name, module_name,
    ...     'import_party.csv')
    >>> _ = shutil.copy(srcfile, dstfile)
    >>> CSVArchive = Model.get('csv.archive')
    >>> archive = CSVArchive()
    >>> archive.profile = profile
    >>> archive.archive_name = 'import_party.csv'
    >>> archive.save()
    >>> archive.click('import_csv')

Get Party::

    >>> Party = Model.get('party.party')
    >>> party, = Party.find([('name', '=', 'Zikzakmedia')])
    >>> len(party.addresses)
    1

Create Parties and multi Addresses::

    >>> srcfile = '%s/%s' % (module_path, 'import_party_multiaddress.csv')
    >>> dstfile = '%s/%s/%s/%s' % (data_path, db_name, module_name,
    ...     'import_party_multiaddress.csv')
    >>> _ = shutil.copy(srcfile, dstfile)
    >>> CSVArchive = Model.get('csv.archive')
    >>> archive = CSVArchive()
    >>> archive.profile = profile
    >>> archive.archive_name = 'import_party_multiaddress.csv'
    >>> archive.save()
    >>> archive.click('import_csv')

Get Addresses::

    >>> Address = Model.get('party.address')
    >>> addresses = Address.find([('party', '=', 'Zikzakmedia')])
    >>> len(addresses)
    4

Create mapping line vat::

    >>> mapping_line = BaseExternalMappingLine()
    >>> mapping.mapping_lines.append(mapping_line)
    >>> mapping_line.sequence = 1
    >>> mapping_line.field = Field.find([
    ...     ('name', '=', 'code'),
    ...     ('model', '=', model_party.id),
    ...     ])[0]
    >>> mapping_line.external_field = 'code'
    >>> mapping_line.mapping_type = 'in_out'
    >>> mapping_line.external_type = 'str'
    >>> mapping.save()

Create CSV Update archive::

    >>> srcfile = '%s/%s' % (module_path, 'update_party.csv')
    >>> dstfile = '%s/%s/%s/%s' % (data_path, db_name, module_name,
    ...     'update_party.csv')
    >>> _ = shutil.copy(srcfile, dstfile)
    >>> CSVArchive = Model.get('csv.archive')
    >>> archive = CSVArchive()
    >>> archive.profile = CSVProfile.find([])[0]
    >>> archive.archive_name = 'update_party.csv'
    >>> archive.save()
    >>> csv_update = CSVArchive.import_csv([archive.id], config.context)

Get Party by code::

    >>> Party = Model.get('party.party')
    >>> parties = Party.find([('code', '=', 'C1')])
    >>> len(parties)
    1