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"