def test_get_company(dbsession): user1 = User.query().first() cid = 1 company = user1.get_company(cid) assert company.name == u'company1' with pytest.raises(KeyError): user1.get_company(3000)
def unique_login(node, value): """ Test login unicity against database """ result = User.query(only_active=False).filter_by(login=value).first() if result: message = u"Le login '{0}' n'est pas disponible.".format( value) raise colander.Invalid(node, message)
def get_user_choices(): """ Return the a list of values for the contractor autocomplete widget """ choices = [(0, u'Tous les entrepreneurs')] choices.extend([(unicode(user.id), u"{0} {1}".format(user.lastname, user.firstname),) for user in User.query().all()]) return choices
def test_gen_cancelinvoice_payment(dbsession, invoice): user = User.query().first() project = Project.query().first() invoice.project = project invoice.owner = user invoice.statusPersonAccount = user invoice.record_payment(mode="c", amount=1500) cinv = invoice.gen_cancelinvoice(user) assert len(cinv.lines) == len(invoice.lines) + len(invoice.discounts) + 1 assert cinv.lines[-1].cost == 1500
def test_company_index(config, content, get_csrf_request_with_db): from autonomie.views.company import company_index avatar = User.query().first() config.add_route('company', '/company/{cid}') config.add_static_view('static', 'autonomie:static') request = get_csrf_request_with_db() request._user = avatar request.user = avatar request.context = avatar.companies[0] response = company_index(request) assert avatar.companies[0].name == response['company'].name
def test_gen_cancelinvoice_payment(self): user = User.query().first() project = Project.query().first() inv = self.getOne() inv.project = project inv.owner = user inv.statusPersonAccount = user inv.record_payment(mode="c", amount=1500) cinv = inv.gen_cancelinvoice(user) self.assertEqual(len(cinv.lines), len(inv.lines) + len(inv.discounts) + 1) self.assertEqual(cinv.lines[-1].cost, 1500)
def upgrade(): logger = logging.getLogger("alembic.migrate_code_compta") op.add_column("company", sa.Column("code_compta", sa.String(30), default=0)) dbsession = DBSESSION() for user in User.query(): code_compta = user.code_compta companies = user.companies if code_compta not in [u"0", None, u""]: if len(companies) == 1: company = companies[0] company.code_compta = code_compta dbsession.merge(company) else: logger.warn(u"User {0} has a code_compta and multiple \ companies".format(user.id))
def auth(form, value): """ Check the login/password content """ log.debug(u" * Authenticating") login = value.get('login') log.debug(u" + Login {0}".format(login)) password = value.get('password') result = User.query().filter_by(login=login).first() if not result or not result.auth(password): log.debug(u" - Authentication Error") message = u"Erreur d'authentification" exc = colander.Invalid(form, message) exc['password'] = message raise exc
def test_gen_cancelinvoice(dbsession, invoice): user = User.query().first() project = Project.query().first() invoice.project = project invoice.owner = user invoice.statusPersonAccount = user dbsession.add(invoice) dbsession.flush() cinv = invoice.gen_cancelinvoice(user) dbsession.add(cinv) dbsession.flush() assert cinv.name == "Avoir 1" assert cinv.total_ht() == -1 * invoice.total_ht() today = datetime.date.today() assert cinv.taskDate == today
def test_gen_cancelinvoice(self): user = User.query().first() project = Project.query().first() inv = self.getOne() inv.project = project inv.owner = user inv.statusPersonAccount = user self.session.add(inv) self.session.flush() cinv = inv.gen_cancelinvoice(user) self.session.add(cinv) self.session.flush() self.assertEqual(cinv.name, "Avoir 1") self.assertEqual(cinv.total_ht(), -1 * inv.total_ht()) today = datetime.date.today() self.assertEqual(cinv.taskDate, today)
def user(content): return User.query().first()
def getone(): return User.query().filter(User.login==APPSTRUCT['login']).first()
def query(self): """ Return the main query for our list view """ return User.query(ordered=False, only_active=False)\ .outerjoin(User.companies)
def getone(self): return User.query().filter(User.login==USER['login']).first()
def getone(): return User.query().filter(User.login == APPSTRUCT['login']).first()
def user(self): return User.query().first()