mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Guard MySQL size calculation returning None (#73331)
This commit is contained in:
parent
2b07082cf6
commit
53b3d2ee87
@ -5,15 +5,18 @@ from sqlalchemy import text
|
|||||||
from sqlalchemy.orm.session import Session
|
from sqlalchemy.orm.session import Session
|
||||||
|
|
||||||
|
|
||||||
def db_size_bytes(session: Session, database_name: str) -> float:
|
def db_size_bytes(session: Session, database_name: str) -> float | None:
|
||||||
"""Get the mysql database size."""
|
"""Get the mysql database size."""
|
||||||
return float(
|
size = session.execute(
|
||||||
session.execute(
|
text(
|
||||||
text(
|
"SELECT ROUND(SUM(DATA_LENGTH + INDEX_LENGTH), 2) "
|
||||||
"SELECT ROUND(SUM(DATA_LENGTH + INDEX_LENGTH), 2) "
|
"FROM information_schema.TABLES WHERE "
|
||||||
"FROM information_schema.TABLES WHERE "
|
"TABLE_SCHEMA=:database_name"
|
||||||
"TABLE_SCHEMA=:database_name"
|
),
|
||||||
),
|
{"database_name": database_name},
|
||||||
{"database_name": database_name},
|
).first()[0]
|
||||||
).first()[0]
|
|
||||||
)
|
if size is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
return float(size)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user