mirror of
https://github.com/osm-search/Nominatim.git
synced 2024-11-23 13:44:36 +03:00
8d082c13e0
Some more functions frrom psycopg are now properly annotated. No ignoring necessary anymore.
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
# SPDX-License-Identifier: GPL-2.0-only
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2022 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Query and access functions for the in-database property table.
|
|
"""
|
|
from typing import Optional, cast
|
|
|
|
from nominatim.db.connection import Connection
|
|
|
|
def set_property(conn: Connection, name: str, value: str) -> None:
|
|
""" Add or replace the property with the given name.
|
|
"""
|
|
with conn.cursor() as cur:
|
|
cur.execute('SELECT value FROM nominatim_properties WHERE property = %s',
|
|
(name, ))
|
|
|
|
if cur.rowcount == 0:
|
|
sql = 'INSERT INTO nominatim_properties (value, property) VALUES (%s, %s)'
|
|
else:
|
|
sql = 'UPDATE nominatim_properties SET value = %s WHERE property = %s'
|
|
|
|
cur.execute(sql, (value, name))
|
|
conn.commit()
|
|
|
|
|
|
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, ))
|
|
|
|
if cur.rowcount == 0:
|
|
return None
|
|
|
|
result = cur.fetchone()
|
|
assert result is not None
|
|
|
|
return cast(Optional[str], result[0])
|