cvm_api/main.py at master · tyasuok/cvm_api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import io
import sys
import requests
import zipfile
import pandas as pd

if __name__=="__main__":
    # a = cvm_builder("http://dados.cvm.gov.br/dados/FI/DOC/INF_DIARIO/DADOS/HIST/inf_diario_fi_2021.zip")
    # a.get_data()
    # print(a.df)

    # a = inf_diario("202112")
    # a.get_data()
    # print(a.list_files())
    # sys.exit()


    # a = inf_diario(2016)
    # a.get_data()
    # print(a.list_files())
    # dfs = []
    # for i in a.list_files():
    #     dfs.append(pd.read_csv(io.BytesIO(a.select_file(i)), sep=";", encoding="ansi"))
    #     print(len(dfs))
    # print(len(dfs))
    # df = pd.concat(dfs)
    # print(df.head())
    # print(df.shape)
    # print(df.info())
    # sys.exit()

    dfs = []
    for i in range(2005, 2021, 1):
        a = inf_diario(i)
        a.get_data()
        print(a.list_files())
        for i in a.list_files():
            dfs.append(pd.read_csv(io.BytesIO(a.select_file(i)), sep=";", encoding="ansi"))
        print(len(dfs))
    print(len(dfs))
    df = pd.concat(dfs)
    df = df.rename(columns=lambda x: x.lower())
    df = df.rename({"vl_patrim_liq": "vl_pl",
                    "captc_dia": "cpt",
                    "resg_dia": "rgt",
                    "nr_cotst": "n_cts",
                   })

    print(df.head())
    print(df.shape)
    print(df.info())
    df.to_csv("fi_doc_info_diario.csv", index=False)