anapaulagomes · October 3, 2024 06:13 · diegoaugustotc · Oct 2, 2024 · anapaulagomes · Oct 3, 2024
diff --git a/README.md b/README.md
diff --git a/classificacao-da-despesa-orcamentaria-por-natureza-2018.csv b/classificacao-da-despesa-orcamentaria-por-natureza-2018.csv
diff --git a/classificacao-da-despesa-orcamentaria-por-natureza-2020.csv b/classificacao-da-despesa-orcamentaria-por-natureza-2020.csv
diff --git a/classification.py b/classification.py
 def identify_code_and_description(row):
    """Identifica classificação da despesa e sua descrição.
    
    Exemplo de entradas esperadas:

    CO	GD	MA	ED	SED	Código	     Código sem máscara   Descrição
    3	0	00	00	00	3.0.00.00.00 30000000 Despesas Correntes
    3 	1	00	00	00	3.1.00.00.00 31000000 Pessoal e Encargos Sociais
    3	1	71	00	00	3.1.71.00.00 31710000 Transferências a Consórcios Públicos Mediante Contrato de Rateio
    3	1	71	70	00	3.1.71.70.00 31717000 Rateio pela Participação em Consórcio Público
    3	1	90	01	01	3.1.90.01.01 31900101 Aposentadorias Custeadas com Recursos do RPPS
    """
    from_pandas = False
    try:
        import pandas as pd
        if isinstance(row, pd.Series) or isinstance(row, pd.DataFrame):
            from_pandas = True
    except:
        pass
    if from_pandas:
        if row.empty or len(row) < 7:
            return
    elif not row or len(row) < 7:
            return

    code_and_description = {'code': row[5], 'description': row[7]}

    is_gd_empty = row[1] == '0'
    is_ma_empty = row[2] == '00'
    is_ed_empty = row[3] == '00'
    is_sed_empty = row[4] == '00'

    is_co = all([
        is_gd_empty,
        is_ma_empty,
        is_ed_empty,
        is_sed_empty,
    ])
    if is_co:
        code_and_description['value'] = row[0]
        code_and_description['group'] = 'CO'
        return code_and_description
    is_gd = all([
        is_ma_empty,
        is_ed_empty,
        is_sed_empty,
    ])
    if is_gd:
        code_and_description['value'] = row[1]
        code_and_description['group'] = 'GD'
        return code_and_description
    is_ma = all([
        is_ed_empty,
        is_sed_empty,
    ])
    if is_ma:
        code_and_description['value'] = row[2]
        code_and_description['group'] = 'MA'
        return code_and_description
    is_ed = is_sed_empty
    if is_ed:
        code_and_description['value'] = row[3]
        code_and_description['group'] = 'ED'
        return code_and_description
    is_sed = not all([
        is_gd_empty,
        is_ma_empty,
        is_ed_empty,
        is_sed_empty,
    ])
    if is_sed:
        code_and_description['value'] = row[4]
        code_and_description['group'] = 'SED'
        return code_and_description
diff --git a/especificacao-despesas-tcm-bahia.csv b/especificacao-despesas-tcm-bahia.csv
diff --git a/test_classification.py b/test_classification.py
 from classification import identify_code_and_description
 import pytest


 @pytest.mark.parametrize('row,expected_description', [
    (
        ['3', '0', '00', '00', '00', '3.0.00.00.00', '3.0.00.00.00', 'Despesas Correntes'],
        {
            'value': '3',
            'group': 'CO',
            'code': '3.0.00.00.00',
            'description': 'Despesas Correntes'
        }
    ),
    (
        ['3', '1', '00', '00', '00', '3.1.00.00.00', '3.0.00.00.00', 'Pessoal e Encargos Sociais'],
        {
            'value': '1',
            'group': 'GD',
            'code': '3.1.00.00.00',
            'description': 'Pessoal e Encargos Sociais'
        }
    ),
    (
        ['3', '1', '71', '00', '00', '3.1.71.00.00', '3.0.00.00.00', 'Transferências a Consórcios Públicos Mediante Contrato de Rateio'],
        {
            'value': '71',
            'group': 'MA',
            'code': '3.1.71.00.00',
            'description': 'Transferências a Consórcios Públicos Mediante Contrato de Rateio'
        }
    ),
    (
        ['3', '1', '71', '70', '00', '3.1.71.70.00', '3.0.00.00.00', 'Rateio pela Participação em Consórcio Público'],
        {
            'value': '70',
            'group': 'ED',
            'code': '3.1.71.70.00',
            'description': 'Rateio pela Participação em Consórcio Público'
        }
    ),
    (
        ['3', '1', '90', '01', '01', '3.1.90.01.01', '3.0.00.00.00', 'Aposentadorias Custeadas com Recursos do RPPS'],
        {
            'value': '01',
            'group': 'SED',
            'code': '3.1.90.01.01',
            'description': 'Aposentadorias Custeadas com Recursos do RPPS'
        }
    ),
    ([], None),
    (None, None),
    (['3', '1', '90'], None)
 ])
 def test_identify_code_and_description(row, expected_description):
    assert identify_code_and_description(row) == expected_description
