Ankit Malik
31 lines
711 B
Python
31 lines
711 B
Python
from db_con.connection import *
|
|
|
|
from log import *
|
|
|
|
|
|
def truncate_table(client, table_name: str) -> None:
|
|
"""
|
|
Truncate a ClickHouse table.
|
|
"""
|
|
query = f"TRUNCATE TABLE {table_name}"
|
|
|
|
print(f"Truncating table: {table_name}")
|
|
client.command(query)
|
|
|
|
log.info(f"Table {table_name} truncated successfully.")
|
|
|
|
|
|
|
|
def delete_rows(client, table_name: str, condition: str) -> None:
|
|
"""
|
|
Delete rows from a ClickHouse table based on a condition.
|
|
"""
|
|
query = f"""
|
|
ALTER TABLE {table_name}
|
|
DELETE WHERE {condition}
|
|
"""
|
|
|
|
print(f"Deleting rows from {table_name} where {condition}")
|
|
client.command(query)
|
|
|
|
log.info("Delete command submitted successfully.") |