def Insert(self,companies):
dao = CompanyDao()
newIds = []
for c in companies:
id = dao.insert(c.name,c.vat_number,c.code)
newIds.append(id)
if len(c.company_types) > 0:
r = [t for t in c.company_types]
dao.updateTypes(id,r)
return self.Get(newIds)
def Update(self,companies):
dao = CompanyDao()
logging.info("UPDATE DAO")
for c in companies:
logging.info("UPDATE COMPANY")
logging.info(str(c.id))
logging.info(str(c.vat_number))
logging.info(str(c.code))
logging.info("******")
dao.update(c.id,c.name,c.vat_number,c.code)
if len(c.company_types) > 0:
r = [t for t in c.company_types]
dao.updateTypes(c.id,r)
return self.Get([c.id for c in companies])
def count_addresses(self):
dao = CompanyDao()
return dao.count_all()
def get_all(self):
dao = CompanyDao()
#rows = dao.GetAllIds()
ids = [r[0] for r in dao.get_all_ids()]
return self.Get(ids)
def AddPerson(self,c,p):
dao = CompanyDao()
dao.joinPerson(c.id,p.id)
c.persons.append(p)
return (self.Get([c.id]))
def add_address(self,c,a):
dao = CompanyDao()
dao.joinAddress(c.id,a.id)
c.addresses.append(a)
return (self.Get([c.id]))
