37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
|
|
import os
|
|
import sys
|
|
import psycopg2
|
|
from pathlib import Path
|
|
from dotenv import load_dotenv
|
|
|
|
# Load .env
|
|
ROOT_DIR = Path(__file__).resolve().parent.parent.parent
|
|
load_dotenv(ROOT_DIR / ".env")
|
|
|
|
def get_db_connection():
|
|
db_host = os.getenv("DB_HOST", "192.168.3.195")
|
|
db_user = os.getenv("DB_USER", "value")
|
|
db_pass = os.getenv("DB_PASSWORD", "Value609!")
|
|
db_name = os.getenv("DB_NAME", "fa3")
|
|
db_port = os.getenv("DB_PORT", "5432")
|
|
|
|
return psycopg2.connect(
|
|
host=db_host, user=db_user, password=db_pass, dbname=db_name, port=db_port
|
|
)
|
|
|
|
def drop_table(table_name):
|
|
print(f"Dropping table {table_name}...")
|
|
try:
|
|
conn = get_db_connection()
|
|
conn.autocommit = True
|
|
with conn.cursor() as cur:
|
|
cur.execute(f"DROP TABLE IF EXISTS {table_name}")
|
|
print(f"✅ Dropped {table_name}")
|
|
conn.close()
|
|
except Exception as e:
|
|
print(f"❌ Error dropping {table_name}: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
drop_table("ifind_int_daily_basic")
|