示例#1
0
def get_by_api_key(api_key: str):
    try:
        user = User.get_user_from_api_key(api_key, get_db_session())
    except NoResultFound:
        raise UserNotFoundException('API Key is wrong, please go to our account https://app.seamlesscloud.io/account,'
                                    ' copy the API Key field and run "smls init <api key>"')
    return user
示例#2
0
def test_user_init():
    uid, name, email = str(uuid4()), "TestUser", "*****@*****.**"
    user = User(uid=uid, name=name, email=email)

    assert isinstance(user, BaseUser)
    assert user.uid == uid
    assert user.name == name
    assert user.email == email
示例#3
0
def test_user_from_dict():
    data = {"uid": str(uuid4()), "name": "TestUser", "email": "*****@*****.**"}
    user = User.from_dict(data)

    assert isinstance(user, BaseUser)
    assert user.uid == data["uid"]
    assert user.name == data["name"]
    assert user.email == data["email"]
示例#4
0
    async def create(self, **user_dict) -> dict:
        """Создание клиента в приложении

        :param user_dict:
        :return:
        """
        user = User.from_dict(user_dict)
        return await self._storage.create(**user.to_dict())
示例#5
0
def get_user_info():
    email = session['profile']['email']
    try:
        user = User.get_user_from_email(email, get_db_session())
        if user:
            return jsonify({'email': user.email, 'api_key': user.api_key}), 200
        return jsonify({'message': 'Unable to find a user'}), 404
    except Exception as e:
        logging.exception(e)
        logging.exception(request.data.decode('utf-8'))
        return jsonify({'message': 'Unable to get an api key'}), 500
示例#6
0
def create(email: str, api_key: str = None):
    session = get_db_session()
    user = User(email=email,
                api_key=api_key or generate_api_key())
    session.add(user)
    db_commit()

    workspace = Workspace(owner_id=user.id)
    session.add(workspace)
    db_commit()
    return user.id
示例#7
0
def test_users():
    users = [User(email=f"{i}@loadtest.com", api_key=generate_api_key())
             for i in range(NUMBER_OF_TEST_USERS)]
    for user in users:
        db_session.add(user)
    db_session.commit()

    yield [row2dict(user) for user in users]

    for user in users:
        db_session.delete(user)
    db_session.commit()
示例#8
0
def get_by_id(user_id: str):
    try:
        user = User.get_user_from_id(user_id, get_db_session())
    except NoResultFound:
        raise UserNotFoundException(f'Cannot find a user with id: {user_id}')
    return user