feat: adicionar o delete_table_row ao menu

This commit is contained in:
aulojor
2025-11-09 20:44:38 -01:00
parent a490bc7756
commit d83a953a12
2 changed files with 10 additions and 9 deletions

View File

@@ -104,7 +104,13 @@ def main():
retInfo = "ID do event não encontrado!" retInfo = "ID do event não encontrado!"
else: else:
db = crud.delete_event(db, eid_choice) table = crud.get_table(db, eid_choice)
crud.show_table(table)
row_choice = _get_usr_input("Escolhe a linha a apagar:", int)
db = crud.delete_table_row(db, eid_choice, row_choice)
new_table = crud.get_table(db, eid_choice)
crud.show_table(new_table)
print(f"Linha {row_choice} apagada com sucesso!")
input() input()
else: else:

View File

@@ -8,7 +8,7 @@ pd.set_option('display.max_columns', 500)
pd.set_option('display.width', 150) pd.set_option('display.width', 150)
HEADER_COLS = ["Data", "Distancia", "Tipo Ev", "Lat", "Long", "Prof", "Magnitudes"] HEADER_COLS = ["Data", "Distancia", "Tipo Ev", "Lat", "Long", "Prof", "Magnitudes"]
TABLE_READ_RET = ["Data", "Lat", "Long", "Distancia", "Tipo Ev"] TABLE_READ_RET = ["Data", "Lat", "Long", "Distancia", "Tipo Ev", "Amplitude"]
def _get_uniques(df) -> pd.DataFrame: def _get_uniques(df) -> pd.DataFrame:
return df.get(["ID", "Data", "Regiao"]).drop_duplicates(subset="ID", keep="first") return df.get(["ID", "Data", "Regiao"]).drop_duplicates(subset="ID", keep="first")
@@ -40,7 +40,6 @@ def read_header(df, event_id):
def show_table(df, retCols=TABLE_READ_RET): def show_table(df, retCols=TABLE_READ_RET):
print(df.loc[:,retCols]) print(df.loc[:,retCols])
def get_table(df, event_id): def get_table(df, event_id):
rows = df[df["ID"] == event_id] rows = df[df["ID"] == event_id]
rows = rows.drop("ID", axis=1) rows = rows.drop("ID", axis=1)
@@ -86,13 +85,9 @@ def delete_event(df, event_id):
print(f"Evento {event_id} apagado!") print(f"Evento {event_id} apagado!")
return new_df return new_df
def delete_table_row(df, event_id, row_number_1): def delete_table_row(df, event_id, row_number):
# Apaga uma linha específica da tabela do evento # Apaga uma linha específica da tabela do evento
row_number_0 = row_number_1 - 1 new_df = df.drop([row_number]).reset_index(drop=True)
table = get_table(df, event_id)
if row_number_0 < 0 or row_number_0 >= len(table):
return f"Linha {row_number_1} não pertence ao evento {event_id}."
new_df = df.drop(table.index[row_number_0])
return new_df return new_df
def create_blank_event(df, event_id): def create_blank_event(df, event_id):