mirror of
https://github.com/firmai/financial-machine-learning.git
synced 2026-07-27 18:57:54 +00:00
Bin Yang
91 lines
3.5 KiB
Python
91 lines
3.5 KiB
Python
import os
|
|
from conf import PROJECT_ROOT_DIR
|
|
import re
|
|
import pandas as pd
|
|
|
|
from git_util import get_repo_attributes_dict, get_github_client, get_repo_path
|
|
|
|
|
|
def get_repo_list():
|
|
repo_df = pd.read_csv(os.path.join(PROJECT_ROOT_DIR, 'raw_data', 'url_list.csv'))
|
|
if 'repo_path' not in repo_df.columns:
|
|
repo_df['repo_path'] = repo_df['url'].apply(get_repo_path)
|
|
return repo_df
|
|
|
|
|
|
def get_repo_status():
|
|
g = get_github_client()
|
|
repo_df = get_repo_list()
|
|
for idx, row in repo_df.iterrows():
|
|
repo_path = row['repo_path']
|
|
if not pd.isna(repo_path):
|
|
try:
|
|
print('processing [{}]'.format(repo_path))
|
|
repo = g.get_repo(repo_path)
|
|
|
|
repo_attr_dict = get_repo_attributes_dict(repo)
|
|
except Exception as ex:
|
|
print(ex)
|
|
repo_attr_dict = {}
|
|
|
|
for k, v in iter(repo_attr_dict.items()):
|
|
repo_df.loc[idx, k] = v
|
|
repo_df.to_csv(os.path.join(PROJECT_ROOT_DIR, 'raw_data', 'url_list.csv'), index=False)
|
|
|
|
|
|
@DeprecationWarning
|
|
def parse_readme_md():
|
|
"""
|
|
|
|
:return:
|
|
usage:
|
|
>>> df = parse_readme_md()
|
|
>>> df.to_csv(os.path.join(PROJECT_ROOT_DIR, 'raw_data', 'url_list.csv'), index=False)
|
|
"""
|
|
file_path = os.path.join(PROJECT_ROOT_DIR, 'README.md')
|
|
with open(file_path) as f:
|
|
lines = f.readlines()[11:] # skip heading
|
|
all_df_list = []
|
|
for line_num in range(len(lines)):
|
|
line = lines[line_num]
|
|
if line.strip().startswith('#'):
|
|
# find a heading
|
|
heading = line.strip().replace('#', '').replace('\n', '').strip()
|
|
# parse until next # or eof
|
|
parsed_list = []
|
|
line_num += 1
|
|
while line_num < len(lines) and not lines[line_num].strip().startswith('#'):
|
|
link_line = lines[line_num].replace('\n', '').strip()
|
|
if len(link_line) > 0:
|
|
# usually in the format of '- [NAME](link) - comment
|
|
split_sections = link_line.split('- ')
|
|
if len(split_sections) == 2:
|
|
comment_str = None
|
|
elif len(split_sections) >= 3:
|
|
comment_str = '-'.join(split_sections[2:]).strip()
|
|
else:
|
|
raise Exception('link_line [{}] not supported'.format(link_line))
|
|
|
|
title_and_link = split_sections[1].strip()
|
|
title = re.search(r'\[(.*?)\]', title_and_link)
|
|
title_str = None
|
|
if title is not None:
|
|
title_str = title.group(1)
|
|
title_and_link = title_and_link.replace('[{}]'.format(title_str), '')
|
|
m_link = re.search(r'\((.*?)\)', title_and_link)
|
|
link_str = None
|
|
if m_link is not None:
|
|
link_str = m_link.group(1)
|
|
parsed_set = (title_str, link_str, comment_str)
|
|
parsed_list.append(parsed_set)
|
|
line_num += 1
|
|
parsed_df = pd.DataFrame(parsed_list, columns=['name', 'url', 'comment'])
|
|
parsed_df['category'] = heading
|
|
all_df_list.append(parsed_df)
|
|
final_df = pd.concat(all_df_list).reset_index(drop=True)
|
|
return final_df
|
|
|
|
|
|
if __name__ == '__main__':
|
|
get_repo_status()
|