def gen_companies(self):
"""
Generate companies as expected
"""
from autonomie.models.company import Company
companies = []
for data in self.activity_companydatas:
# Try to retrieve an existing company (and avoid duplicates)
company = Company.query().filter(
Company.name == data.name
).first()
if company is None:
company = Company(
name=data.name,
goal=data.title,
email=self.coordonnees_email1,
phone=self.coordonnees_tel,
mobile=self.coordonnees_mobile,
)
if data.activity is not None:
company.activities.append(data.activity)
company.employees.append(self.user)
companies.append(company)
return companies
def company2(dbsession, user):
from autonomie.models.company import Company
company = Company(
name=u"Company 2",
email=u"*****@*****.**",
)
company.employees = [user]
dbsession.add(company)
dbsession.flush()
user.companies = [company]
user = dbsession.merge(user)
dbsession.flush()
return company
def _mk_company(name, email, code_compta):
from autonomie.models.company import Company
company = Company(
name=name,
email=email,
code_compta=code_compta,
)
company.employees = [user]
dbsession.add(company)
dbsession.flush()
user.companies = [company]
dbsession.merge(user)
dbsession.flush()
return company
def other_company(dbsession, other_user):
from autonomie.models.company import Company
company = Company(
name=u"Company2",
email=u"*****@*****.**",
code_compta="0USER2",
)
company.employees = [other_user]
dbsession.add(company)
dbsession.flush()
other_user.companies = [company]
other_user = dbsession.merge(other_user)
dbsession.flush()
return company
def _add_company(self, name, user):
"""
Add a company 'name' in the database
( set its goal by default )
"""
logger.info(u"Adding company : %s" % name)
company = Company()
company.name = name
company.goal = u"Entreprise de {0}".format(
format_account(user, reverse=False))
company.contribution = self.request.config.get('contribution_cae')
company = self.dbsession.merge(company)
self.dbsession.flush()
return company
def populate_db(session):
from autonomie.models.user import User
user = User(
login='******',
firstname='user1_firstname',
lastname="user1_lastname",
email="*****@*****.**"
)
user.set_password('o')
session.add(user)
from autonomie.models.project import Project
project = Project(
name='Projet 1',
code='P001',
definition="Projet 1"
)
session.add(project)
from autonomie.models.customer import Customer
cust = Customer(
code='C001',
name='Client1',
contactLastName=u'Client Lastname',
address=u'15 rue Victore Hugo',
zipCode='69003',
city='Lyon',
)
cust.projects.append(project)
session.add(cust)
from autonomie.models.project import Phase
phase = Phase(name='Phase de test')
phase.project = project
session.add(phase)
from autonomie.models.company import Company
c = Company(
name="company1",
goal="Company of user1",
phone='0457858585',
)
c.employees.append(user)
c.customers.append(cust)
c.projects.append(project)
session.add(c)
from autonomie.scripts import fake_database
fake_database.set_configuration()
def add_company(user, company_name, goal=""):
company = Company()
company.name = company_name
company.goal = goal or u"Entreprise de %s" % user.login
user.companies.append(company)
session = DBSESSION()
session.add(company)
session.flush()
print "Added company for %s: %s" % (user.login, company_name)
return company
def submit_success(self, appstruct):
"""
Edit the database entry and return redirect
"""
user_id = appstruct.get('user_id')
company = Company()
company.activities = fetch_activities_objects(appstruct)
company = merge_session_with_post(company, appstruct)
if user_id is not None:
user_account = User.get(user_id)
if user_account is not None:
company.employees.append(user_account)
self.dbsession.add(company)
self.dbsession.flush()
message = u"L'entreprise '{0}' a bien été ajoutée".format(company.name)
self.session.flash(message)
return HTTPFound(self.request.route_path("company", id=company.id))
def company2(dbsession):
from autonomie.models.company import Company
comp = Company(name="New company", email="*****@*****.**")
dbsession.add(comp)
dbsession.flush()
return comp
def company_125(dbsession):
from autonomie.models.company import Company
c = Company(name=u'testit', code_compta=CODE, email="*****@*****.**")
dbsession.add(c)
return c
def test_get_company_id():
assert Company(id=1).get_company_id() == 1
