74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
# This file is part of csv_import_getmail module for Tryton.
|
|
# The COPYRIGHT file at the top level of this repository contains
|
|
# the full copyright notices and license terms.
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.model import ModelSQL, fields
|
|
from email.utils import parseaddr
|
|
import logging
|
|
|
|
__all__ = ['CSVProfile', 'CSVProfileParty']
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CSVProfile:
|
|
__metaclass__ = PoolMeta
|
|
__name__ = 'csv.profile'
|
|
parties = fields.Many2Many('csv.profile-party.party',
|
|
'profile', 'party', 'Parties')
|
|
|
|
@classmethod
|
|
def getmail(self, server, messages):
|
|
pool = Pool()
|
|
GetMail = pool.get('getmail.server')
|
|
CSVArchive = pool.get('csv.archive')
|
|
CSVProfile = pool.get('csv.profile')
|
|
|
|
for message in messages:
|
|
msgeid = str(message.uid)
|
|
if not message.attachments:
|
|
logger.info('Not attachments. Continue')
|
|
continue
|
|
if not message.from_addr:
|
|
logger.info('Not from address email. Continue')
|
|
continue
|
|
|
|
sender = parseaddr(message.from_addr)[1]
|
|
party, _ = GetMail.get_party_from_email(sender)
|
|
if not party:
|
|
logger.info('Not party from email %s' % sender)
|
|
continue
|
|
csv_profiles = CSVProfile.search([('parties', 'in', [party.id])])
|
|
if not csv_profiles:
|
|
logger.info('Not profile from party %s' % party.name)
|
|
continue
|
|
csv_profile = csv_profiles[0]
|
|
|
|
logger.info('Process email: %s' % (msgeid))
|
|
|
|
attch = None
|
|
for attachment in message.attachments:
|
|
if attachment[0][-3:].upper() == 'CSV':
|
|
attch = True
|
|
logger.info('Process import CSV: %s' % (msgeid))
|
|
csv_archive = CSVArchive()
|
|
csv_archive.profile = csv_profile
|
|
csv_archive.data = attachment[1]
|
|
csv_archive.on_change_profile()
|
|
csv_archive.save()
|
|
CSVArchive().import_csv([csv_archive])
|
|
if not attch:
|
|
logger.info('Not attachment CSV: %s' % (msgeid))
|
|
|
|
return True
|
|
|
|
|
|
class CSVProfileParty(ModelSQL):
|
|
'Profile - Party'
|
|
__name__ = 'csv.profile-party.party'
|
|
_table = 'csv_profile_party_rel'
|
|
profile = fields.Many2One('csv.profile', 'CSV Profile', ondelete='CASCADE',
|
|
required=True, select=True)
|
|
party = fields.Many2One('party.party', 'Party',
|
|
ondelete='CASCADE', required=True, select=True)
|