Coluna	Descrição
CO	Categoria Econômica
GD	Grupo de Natureza da Despesa
MA	Modalidade de Aplicação
ED	Elemento de Despesa
SED	Item de Despesa (subelemento)
Código	Código completo
Código sem máscara	Código completo sem máscara (pontos)
Descrição	Descrição da despesa
Tipo
	def identify_code_and_description(row):
	"""Identifica classificação da despesa e sua descrição.

	Exemplo de entradas esperadas:

	CO GD MA ED SED Código Código sem máscara Descrição
	3 0 00 00 00 3.0.00.00.00 30000000 Despesas Correntes
	3 1 00 00 00 3.1.00.00.00 31000000 Pessoal e Encargos Sociais
	3 1 71 00 00 3.1.71.00.00 31710000 Transferências a Consórcios Públicos Mediante Contrato de Rateio
	3 1 71 70 00 3.1.71.70.00 31717000 Rateio pela Participação em Consórcio Público
	3 1 90 01 01 3.1.90.01.01 31900101 Aposentadorias Custeadas com Recursos do RPPS
	"""
	from_pandas = False
	try:
	import pandas as pd
	if isinstance(row, pd.Series) or isinstance(row, pd.DataFrame):
	from_pandas = True
	except:
	pass
	if from_pandas:
	if row.empty or len(row) < 7:
	return
	elif not row or len(row) < 7:
	return

	code_and_description = {'code': row[5], 'description': row[7]}

	is_gd_empty = row[1] == '0'
	is_ma_empty = row[2] == '00'
	is_ed_empty = row[3] == '00'
	is_sed_empty = row[4] == '00'

	is_co = all([
	is_gd_empty,
	is_ma_empty,
	is_ed_empty,
	is_sed_empty,
	])
	if is_co:
	code_and_description['value'] = row[0]
	code_and_description['group'] = 'CO'
	return code_and_description
	is_gd = all([
	is_ma_empty,
	is_ed_empty,
	is_sed_empty,
	])
	if is_gd:
	code_and_description['value'] = row[1]
	code_and_description['group'] = 'GD'
	return code_and_description
	is_ma = all([
	is_ed_empty,
	is_sed_empty,
	])
	if is_ma:
	code_and_description['value'] = row[2]
	code_and_description['group'] = 'MA'
	return code_and_description
	is_ed = is_sed_empty
	if is_ed:
	code_and_description['value'] = row[3]
	code_and_description['group'] = 'ED'
	return code_and_description
	is_sed = not all([
	is_gd_empty,
	is_ma_empty,
	is_ed_empty,
	is_sed_empty,
	])
	if is_sed:
	code_and_description['value'] = row[4]
	code_and_description['group'] = 'SED'
	return code_and_description
	from classification import identify_code_and_description
	import pytest


	@pytest.mark.parametrize('row,expected_description', [
	(
	['3', '0', '00', '00', '00', '3.0.00.00.00', '3.0.00.00.00', 'Despesas Correntes'],
	{
	'value': '3',
	'group': 'CO',
	'code': '3.0.00.00.00',
	'description': 'Despesas Correntes'
	}
	),
	(
	['3', '1', '00', '00', '00', '3.1.00.00.00', '3.0.00.00.00', 'Pessoal e Encargos Sociais'],
	{
	'value': '1',
	'group': 'GD',
	'code': '3.1.00.00.00',
	'description': 'Pessoal e Encargos Sociais'
	}
	),
	(
	['3', '1', '71', '00', '00', '3.1.71.00.00', '3.0.00.00.00', 'Transferências a Consórcios Públicos Mediante Contrato de Rateio'],
	{
	'value': '71',
	'group': 'MA',
	'code': '3.1.71.00.00',
	'description': 'Transferências a Consórcios Públicos Mediante Contrato de Rateio'
	}
	),
	(
	['3', '1', '71', '70', '00', '3.1.71.70.00', '3.0.00.00.00', 'Rateio pela Participação em Consórcio Público'],
	{
	'value': '70',
	'group': 'ED',
	'code': '3.1.71.70.00',
	'description': 'Rateio pela Participação em Consórcio Público'
	}
	),
	(
	['3', '1', '90', '01', '01', '3.1.90.01.01', '3.0.00.00.00', 'Aposentadorias Custeadas com Recursos do RPPS'],
	{
	'value': '01',
	'group': 'SED',
	'code': '3.1.90.01.01',
	'description': 'Aposentadorias Custeadas com Recursos do RPPS'
	}
	),
	([], None),
	(None, None),
	(['3', '1', '90'], None)
	])
	def test_identify_code_and_description(row, expected_description):
	assert identify_code_and_description(row) == expected_description