{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from impuestos import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "786384 0\n",
      "1747440 0.04\n"
     ]
    }
   ],
   "source": [
    "sueldo_bruto  = 1_200_000\n",
    "\n",
    "impuesto = 0\n",
    "for tramo, rate in TRAMOS.items():\n",
    "    print(tramo, rate)\n",
    "    delta = sueldo_bruto - tramo\n",
    "    if delta < 0:\n",
    "        break\n",
    "    impuesto += delta*rate"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[786384, 1747440, 2912400, 4077360, 5242320, 6989760, 18056880, 99999999]"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "list(TRAMOS.keys())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "def descomponer_en_tramos(sueldo_bruto, tramos=TRAMOS):\n",
    "    descomp = []\n",
    "    impuestos = []\n",
    "    tramo_anterior = 0\n",
    "    for tramo, descuento in tramos.items():\n",
    "        delta = min(sueldo_bruto, tramo) - tramo_anterior\n",
    "        if delta>0:\n",
    "            descomp.append(delta)\n",
    "            impuestos.append(int(delta*descuento))\n",
    "        tramo_anterior = tramo\n",
    "    return descomp, impuestos"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "([786384, 413616], [0, 16544])"
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "sueldo_bruto  = 1_200_000\n",
    "descomponer_en_tramos(sueldo_bruto, TRAMOS)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [],
   "source": [
    "def get_table(sueldo_bruto, tramos=TRAMOS):\n",
    "    _tramos = [0]+list(tramos.keys())\n",
    "    tasas = tramos.values()\n",
    "    data = [[desde, hasta, monto, tasa, impuesto] for  desde, hasta, tasa, monto, impuesto in zip(_tramos[:-1], _tramos[1:], tasas, *descomponer_en_tramos(sueldo_bruto))]\n",
    "    df = pd.DataFrame(data=data, columns=[\"Desde\", \"Hasta\", \"Monto\", \"Tasa\", \"Impuesto\"])\n",
    "    return df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Desde</th>\n",
       "      <th>Hasta</th>\n",
       "      <th>Monto</th>\n",
       "      <th>Tasa</th>\n",
       "      <th>Impuesto</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>786384</td>\n",
       "      <td>786384</td>\n",
       "      <td>0.000</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>786384</td>\n",
       "      <td>1747440</td>\n",
       "      <td>961056</td>\n",
       "      <td>0.040</td>\n",
       "      <td>38442</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1747440</td>\n",
       "      <td>2912400</td>\n",
       "      <td>1164960</td>\n",
       "      <td>0.080</td>\n",
       "      <td>93196</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>2912400</td>\n",
       "      <td>4077360</td>\n",
       "      <td>1164960</td>\n",
       "      <td>0.135</td>\n",
       "      <td>157269</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>4077360</td>\n",
       "      <td>5242320</td>\n",
       "      <td>1164960</td>\n",
       "      <td>0.230</td>\n",
       "      <td>267940</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>5242320</td>\n",
       "      <td>6989760</td>\n",
       "      <td>1747440</td>\n",
       "      <td>0.304</td>\n",
       "      <td>531221</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>6989760</td>\n",
       "      <td>18056880</td>\n",
       "      <td>3010240</td>\n",
       "      <td>0.350</td>\n",
       "      <td>1053584</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     Desde     Hasta    Monto   Tasa  Impuesto\n",
       "0        0    786384   786384  0.000         0\n",
       "1   786384   1747440   961056  0.040     38442\n",
       "2  1747440   2912400  1164960  0.080     93196\n",
       "3  2912400   4077360  1164960  0.135    157269\n",
       "4  4077360   5242320  1164960  0.230    267940\n",
       "5  5242320   6989760  1747440  0.304    531221\n",
       "6  6989760  18056880  3010240  0.350   1053584"
      ]
     },
     "execution_count": 22,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "get_table(10000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3.10.2 ('st')",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.2"
  },
  "orig_nbformat": 4,
  "vscode": {
   "interpreter": {
    "hash": "885b610356171de8141a812cff200370cfd21ab3bf3d712138f61360bb82c1dd"
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}