Merge pull request #3201 from lonvia/tweak-expected-count

[nominatim.git] / nominatim / api / status.py

diff --git a/nominatim/api/status.py b/nominatim/api/status.py
index a992460cc0952a29d2f2066ddc58074d192b15c8..61e36cc36488c097f58dd77d604b3975879500b4 100644 (file)
--- a/nominatim/api/status.py
+++ b/nominatim/api/status.py
@@ -7,58 +7,40 @@
 """
 Classes and function releated to status call.
 """
 """
 Classes and function releated to status call.
 """

-from typing import Optional, cast
+from typing import Optional

 import datetime as dt
 import datetime as dt

+import dataclasses

 
 import sqlalchemy as sa
 
 import sqlalchemy as sa

-import asyncpg

 
 from nominatim.api.connection import SearchConnection
 from nominatim import version
 
 
 from nominatim.api.connection import SearchConnection
 from nominatim import version
 

+@dataclasses.dataclass

 class StatusResult:
     """ Result of a call to the status API.
     """
 class StatusResult:
     """ Result of a call to the status API.
     """

-
-    def __init__(self, status: int, msg: str):
-        self.status = status
-        self.message = msg
-        self.software_version = version.NOMINATIM_VERSION
-        self.data_updated: Optional[dt.datetime]  = None
-        self.database_version: Optional[version.NominatimVersion] = None
-
-
-async def _get_database_date(conn: SearchConnection) -> Optional[dt.datetime]:
-    """ Query the database date.
-    """
-    sql = sa.text('SELECT lastimportdate FROM import_status LIMIT 1')
-    result = await conn.execute(sql)
-
-    for row in result:
-        return cast(dt.datetime, row[0])
-
-    return None
-
-
-async def _get_database_version(conn: SearchConnection) -> Optional[version.NominatimVersion]:
-    sql = sa.text("""SELECT value FROM nominatim_properties
-                     WHERE property = 'database_version'""")
-    result = await conn.execute(sql)
-
-    for row in result:
-        return version.parse_version(cast(str, row[0]))
-
-    return None
+    status: int
+    message: str
+    software_version = version.NOMINATIM_VERSION
+    data_updated: Optional[dt.datetime] = None
+    database_version: Optional[version.NominatimVersion] = None

 
 
 async def get_status(conn: SearchConnection) -> StatusResult:
     """ Execute a status API call.
     """
     status = StatusResult(0, 'OK')
 
 
 async def get_status(conn: SearchConnection) -> StatusResult:
     """ Execute a status API call.
     """
     status = StatusResult(0, 'OK')

+
+    # Last update date
+    sql = sa.select(conn.t.import_status.c.lastimportdate).limit(1)
+    status.data_updated = await conn.scalar(sql)
+
+    # Database version

     try:
     try:

-        status.data_updated = await _get_database_date(conn)
-        status.database_version = await _get_database_version(conn)
-    except asyncpg.PostgresError:
-        return StatusResult(700, 'Database connection failed')
+        verstr = await conn.get_property('database_version')
+        status.database_version = version.parse_version(verstr)
+    except ValueError:
+        pass

 
     return status
 
     return status