60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
|
#coding=utf-8
|
|||
|
|
import os
|
|||
|
|
import json
|
|||
|
|
import utils
|
|||
|
|
import LicenseRepair
|
|||
|
|
from LicenseDataset import Licensedataset
|
|||
|
|
from treelib import Tree, Node
|
|||
|
|
|
|||
|
|
rootDir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
unDir = os.path.join(rootDir, 'repos')
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# （先加载好corenlp）
|
|||
|
|
from stanfordcorenlp import StanfordCoreNLP
|
|||
|
|
nlp = StanfordCoreNLP(os.path.join(rootDir, 'model', 'stanford-corenlp-4.2.0'))
|
|||
|
|
# （加载ee5模型）
|
|||
|
|
from EE5.LocateTerms.nermodel.ner_model import NERModel
|
|||
|
|
from EE5.LocateTerms.nermodel.config import Config
|
|||
|
|
ner_config_ee5 = Config()
|
|||
|
|
ner_model_ee5 = NERModel(ner_config_ee5)
|
|||
|
|
ner_model_ee5.build()
|
|||
|
|
ner_model_ee5.restore_session(ner_config_ee5.dir_model)
|
|||
|
|
print('【ee5 loaded. 】')
|
|||
|
|
# （加载re模型）
|
|||
|
|
from RE import re_predict
|
|||
|
|
re_args, re_model = re_predict.load_re_model()
|
|||
|
|
print('【re loaded. 】')
|
|||
|
|
# ner_model_ee5 = None
|
|||
|
|
# re_args = None
|
|||
|
|
# re_model = None
|
|||
|
|
# （加载ac模型）
|
|||
|
|
from tgrocery import Grocery
|
|||
|
|
ac_model = Grocery(os.path.dirname(os.path.abspath(__file__))+'/'+'AC/ossl2_ac')
|
|||
|
|
ac_model.load()
|
|||
|
|
print('【ac loaded. 】')
|
|||
|
|
## 加载ld
|
|||
|
|
ld = Licensedataset()
|
|||
|
|
ld.load_licenses_from_csv(nlp, ld, ner_model_ee5, re_args, re_model, ac_model)
|
|||
|
|
print('【ld loaded. 】')
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
repo = "XXXX"
|
|||
|
|
lr, fg_hasPL, num_fixable, num_incom, num_repair, methods_repair \
|
|||
|
|
= LicenseRepair.runLicenseRepair(repo, nlp, ld, ner_model_ee5, re_args, re_model, ac_model)
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# （关闭corenlp）
|
|||
|
|
nlp.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|