Exemple #1
0
    def post(self):
        resp = dict(RESPONSE)
        args = self.parser.parse_args()
        user_identity = get_jwt_identity()
        current_user = User.query.get(user_identity)
        if not current_user:
            resp['error'] = True
            resp['msg'] = f'User {user_identity} does\'t exists'
            resp['data'] = None
            return resp, 404

        exists_name = db.session.query(User.id).filter_by(name_lower=args['name'].lower()).scalar() is not None
        if exists_name:
            resp['error'] = True
            resp['msg'] = f'Notebook with {args["name"]} name exists'
            resp['data'] = None
            return resp, 401

        notebook = Notebook(name=args['name'])
        notebook.user = current_user
        notebook.user_id = current_user.id
        db.session.add(notebook)
        db.session.commit()
        resp['data'] = marshal(notebook, notebook_fields)
        return resp, 200
Exemple #2
0
def fill(db):
    user = User(name='user', email='*****@*****.**')
    user.set_password('pass')

    notebook = Notebook(name='First notebook')
    notebook.user = user
    notebook.user_id = user.id

    note = Note(name='First note', description='Note\'s description')
    note.notebook = notebook
    note.notebook_id = notebook.id
    note.user = user
    note.user_id = user.id

    db.session.add(user)
    db.session.add(note)
    db.session.add(notebook)
    db.session.commit()