X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/c3788d765ed4e7ddf00794085af757714bc102cf..17bbe2637a32e93b19ff887a896bf1b444cd9792:/nominatim/db/properties.py diff --git a/nominatim/db/properties.py b/nominatim/db/properties.py index 19c09006..3624c950 100644 --- a/nominatim/db/properties.py +++ b/nominatim/db/properties.py @@ -7,8 +7,11 @@ """ Query and access functions for the in-database property table. """ +from typing import Optional, cast -def set_property(conn, name, value): +from nominatim.db.connection import Connection + +def set_property(conn: Connection, name: str, value: str) -> None: """ Add or replace the propery with the given name. """ with conn.cursor() as cur: @@ -23,12 +26,19 @@ def set_property(conn, name, value): cur.execute(sql, (value, name)) conn.commit() -def get_property(conn, name): - """ Return the current value of the given propery or None if the property + +def get_property(conn: Connection, name: str) -> Optional[str]: + """ Return the current value of the given property or None if the property is not set. """ + if not conn.table_exists('nominatim_properties'): + return None + with conn.cursor() as cur: cur.execute('SELECT value FROM nominatim_properties WHERE property = %s', (name, )) - return cur.fetchone()[0] if cur.rowcount > 0 else None + if cur.rowcount == 0: + return None + + return cast(Optional[str], cur.fetchone()[0]) # type: ignore[no-untyped-call]