def add(category): """Add new category""" session = DBSession() session.add(category) session.commit() session.close()
def add(user): """Add new user""" session = DBSession() session.add(user) session.commit() session.close()
def add(post, categories): """Add new post and return it's id. Arguments: post - Posts object categories - list of categories id """ session = DBSession() session.add(post) # add categories to the post for id in categories: category = session.query(Categories).get(id) post.categories.append(category) id = post.id session.commit() session.close() return id