def import_all_jdk_class_belong_to_relation():
# get all-version library
cur.execute("select class_id,package_id from jdk_class")
data_list = cur.fetchall()
result_tuples = []
for i in range(0, len(data_list)):
old_class_id = data_list[i][0]
relation = create_class_belong_to_relation(
old_class_id=old_class_id, old_package_id=data_list[i][1])
if relation is None:
logger.info("None Relation")
logger.info(data_list[i])
continue
relation = relation.find_or_create(session, autocommit=False)
result_tuples.append((relation, old_class_id))
session.commit()
for item in result_tuples:
(relation, old_id) = item
record = KnowledgeTableColumnMapRecord(jdk_class_knowledge_table,
api_relation_table, old_id,
relation.id, "package_id")
record.create(session, autocommit=False)
session.commit()
logger.info("import class belong to relation completed!")
def import_all_android_class_extends_relation():
# get all-version library
cur.execute("select id,parent_class from androidAPI_class")
data_list = cur.fetchall()
result_tuples = []
for i in range(0, len(data_list)):
old_class_id = data_list[i][0]
parent_class_name = data_list[i][1]
relation = create_class_extends_relation(old_class_id=old_class_id,
parent_class_name=parent_class_name)
if relation is None:
logger.info("None Relation")
logger.info(data_list[i])
continue
relation = relation.find_or_create(session, autocommit=False)
session.commit()
for item in result_tuples:
(relation, old_id) = item
record = KnowledgeTableColumnMapRecord(android_class_knowledge_table, api_relation_table, old_id,
relation.id, "extend_class")
record.create(session, autocommit=False)
session.commit()
logger.info("import class extends relation completed!")
def commit_to_server(result_tuples):
print "success=", len(result_tuples)
session.commit()
for item in result_tuples:
(relation, old_id) = item
record = KnowledgeTableColumnMapRecord(android_method_knowledge_table, api_relation_table, old_id,
relation.id, "class_id")
record.create(session, autocommit=False)
session.commit()
def commit_to_server(result_tuples):
print "success=", len(result_tuples)
session.commit()
for item in result_tuples:
(relation, old_id) = item
record = KnowledgeTableColumnMapRecord(jdk_package_knowledge_table, api_belong_to_table, old_id,
relation.id, "library_id")
record.create(session, autocommit=False)
session.commit()
def is_imported(row_id):
if KnowledgeTableColumnMapRecord.exist_import_record(
session, jdk_method_knowledge_table, api_relation_table, row_id,
"class_id"):
return True
else:
return False
def is_imported(row_id):
if KnowledgeTableColumnMapRecord.exist_import_record(
session, jdk_package_knowledge_table, api_belong_to_table, row_id,
"library_id"):
return True
else:
return False
def test_exist_import_record(self):
session = EngineFactory.create_session()
jdk_method_knowledge_table = KnowledgeTableFactory.get_jdk_method_table(session)
api_relation_table = KnowledgeTableFactory.get_api_relation_table(session)
api_knowledge_table = KnowledgeTableFactory.get_api_entity_table(session)
result= KnowledgeTableColumnMapRecord.exist_import_record(session, jdk_method_knowledge_table, api_relation_table,
1,
"class_id")
print result
self.assertEqual(result,True)
