gavindiaz/financial-machine-learning
1
0
Fork 0
mirror of https://github.com/firmai/financial-machine-learning.git synced 2026-08-21 23:08:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

setup github action to update wiki weekly as well

Browse Source
This commit is contained in:
Bin Yang
2021-03-31 17:07:14 -04:00
parent 64f9ec4adf
commit a9f0c64c74
20 changed files with 410 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wiki_gen.py
+37 -8
View File
@@ -1,6 +1,25 @@
from conf import PROJECT_ROOT_DIR
import os
import pandas as pd
import numpy as np
from git_status import get_repo_list
def get_wiki_status_color(input_text):
if input_text is None or input_text == 'inactive':
result_text = "![inactive](https://placehold.it/15/FF0000/000000?text=+)"
else:
result_text = "![active](https://placehold.it/15/00FF00/000000?text=+)"
return '<sub>{}</sub>'.format(result_text)
def get_wiki_rating(input_rating):
result_text = ''
if input_rating is not None and not np.isnan(input_rating):
rating = int(input_rating)
result_text = ':star:x{}'.format(rating)
return '<sub>{}</sub>'.format(result_text)
def generate_wiki_per_category(output_path):
@@ -8,21 +27,31 @@ def generate_wiki_per_category(output_path):
:param output_path:
"""
repo_path = os.path.join(PROJECT_ROOT_DIR, 'raw_data', 'url_list.csv')
repo_df = pd.read_csv(repo_path)
repo_df = get_repo_list()
for category in repo_df['category'].unique():
category_df = repo_df[repo_df['category'] == category].copy()
url_md_list = []
for idx, irow in category_df[['name', 'url']].iterrows():
url_md_list.append('[{}]({})'.format(irow['name'], irow['url']))
url_md_list.append('<sub>[{}]({})</sub>'.format(irow['name'], irow['url']))
formatted_df = pd.DataFrame({
'repo': url_md_list,
'comment': category_df['comment'],
'created_at': category_df['created_at'],
'last_commit': category_df['last_commit'],
'star_count': category_df['star_count'],
'comment': category_df['comment'].apply(lambda x: '<sub>{}</sub>'.format(x)),
'created_at': category_df['created_at'].apply(lambda x: '<sub>{}</sub>'.format(x)),
'last_commit': category_df['last_commit'].apply(lambda x: '<sub>{}</sub>'.format(x)),
'star_count': category_df['star_count'].apply(lambda x: '<sub>{}</sub>'.format(x)),
'repo_status': category_df['repo_status'],
'rating': category_df['rating']
})
# add color for the status
formatted_df['repo_status'] = formatted_df['repo_status'].apply(lambda x: get_wiki_status_color(x))
formatted_df['rating'] = formatted_df['rating'].apply(lambda x: get_wiki_rating(x))
output_path_full = os.path.join(output_path, '{}.md'.format(category))
with open(output_path_full, 'w') as f:
f.write(formatted_df.to_markdown(index=False))
if __name__ == '__main__':
local_path = os.path.join(PROJECT_ROOT_DIR, 'generated_wiki')
generate_wiki_per_category(local_path)