19 lines
645 B
Python
19 lines
645 B
Python
from src.services.postgresql_manager import PostgreSQLManager
|
|
from src.utils.config import config
|
|
from src.models.activity import Activity
|
|
from sqlalchemy import text
|
|
|
|
def backfill():
|
|
print("Backfilling is_estimated_power defaults...")
|
|
db_manager = PostgreSQLManager(config.DATABASE_URL)
|
|
with db_manager.get_db_session() as session:
|
|
# Update all NULLs to False
|
|
result = session.execute(
|
|
text("UPDATE activities SET is_estimated_power = false WHERE is_estimated_power IS NULL")
|
|
)
|
|
session.commit()
|
|
print(f"Updated {result.rowcount} rows.")
|
|
|
|
if __name__ == "__main__":
|
|
backfill()
|