def add_comment(cls, post_id, user_id, content): if not is_id_valid(post_id): raise InvalidFieldError("post id is invalid", ["post_id"]) # not need to check user id for logged in user # if not is_id_valid(user_id): # raise InvalidFieldError("user id is invalid", ["user_id"]) post = DBPost.get_by_id(post_id) if not post: raise PostNotFoundError(post_id=post_id) if len(content) < 10: raise InvalidFieldError("comment is too short", ["content"]) args = { "post_id": post_id, "user_id": user_id, "content": content } comment = DBComment(**args) try: comment.save() return comment except: return None
def set_up_db(): args = { "email": "*****@*****.**", "password": hashlib.md5("123456").hexdigest(), "first_name": "Admin", "last_name": "Nguyen", "brief": "Hello world", "role": "manager" } user = User(**args) user.save() args["email"] = "*****@*****.**" args["first_name"] = "Editor2" args["role"] = 'editor' user = User(**args) user.save() args["email"] = "*****@*****.**" args["first_name"] = "Editor3" user = User(**args) user.save() args["email"] = "*****@*****.**" args["first_name"] = "Editor4" user = User(**args) user.save() args = { "title": "Post 1 title", "content": """<p>Lorem ipsum dolor sit amet, consectetur adipisicing elit. Eveniet doloribus qui, adipisci inventore sequi fugiat dolores ullam, provident a, accusantium, necessitatibus ab nisi aliquam. Ipsam voluptas dolores magni necessitatibus provident.</p> <p>Sunt quo placeat fugiat nesciunt vel assumenda dolorem incidunt provident eligendi ipsa, quam autem optio id nostrum beatae corporis a. Tempore saepe quod nemo hic magni in veritatis illum natus.</p> <p>Et beatae ipsam repellat officiis similique cupiditate distinctio expedita rem at, aut aspernatur, voluptate quibusdam! Voluptatum aut quos porro eos nulla officiis adipisci magnam perferendis, dicta minima quis eligendi enim.</p> <p>Sed itaque dignissimos eligendi reprehenderit, nesciunt ducimus voluptates dolores suscipit fugit ipsam aperiam praesentium laborum odit qui libero ipsum tempora, eos quis hic, sapiente perspiciatis amet labore voluptatibus alias. Vitae.</p>""", "user_id": 1, "categories": ["c++","python"] } post = Post(**args) post.save() args["title"] = "Post 2 title" args["user_id"] = 2 post = Post(**args) post.save() args["title"] = "Post 3 title" args["user_id"] = 3 post = Post(**args) post.save() args["title"] = "Post 4 title" args["user_id"] = 4 post = Post(**args) post.save() args["title"] = "Post Hello title" args["user_id"] = 1 args["categories"] = ["non-it"] post = Post(**args) post.save() args["title"] = "Good morning" args["user_id"] = 2 post = Post(**args) post.save() args["title"] = "First day at Moscow" args["user_id"] = 2 args["categories"] = ["journey","non-it"] post = Post(**args) post.save() args["title"] = "Surprising" args["user_id"] = 3 post = Post(**args) post.save() args["title"] = "So awesome lake" args["user_id"] = 3 post = Post(**args) post.save() args["title"] = "My new Phone" args["user_id"] = 3 args["categories"] = ["photo","non-it"] post = Post(**args) post.save() args["title"] = "Photo with new phone" args["user_id"] = 3 post = Post(**args) post.save() args["title"] = "List of useful app for Blackberry" args["user_id"] = 3 args["categories"] = ["uncategorized"] post = Post(**args) post.save() cmt_args={ "content":"<p>Lorem ipsum dolor sit amet, consectetur adipisicing elit. Ut ipsum ad, mollitia repellendus harum dignissimos rem beatae, dolore minus. Sapiente saepe mollitia magnam molestiae natus officiis corrupti voluptatibus, qui repudiandae.</p>", "post_id":1, "user_id":2, } cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 1 cmt_args["user_id"] = 3 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 1 cmt_args["user_id"] = 4 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 3 cmt_args["user_id"] = 4 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 3 cmt_args["user_id"] = 2 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 4 cmt_args["user_id"] = 1 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 7 cmt_args["user_id"] = 1 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 4 cmt_args["user_id"] = 2 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 3 cmt_args["user_id"] = 3 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 2 cmt_args["user_id"] = 4 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 2 cmt_args["user_id"] = 1 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 2 cmt_args["user_id"] = 3 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 2 cmt_args["user_id"] = 4 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 5 cmt_args["user_id"] = 3 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 5 cmt_args["user_id"] = 2 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 5 cmt_args["user_id"] = 3 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 5 cmt_args["user_id"] = 4 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 6 cmt_args["user_id"] = 4 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 6 cmt_args["user_id"] = 1 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 9 cmt_args["user_id"] = 1 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 9 cmt_args["user_id"] = 1 cmt = Comment(**cmt_args) cmt.save() cmt_args["post_id"] = 9 cmt_args["user_id"] = 3 cmt = Comment(**cmt_args) cmt.save()