示例#1
0
def get_reshaped_dfs():
    dfa, dfq, dfm = read_dfs()
    check_for_dups(dfa)
    dfa = reshape_a(dfa)
    dfq = reshape_q(dfq)
    dfm = reshape_m(dfm)
    return dfa, dfq, dfm
示例#2
0
    if 'year' in dfm.columns.values:
        dfm = dfm.drop(['year', 'month'], 1)
    return dfm

def get_dfq():
    dfa, dfq, dfm = get_reshaped_dfs()
    if 'year' in dfq.columns.values:
        dfq = dfq.drop(['year','qtr'], 1)
    return dfq
	
def get_dfa():
    dfa, dfq, dfm = get_reshaped_dfs()
    if 'year' in dfa.columns.values:
        dfa = dfa.drop(['year'], 1)
    return dfa

def get_dfs():
    return get_dfa(), get_dfq(), get_dfm()
    
def db_dump():
    dfa, dfq, dfm = get_reshaped_dfs()
    write_to_xl(dfa, dfq, dfm)
    write_to_csv(dfa, dfq, dfm)

if __name__ == "__main__":
    # repeat db_dump() here
    dfa, dfq, dfm = read_dfs()
    check_for_dups(dfa)
    dfa, dfq, dfm = reshape_all(dfa, dfq, dfm)
    write_to_xl(dfa, dfq, dfm)
    write_to_csv(dfa, dfq, dfm)