Saltar a contenido

Ejemplo de Documentación Python

"""Impresora de columna de hoja de cálculo

Este script permite al usuario imprimir en la consola todas las columnas del
hoja de cálculo. Se supone que la primera fila de la hoja de cálculo es la
Ubicación de las columnas.

Esta herramienta acepta archivos de valores separados por comas (.csv), así como Excel
(.xls, .xlsx) archivos.

Este script requiere que `pandas` esté instalado dentro de Python
entorno en el que está ejecutando este script.

Este archivo también se puede importar como un módulo y contiene lo siguiente
funciones:

    * get_spreadsheet_cols - devuelve los encabezados de columna del archivo
    * main - la función principal del script
"""

import argparse

import pandas as pd


def get_spreadsheet_cols(file_loc, print_cols=False):

"""Obtiene e imprime las columnas de encabezado de la hoja de cálculo

    Parameters
    ----------
    file_loc : str

        La ubicación del archivo de la hoja de cálculo.
    print_cols : bool, optional
Un indicador utilizado para imprimir las columnas en la consola (el valor predeterminado es False)

    Returns
    -------
    list
        una lista de cadenas utilizadas que son las columnas de encabezado
    """

    file_data = pd.read_excel(file_loc)
    col_headers = list(file_data.columns.values)

    if print_cols:
        print("\n".join(col_headers))

    return col_headers


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument(
        'input_file',
        type=str,
        help="El archivo de hoja de cálculo para imprimir las columnas de"
    )
    args = parser.parse_args()
    get_spreadsheet_cols(args.input_file, print_cols=True)


if __name__ == "__main__":
    main()