import yaml
import pandas as pd
import argparse


parser = argparse.ArgumentParser()
parser.add_argument('--index_yml', default='/home/mao/Downloads/operation-work/explore/index.yml', type=str)
parser.add_argument('--sort_values', default='notification count', type=str)
parser.add_argument('--result_name', default='weekly_growth.csv', type=str)
args = parser.parse_args()

with open(args.index_yml, 'r', encoding='utf-8') as stream:
    try:
        parse_yml = yaml.safe_load(stream)
        print(parse_yml)
    except yaml.YAMLError as exec:
        print(exec)
    

parse_yml['projects']
    

csv = pd.read_csv(args.result_name, usecols=['project full path', 'project url', 'notification count', 'star count'])
    

csv.head()
    

csv = csv.sort_values(args.sort_values, ascending=False)
    

csv.head()
    

csv.iloc[0, 0]
    

for i in range(len(parse_yml['projects'])):
    parse_yml['projects'][i] = csv.iloc[i+25, 0]
    

parse_yml['projects']
    

with open(args.index_yml, 'w', encoding='utf-8') as stream:
    try:
        yaml.safe_dump(parse_yml, stream)
    except yaml.YAMLError as exec:
        print(exec)