class Master(object):
"""
"""
def __init__(self, edgar_folder):
"""
@param edgar_folder:
@return:
"""
self.edgar = EDGAR(edgar_folder)
self.index = self.edgar.load_search_idx()
"""@type: dict"""
def get_summary(self, ticker):
output = {"error": "error"}
company_data = self.index.get(ticker, None)
if company_data:
paths = company_data["file_list"]
txt_files = [self.edgar.local_folder+path for path in paths]
lr = LexRank(txt_files)
lr.lex_rank.tolist()
output = {'Full_Text': lr.original,
'Cosine_Matrix': lr.adj.tolist(),
'Lex_Rank': lr.lex_rank.reshape((-1,)).tolist(),
'Company_Name': self.index[ticker]["company_name"]}
return json.dumps(output)
def __init__(self, edgar_folder):
"""
@param edgar_folder:
@return:
"""
self.edgar = EDGAR(edgar_folder)
self.index = self.edgar.load_search_idx()
"""@type: dict"""
