def create_model(self, store): person = BasePersonRoleEditor.create_model(self, store) transporter = person.transporter if transporter is None: transporter = Transporter(person=person, store=store) return transporter
def process_one(self, data, fields, store): person = Person(store=store, name=data.name, phone_number=data.phone_number, mobile_number=data.mobile_number) Company(person=person, store=store, cnpj=data.cnpj, fancy_name=data.name, state_registry=data.state_registry) ctloc = CityLocation.get_or_create(store=store, city=data.city, state=data.state, country=data.country) streetnumber = data.streetnumber and int(data.streetnumber) or None Address(is_main_address=True, person=person, city_location=ctloc, store=store, street=data.street, streetnumber=streetnumber, district=data.district) Transporter(person=person, open_contract_date=self.parse_date(data.open_contract), freight_percentage=decimal.Decimal( data.freight_percentage), store=store)
def create_transporter(self, name=u'John'): from stoqlib.domain.person import Company, Transporter person = self.create_person(name) Company(person=person, store=self.store) return Transporter(person=person, store=self.store)