matt-bernhardt/trapp

# -*- coding: utf-8 -*-
from trapp.importer import Importer
from trapp.player import Player


class ImporterPlayers(Importer):

    def correctValues(self):
        for record in self.records:
            if (record['DOB'] != ''):
                record['DOB'] = self.source.recoverDate(record['DOB'])

        return True

    def importRecord(self, record):
        record['PlayerName'] = (
            record['FirstName'] + " " + record['LastName']
        ).strip()

        self.log.message('Importing player ' + str(record))

        p = Player()
        p.connectDB()

        # Does the record exist?
        found = p.lookupIDbyName(record, self.log)
        if (len(found) == 0):
            # Nothing found, so we import
            p.saveDict(record, self.log)
            self.imported += 1
        elif (len(found) == 1):
            # Found one record, so we update
            record['PlayerID'] = found[0]
            p.saveDict(record, self.log)
            self.updated += 1
        else:
            # Something(s) found, so we skip
            self.processMissingRecords(found, len(found))

        self.log.message('')

        return True