def handle(self, message):
'''
@description:
@param {type}
@return:
'''
logging.debug("'AnnotationMetaLabelHandler' receive message %s",
message.to_json())
if message.data is None:
return self.replyMessage(message, state=False, msg='50101')
data = message.data
if 'projectId' not in data or 'imageId' not in data or 'action' not in data:
return self.replyMessage(message,
state=False,
msg="50102",
data=None)
if data['projectId'] != self.user.projectId:
return self.replyMessage(message, state=False, msg="50103")
context = TLPContext()
project = context.get_project(self.user.projectId)
mysql = MysqlManager()
try:
image_table_name = '`AnnotationProjectImage' + str(
project.index) + '`'
target_image_result = mysql.selectOne(
"""select * from """ + image_table_name + """ where id = %s""",
(data['imageId'], ))
if target_image_result[0] == 0:
# 目标图片不存在
return self.replyMessage(message, state=False, msg="50104")
image = AnnotationProjectImage.create_by_database_result(
target_image_result[1])
if image.annotationUserId != self.user.userId:
# 当前用户没有锁定图片
return self.replyMessage(message, state=False, msg="50105")
action = data['action']
meta_label_table_name = '`AnnotationProjectImageMetaLabel' + str(
project.index) + '`'
if action == 'add':
# 为图片增加新的meta标签
if 'labelId' not in data:
# 要增加的label关联不存在
return self.replyMessage(message, state=False, msg="50106")
labelId = data['labelId']
attribute = data['attribute'] if (
'attribute' in data
and data['attribute'] is not None) else None
# 增加默认的置信度的设置
if attribute is not None:
attribute = self._merge_default_attribute(
context.get_default_attribute(), attribute)
sql = """insert into """ + meta_label_table_name + """ (`id`, `imageId`, `labelId`, `type`, `version`, `attribute`, `userId`, `createTime`, `updateTime`) values (%s, %s, %s, %s, %s, %s, %s, %s, %s)"""
now = datetime.datetime.today()
newId = str(uuid.uuid4())
count = mysql.insertOne(
sql, (newId, image.id, labelId, 'MANUAL', '0', attribute,
self.user.userId, now, now))
return self.replyMessage(message,
state=True,
msg="add meta label success.",
id=newId,
imageId=image.id,
labelId=labelId)
# if count:
# return self.replyMessage(message, state=True, msg="add meta label success.", id=newId, imageId=image.id, labelId=labelId)
# else:
# return self.replyMessage(message, state=False, msg="50107", imageId=image.id, labelId=labelId) # 插入为写入数据库
elif action == 'delete':
# 为图片取消一个meta标签
if 'id' not in data:
# 要删除的ID不存在
return self.replyMessage(message, state=False, msg="50107")
delete_id = data['id']
sql = """delete from """ + meta_label_table_name + """ where id = %s"""
count = mysql.delete(sql, (delete_id, ))
return self.replyMessage(message,
state=True,
msg="delete meta label success.",
id=delete_id,
imageId=image.id)
# if count:
# return self.replyMessage(message, state=True, msg="delete meta label success.", id=delete_id, imageId=image.id)
# else:
# return self.replyMessage(message, state=False, msg="50108", id=delete_id, imageId=image.id) # 删除未成功执行
elif action == 'update':
# 为图像的Meta修改属性
if 'id' not in data:
# 要删除的ID不存在
return self.replyMessage(message, state=False, msg="50107")
update_id = data['id']
attribute = data['attribute'] if (
'attribute' in data
and data['attribute'] is not None) else None
sql = """update """ + meta_label_table_name + """ set attribute = %s where id = %s"""
count = mysql.update(sql, (attribute, update_id))
return self.replyMessage(
message,
state=True,
msg="update meta label attribute success.",
id=update_id,
imageId=image.id)
# if count:
# return self.replyMessage(message, state=True, msg="update meta label attribute success.", id=update_id, imageId=image.id)
# else:
# return self.replyMessage(message, state=False, msg="50108", id=update_id, imageId=image.id) # 删除未成功执行
else:
# 未知的meta操作
return self.replyMessage(message,
state=False,
msg="50109",
imageId=image.id) # 未知的Action操作目标
finally:
mysql.destory()
def handle(self, message):
'''
@description:
@param {type}
@return:
'''
logging.debug("'AnnotationRegionLabelHandler' receive message %s",
message.to_json())
if message.data is None:
return self.replyMessage(message, state=False, msg='40101')
data = message.data
if 'projectId' not in data or 'imageId' not in data or 'action' not in data:
return self.replyMessage(message,
state=False,
msg="40102",
data=None)
if data['projectId'] != self.user.projectId:
return self.replyMessage(message, state=False, msg="40103")
context = TLPContext()
project = context.get_project(self.user.projectId)
mysql = MysqlManager()
try:
image_table_name = '`AnnotationProjectImage' + str(
project.index) + '`'
region_table_name = '`AnnotationProjectImageRegion' + str(
project.index) + '`'
region_label_table_name = '`AnnotationProjectImageRegionLabel' + str(
project.index) + '`'
target_image_result = mysql.selectOne(
"""select * from """ + image_table_name + """ where id = %s""",
(data['imageId'], ))
if target_image_result[0] == 0:
# 目标图片不存在
return self.replyMessage(message, state=False, msg="40104")
image = AnnotationProjectImage.create_by_database_result(
target_image_result[1])
if image.annotationUserId != self.user.userId:
# 当前用户没有锁定图片
return self.replyMessage(message, state=False, msg="40105")
action = data['action']
userId = self.user.userId
now = datetime.datetime.today()
insert_region_sql = """insert into """ + region_table_name + """ (`id`, `imageId`, `type`, `index`, `shape`, `shapeData`, `userId`, `createTime`, `updateTime`) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)"""
insert_region_label_sql = """insert into """ + region_label_table_name + """(`id`, `imageId`, `regionId`, `labelId`, `type`, `version`, `attribute`, `userId`, `createTime`, `updateTime`) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)"""
update_region_label_sql = """update """ + region_label_table_name + """ set attribute = %s, updateTime = %s where id = %s"""
if action == 'save-all':
# 更新全部的页面Region标签
if 'regions' not in data:
return self.replyMessage(message, state=False, msg="40109")
regions = data['regions']
(insert_region_list, insert_region_lable_list,
update_region_list, update_region_label_list
) = self.__create_region_and_region_label_data(
context, regions, image.id, userId, now)
# if not insert_region_list and not update_region_list and not update_region_label_list:
# return self.replyMessage(message, state=False, msg="40110") # 没有可以写入的数据
if insert_region_list:
insert_region_data = []
for region_obj in insert_region_list:
insert_region_data.append(
region_obj.to_value_list(
('id', 'imageId', 'type', 'index', 'shape',
'shapeData', 'userId', 'createTime',
'updateTime')))
# 新的标注形状写入
mysql.close_transaction_insert_many(
insert_region_sql, insert_region_data)
if insert_region_lable_list:
# 继续写入需要新的label数据
insert_region_label_data = []
for label_object in insert_region_lable_list:
insert_region_label_data.append(
label_object.to_value_list(
('id', 'imageId', 'regionId', 'labelId',
'type', 'version', 'attribute', 'userId',
'createTime', 'updateTime')))
# 新的标注属性
mysql.close_transaction_insert_many(
insert_region_label_sql, insert_region_label_data)
if update_region_list:
# 如果有需要更新的矩形数据--更新
update_region_sql = """update """ + region_table_name + """ set shapeData = %s, updateTime = %s where id = %s"""
for region_obj in update_region_list:
mysql.update(update_region_sql,
parameter=(region_obj.shapeData,
region_obj.updateTime,
region_obj.id),
auto_commit=False)
if update_region_label_list:
# 如果有需要更新的矩形标签--更新
for label_object in update_region_label_list:
mysql.update(update_region_label_sql,
parameter=(label_object.attribute,
label_object.updateTime,
label_object.id),
auto_commit=False)
if 'deleteRegions' in data:
# 删除对应的区域和属性
delete_region_sql = """delete from """ + region_table_name + """ where id = %s"""
deleteRegions = data['deleteRegions']
for region_id in deleteRegions:
mysql.delete(sql=delete_region_sql,
parameter=(region_id, ),
auto_commit=False)
if 'deleteRegionLabels' in data:
# 删除对应的属性
delete_region_label_sql = """delete from """ + region_label_table_name + """ where id = %s"""
deleteRegionLabels = data['deleteRegionLabels']
for region_label_id in deleteRegionLabels:
mysql.delete(sql=delete_region_label_sql,
parameter=(region_label_id, ),
auto_commit=False)
# 都执行成功了，commit一次到数据库
mysql.end()
get_all_region_sql = """select * from """ + region_table_name + """ where imageId = %s"""
image_all_region_result = mysql.selectAll(
get_all_region_sql, (image.id, ))
image_all_region_label_result = []
result_data = []
if image_all_region_result[0]:
# 有相关的数据，返回的时候返回当前数据的所有结构
get_all_region_label_where_sql = ""
for region in image_all_region_result[1]:
get_all_region_label_where_sql += "regionId = '" + region[
"id"].decode("utf-8") + "' or "
get_all_region_label_where_sql = get_all_region_label_where_sql[
0:-3]
get_all_region_label_sql = """select * from """ + region_label_table_name + """ where """ + get_all_region_label_where_sql
image_all_region_label_result = mysql.selectAll(
get_all_region_label_sql)
for region in image_all_region_result[1]:
region_dict = AnnotationProjectImageRegion.convert_database_result_2_dict(
region)
region_dict["labels"] = []
if image_all_region_label_result[0]:
for region_label in image_all_region_label_result[
1]:
if region_label["regionId"].decode(
"utf-8") == region_dict['id']:
region_dict["labels"].append(
AnnotationProjectImageRegionLabel.
convert_database_result_2_dict(
region_label))
result_data.append(region_dict)
return self.replyMessage(
message,
state=True,
msg="save all region lable success.",
imageId=image.id,
regions=result_data,
action="save-all")
else:
# 没有形状信息就直接返回一个图片的ID
return self.replyMessage(
message,
state=True,
msg="save all region lable success.",
imageId=image.id,
action="save-all")
elif action == "save-one":
if "regionId" not in data:
return self.replyMessage(message,
state=False,
msg="请指定要标签的区域信息")
region_id = data['regionId']
if "labelId" not in data or not data['labelId']:
return self.replyMessage(message,
state=False,
msg="请指定要标签的模板信息")
label_id = data['labelId']
lid = None
if "id" in data and len(data['id']) > 0:
lid = data["id"]
attribute = None
if "attribute" in data and data['attribute']:
attribute = data["attribute"]
# 增加默认的置信度的设置
if attribute is not None:
attribute = self._merge_default_attribute(
context.get_default_attribute(), attribute)
if region_id == "":
region_id = str(uuid.uuid4())
new_region = AnnotationProjectImageRegion(
id=region_id,
imageId=image.id,
type='MANUAL',
index=data['index'],
shape=data['shape'],
shapeData=data['shapeData'],
userId=userId,
createTime=now,
updateTime=now)
mysql.close_transaction_insert_many(
insert_region_sql, (new_region.to_value_list(
('id', 'imageId', 'type', 'index', 'shape',
'shapeData', 'userId', 'createTime',
'updateTime')), ))
label_obj = AnnotationProjectImageRegionLabel(
id=lid,
imageId=image.id,
regionId=region_id,
labelId=label_id,
type="MANUAL",
version="0",
attribute=attribute,
userId=userId,
createTime=now,
updateTime=now)
if label_obj.id is None:
# insert
label_obj.id = str(uuid.uuid4())
mysql.close_transaction_insert_many(
insert_region_label_sql, (label_obj.to_value_list(
('id', 'imageId', 'regionId', 'labelId', 'type',
'version', 'attribute', 'userId', 'createTime',
'updateTime')), ))
else:
# update
mysql.update(sql=update_region_label_sql,
parameter=(label_obj.attribute, now,
label_obj.id),
auto_commit=False)
# 都执行成功了，commit一次到数据库
mysql.end()
return self.replyMessage(message,
state=True,
msg="区域标签保存成功",
action="save-one",
labelInfo=label_obj.to_dict())
except Exception as e:
mysql.end(False)
raise RunTimeException("save all regions code error.")
finally:
mysql.destory()