def post(self, channel_name):
channel_id = ChannelModel.find_id_by_name(channel_name)
if channel_id:
random_data = randomData()
message_id = hashData(str(random_data))
if MessageModel.find_msg_by_channel_id_msg_id(
channel_id, message_id):
return {'message': 'Bad message id'}, 400
data = Message.parser.parse_args()
date_now = datetime.utcnow()
user_id = get_jwt_identity()
user = UserModel.find_by_id(user_id)
print(user.username)
message = MessageModel(message_id, channel_id, data['content'],
date_now, user.username,
UserModel.get_avatar(user.username))
message.save_to_db()
return message.json(), 201
else:
return {'message': 'Channel not exist.'}, 404
def delete(self, channel_name, message_id):
#claims = get_jwt_claims()
#if not claims['is_admin']:
# return {'message': 'Admin privilege required.'}, 401
channel_id = ChannelModel.find_id_by_name(channel_name)
if channel_id:
message = MessageModel.find_msg_by_channel_id_msg_id(
channel_id, message_id)
if message:
message.delete_from_db()
return {'message': 'Message was deleted.'}, 201
else:
return {'message': 'Message not found.'}, 404
