def handle(self, message):
'''
@description:
@param {type}
@return:
'''
logging.debug("'ListImageAnnotationInfoHandler' receive message %s",
message.to_json())
if message.data is None:
return self.replyMessage(message, state=False, msg='请填写需要提交的数据')
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="请填写关键的信息",
data=None)
if data['projectId'] != self.user.projectId:
return self.replyMessage(message,
state=False,
msg="您不属于当前项目，无法进行操作，请检查权限")
context = TLPContext()
project = context.get_project(self.user.projectId)
project_index_str = str(project.index)
mysql = MysqlManager()
try:
image_table_name = '`AnnotationProjectImage' + project_index_str + '`'
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="要操作的图片没有找到")
image = AnnotationProjectImage.create_by_database_result(
target_image_result[1])
# if image.annotationUserId != self.user.userId:
# # 当前用户没有锁定图片
# return self.replyMessage(message, state=False, msg="您并没有锁定要操作的图片，请确认信息")
action = data['action']
meta_label_table_name = '`AnnotationProjectImageMetaLabel' + project_index_str + '`'
image_region_table_name = '`AnnotationProjectImageRegion' + project_index_str + '`'
image_region_label_table_name = '`AnnotationProjectImageRegionLabel' + project_index_str + '`'
if action == 'image-meta-label':
# 查询图片得meta信息
return self.replyMessage(message,
state=True,
msg="select meta lable success.",
metaLabels=self.__select_meta_label(
mysql, meta_label_table_name,
image.id),
action=action)
elif action == 'image-region-label':
# 查询图片得区域和区域得label信息
return self.replyMessage(
message,
state=True,
msg="select region lable success.",
regions=self.__select_region_and_region_label(
mysql, image_region_table_name,
image_region_label_table_name, image.id),
action=action)
elif action == 'region-label':
# 查询单个形状得label信息
if 'regionId' not in data:
return self.replyMessage(message,
state=False,
msg="region id not found.",
regionLabels=[],
action=action)
regionId = data['regionId']
return self.replyMessage(
message,
state=True,
msg="select region label success.",
regionLabels=self.__select_one_region_label(
mysql, image_region_label_table_name, regionId),
action=action)
elif action == 'all':
# 查询所有信息
if context.get_label_filter_condition(id(self.websocket)):
(
meta_label_list, region_list
) = self.__select_all_region_and_label_use_label_filter_condition(
mysql, image.id, project_index_str,
context.get_label_filter_condition(id(self.websocket)))
return self.replyMessage(message,
state=True,
msg="select all label success.",
metaLabels=meta_label_list,
regionLabels=region_list,
action=action)
else:
meta_label_list = self.__select_meta_label(
mysql, meta_label_table_name, image.id)
region_label_list = self.__select_region_and_region_label(
mysql, image_region_table_name,
image_region_label_table_name, image.id)
return self.replyMessage(message,
state=True,
msg="select all label success.",
metaLabels=meta_label_list,
regionLabels=region_label_list,
action=action)
else:
return self.replyMessage(message,
state=False,
msg="不明确得动作定义，请指定查询内容")
except Exception as e:
mysql.end(False)
raise RunTimeException("get all regions code error.")
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()