Merge pull request #93 from staticdev/enhancement/add-2021

Add 2021

README.rst

@@ -38,7 +38,7 @@ IRPF Investidor
 
 Programa auxiliar para calcular custos de ações, ETFs e FIIs. Este programa foi feito para calcular emolumentos, taxa de liquidação e custo total para a declaração de Bens e Direitos do Imposto de Renda Pessoa Física.
 
-**Essa aplicação foi testada e configurada para calcular tarifas referentes aos anos de 2019 a 2020 (IRPF 2020/2021) e não faz cálculos para compra e venda no mesmo dia (Day Trade), contratos futuros e Índice Brasil 50.**
+**Essa aplicação foi testada e configurada para calcular tarifas referentes aos anos de 2019 a 2021 (IRPF 2020/2022) e não faz cálculos para compra e venda no mesmo dia (Day Trade), contratos futuros e Índice Brasil 50.**
 
 
 Requisitos
@@ -78,6 +78,8 @@ Uso
 
 1. Entre na `Área do Investidor`_ da B3, faça login e entre no menu Extratos e Informativos → Negociação de Ativos → Escolha uma corretora e as datas 1 de Janeiro e 31 de Dezembro do ano em que deseja declarar. Em seguida clique no botão “Exportar para EXCEL”. Ele irá baixar o arquivo “InfoCEI.xls”.
 
+**Ainda não é possível rodar o programa usando os novos arquivos XLSX, gerar no formato antigo.**
+
 Você pode combinar lançamentos de anos diferentes em um mesmo documento colando as linhas de um relatório em outro, mas mantenha a ordem cronológica.
 
 2. Execute o programa através do comando:

pyproject.toml

@@ -48,6 +48,7 @@ sphinx-click = ">=3.0.2"
 typeguard = ">=2.13.3"
 xdoctest = {extras = ["colors"], version = ">=0.15.10"}
 pyfakefs = ">=4.5.3"
+bandit = "==1.7.2"
 
 [tool.poetry.scripts]
 irpf-investidor = "irpf_investidor.__main__:main"

src/irpf_investidor/b3.py

@@ -71,7 +71,7 @@
         datetime.datetime(2020, 11, 1), datetime.datetime(2020, 12, 1), 0.00003247
     ),
     RatePeriod(
-        datetime.datetime(2020, 12, 1), datetime.datetime(2021, 1, 1), 0.00003020
+        datetime.datetime(2020, 12, 1), datetime.datetime(2022, 1, 1), 0.00003020
     ),
 ]
 EMOLUMENTOS_AUCTION_RATE = 0.00007
@@ -700,6 +700,10 @@ def get_asset_info(code: str) -> AssetInfo:
     """
     if code in STOCKS:
         return AssetInfo("STOCKS", STOCKS[code])
+    # STOCKS CAN END IN F
+    elif code.endswith("F") and code[:-1] in STOCKS:
+        code = code[:-1]
+        return AssetInfo("STOCKS", STOCKS[code])
     # ETF and FII code can end in 11 or 11B
     if len(code) == 6 and code.endswith("11"):
         code = code[:-2]

src/irpf_investidor/report_reader.py

@@ -20,6 +20,8 @@
     "STOCKS": "31 (Ações)",
     "NOT_FOUND": "Não encontrado",
 }
+FIRST_IMPLEMENTED_YEAR = 2019
+LAST_IMPLEMENTED_YEAR = 2021
 
 
 def get_xls_filename() -> str:
@@ -45,7 +47,11 @@ def validate_period(first: str, second: str) -> int:
     """Consider the year from the first trade date."""
     first_year = int(first[-4:])
     second_year = int(second[-4:])
-    if first_year <= second_year and first_year > 2018 and second_year < 2021:
+    if (
+        first_year <= second_year
+        and first_year >= FIRST_IMPLEMENTED_YEAR
+        and second_year <= LAST_IMPLEMENTED_YEAR
+    ):
         return second_year
     return sys.exit(
         f"Erro: o período de {first} a {second} não é válido, favor verificar "

tests/test_b3.py

@@ -27,6 +27,13 @@ def test_get_asset_info_stock() -> None:
     assert asset_info.cnpj == "33.000.167/0001-01"
 
 
+def test_get_asset_info_stock_fractionary() -> None:
+    """Return STOCKS."""
+    asset_info = b3.get_asset_info("PETR4F")
+    assert asset_info.category == "STOCKS"
+    assert asset_info.cnpj == "33.000.167/0001-01"
+
+
 def test_get_asset_info_not_found() -> None:
     """Return NOT_FOUND."""
     asset_info = b3.get_asset_info("OMG3M3")

tests/test_report_reader.py

@@ -90,9 +90,9 @@ def test_get_xls_filename_download_folder(
 def test_validate_period_success() -> None:
     """Return reference year."""
     first_date = "01/01/2019"
-    second_date = "31/12/2020"
+    second_date = "31/12/2021"
 
-    assert report_reader.validate_period(first_date, second_date) == 2020
+    assert report_reader.validate_period(first_date, second_date) == 2021
 
 
 def test_validate_period_wrong_start_finish() -> None: