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
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)