universe.py

import os
from os import walk
from constants import STAR_DIRECTORY
import pandas as pd
from rich.table import Table
from starsystem import load_starsystem_yaml


def universe_save():
    list_systems = []
    path_starsystems = os.path.abspath(STAR_DIRECTORY)
    for (dirpath, dirnames, filenames) in walk(path_starsystems):
        print("Collecting filenames")
        for file in filenames:
            system = load_starsystem_yaml(file)
            list_systems.append(system)

    # Convert list of dicts to a dataframe
    system_df = pd.DataFrame(list_systems)

    link = os.path.abspath(STAR_DIRECTORY)+"universe.csv"
    #save to the star directory
    system_df.to_csv(link, index=True)

    return link

def build_universe_table():
    list_systems = []
    path_starsystems = os.path.abspath(STAR_DIRECTORY)
    for (dirpath, dirnames, filenames) in walk(path_starsystems):
        print("Collecting filenames")
        for file in filenames:
            system = load_starsystem_yaml(file)
            list_systems.append(system)

    # Convert list of dicts to a dataframe
    system_df = pd.DataFrame(list_systems)
    # Create a Table object
    table = Table(title="Visited Systems")

    # Add columns from the DataFrame
    for column_name in system_df.columns:
        table.add_column(column_name)

    # Add rows from the DataFrame
    for index, row in system_df.iterrows():
        row = row.str.split(', ', expand=True)
        table.add_row(str(row))  # Unpack row values as arguments

    return table