From 0226b76e0ae40ac8596f2ae1470d60b62faec8a3 Mon Sep 17 00:00:00 2001 From: bastshoes Date: Tue, 22 Oct 2019 18:39:26 +0300 Subject: [PATCH] Add support SQL VACUUM for PostgeSQL (#28106) * Add support SQL VACUUM for PostgeSQL VACUUM PostgreSQL DB if repack is true * Update tests --- homeassistant/components/recorder/purge.py | 4 ++-- tests/components/recorder/test_purge.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/recorder/purge.py b/homeassistant/components/recorder/purge.py index 2ac0b38c694..089476245fe 100644 --- a/homeassistant/components/recorder/purge.py +++ b/homeassistant/components/recorder/purge.py @@ -34,8 +34,8 @@ def purge_old_data(instance, purge_days, repack): _LOGGER.debug("Deleted %s events", deleted_rows) # Execute sqlite vacuum command to free up space on disk - if repack and instance.engine.driver == "pysqlite": - _LOGGER.debug("Vacuuming SQLite to free space") + if repack and instance.engine.driver in ("pysqlite", "postgresql"): + _LOGGER.debug("Vacuuming SQL DB to free space") instance.engine.execute("VACUUM") except SQLAlchemyError as err: diff --git a/tests/components/recorder/test_purge.py b/tests/components/recorder/test_purge.py index 1c676e203d2..7e06dcd1e5e 100644 --- a/tests/components/recorder/test_purge.py +++ b/tests/components/recorder/test_purge.py @@ -174,5 +174,5 @@ class TestRecorderPurge(unittest.TestCase): self.hass.data[DATA_INSTANCE].block_till_done() assert ( mock_logger.debug.mock_calls[3][1][0] - == "Vacuuming SQLite to free space" + == "Vacuuming SQL DB to free space" )