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)
