Merge pull request #3 from aulojor/main
feat: adicionar o delete_table_row ao menu
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
@@ -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):
|
||||||
|
|||||||
Reference in New Issue
Block a user