def add_comment_to_company(company_id, data): #user = UserTokenRepo().get_auth_user(request.headers.get('Authorization')) user = auth.user() company = CompanyRepo().get_company(company_id) msg, attachment_obj, upload_data = data.get('comment', None), data.get( 'attachment', None), None if attachment_obj is not None and attachment_obj.filename != '': data = uploader.upload(attachment_obj) data['uploaderId'] = data['id'] del data['id'] upload_data = UploadRepo().create(data) elif msg is None or msg == '': raise SException("Please provide some comment", redirect_url='/companies/' + company.code + '/' + company.slug) comment = CommentRepo().create({ 'user': user, 'company': company, 'message': msg, 'type': data.get('type'), 'attachment': upload_data }) return company return transformers.transform_comment(comment)
def reply_to_comment(company_id, comment_id, message, reply_to_id=None): user = auth.user() company = CompanyRepo().get_company(company_id) comment = CommentRepo().get(id=comment_id) reply = ReplyCommentRepo().create({"user": user, "message": message}) comment.replies.append(reply) comment.save() return company
def list_of_watchlist(per_page=5, page=1): user = UserRepo().objects.no_dereference().filter( id=auth.user().id).first() company_ids = [u.id for u in user.favourites] #return CompanyRepo().set_transformer(transformers.company)\ return CompanyRepo().skip_list('slice__history', 0, 1)\ .set_excludes([])\ .filter_self(id__in=company_ids)\ .paginate_self(per_page=per_page, page=page)
def unlike_a_comment(company_id, comment_id): user = auth.user() company = CompanyRepo().get_company(company_id) comment = CommentRepo().get(id=comment_id) CommentRepo().unlike(comment, user) return company
def remove_from_watchlist(company_id): company = CompanyRepo().get_company(company_id) user = UserRepo().objects.no_dereference()\ .filter(id=auth.user().id)\ .update(pull__favourites=company) return "Removed successfully"
def add_to_watchlist(company_id): company = CompanyRepo().get_company(company_id) user = UserRepo().objects.no_dereference()\ .filter(id=auth.user().id)\ .update(add_to_set__favourites=company) return "Added successfully"