def room_create(tokenData):
values = request.get_json()
user = User.query.filter_by(id=tokenData['userId']).first()
if user:
room = Room(name=values['name'],
description=values['description'],
user_set=str(tokenData['userId']),
owner=user.id,
avatar_image='')
db.session.add(room)
db.session.flush()
room.room_hash_id = hashlib.md5(
f'{room.id}{time.time()}'.encode('utf-8')).hexdigest()
room.user_set = f'{room.user_set},{user.id}' if room.user_set else user.id
if values['avatarImage']:
avatartImageList = values['avatarImage'].split(',')
suffix = avatartImageList[0].split('/')[1].split(';')[0]
filename = f'room_avatar/{room.room_hash_id}.{suffix}'
print(filename)
with open(f'media/{filename}', 'wb') as f:
f.write(base64.b64decode(avatartImageList[1]))
room.avatar_image = filename
user.room_id_set = f'{user.room_id_set},{room.id}' if user.room_id_set else room.id
db.session.commit()
return jsonify({
'data': {
'room': JSONHelper.model_to_json(room)
},
'message': '成功',
'status': 200
})
return jsonify({'data': '', 'message': '失败失败', 'status': 500})
