read_bodies.py

# -*- coding: utf-8 -*-
"""
Created on Mon Jan 29 10:30:53 2024

@author: miguelangel
"""

import pathlib

def leer_parámetros(fichero):
    # Diccionario para almacenar los valores
    valores = {}
    # Palabras clave a buscar
    palabras_clave = ['Mass', 'Size', 'SidRotPeriod']
    # Abrir el archivo en modo lectura
    with open(fichero, 'r') as archivo:
        # Iterar sobre cada línea en el archivo
        for linea in archivo:
            # Dividir la línea en palabras
            palabras = linea.split()
            # Verificar si la palabra clave está en la línea
            for palabra in palabras_clave:
                if palabra in palabras:
                    # Encontrar el índice de la palabra clave
                    indice = palabras.index(palabra)
                    # Obtener el valor numérico que sigue a la palabra clave
                    valor = palabras[indice + 2]
                    # Almacenar el valor en el diccionario
                    valores[palabra.lower()] = float(valor)
    return valores


#Lista para guardar los resultados
bodies_data = ['Planet name\tMass (kg)\tRadius (m)\tPeriod (s)']
#Lista con todos los ficheros del directorio:
listaArchivos = [f for f in pathlib.Path().glob("./bodies/todos/*.cfg")]

for archivo in listaArchivos:
    resultados = leer_parámetros(archivo)
    # Obtener los valores de las variables
    nombre = archivo.name.split('.')[0]
    masa = resultados.get('mass', None)
    radio = resultados.get('size', None)
    periodo = resultados.get('sidrotperiod', None)
    # Imprimir los resultados
    #print(f'{nombre}\t-> Masa: {masa} \t Radio: {radio} \t Periodo: {periodo}')
    línea = '\t'.join(list(map(str,[nombre, masa, radio, periodo])))
    bodies_data.append(línea)


# lo volcamos en un archivo de texto
with open('bodies_data.cfg', "w") as f:
    for l in bodies_data:
       f.write(l + '\n')