feat(db): save aruco flag in db

Author: matthiasschaub

sketch_map_tool/database/client_celery.py

@@ -39,6 +39,7 @@ def insert_map_frame(
     format_: PaperFormat,
     orientation: str,
     layer: Layer,
+    aruco: bool,
 ):
     """Insert map frame alongside map generation parameters into the database.
 
@@ -56,6 +57,7 @@ def insert_map_frame(
             orientation VARCHAR,
             layer VARCHAR,
             version VARCHAR,
+            aruco BOOLEAN DEFAULT FALSE,
             ts TIMESTAMP WITH TIME ZONE DEFAULT now()
             )
     """
@@ -69,6 +71,7 @@ def insert_map_frame(
             format,
             orientation,
             layer,
+            aruco,
             version
             )
         VALUES (
@@ -80,6 +83,7 @@ def insert_map_frame(
             %s,
             %s,
             %s,
+            %s,
             %s)
     """
     with db_conn.cursor() as curs:
@@ -95,6 +99,7 @@ def insert_map_frame(
                 str(format_),
                 orientation,
                 layer,
+                aruco,
                 __version__,
             ),
         )

sketch_map_tool/tasks.py

@@ -120,7 +120,15 @@ def generate_sketch_map(
         layer,
         aruco,
     )
-    db_client_celery.insert_map_frame(map_img, uuid, bbox, format_, orientation, layer)
+    db_client_celery.insert_map_frame(
+        map_img,
+        uuid,
+        bbox,
+        format_,
+        orientation,
+        layer,
+        aruco,
+    )
     return map_pdf
 
 

tests/fixtures/cassette/test_as_image

@@ -20,6 +20,7 @@ def test_generate_sketch_map(bbox, format_, size, scale, layer):
             size,
             scale,
             layer,
+            False,
         )
     )
     result = task.get(timeout=180)

tests/fixtures/cassette/test_generate_sketch_map

@@ -81,9 +81,18 @@ def test_close_open_connection():
     client_celery.open_connection()
 
 
-def test_write_map_frame(flask_app, map_frame, bbox, format_, orientation, layer):
+def test_write_map_frame(
+    flask_app,
+    map_frame,
+    bbox,
+    format_,
+    orientation,
+    layer,
+):
     uuid = uuid4()
-    client_celery.insert_map_frame(map_frame, uuid, bbox, format_, orientation, layer)
+    client_celery.insert_map_frame(
+        map_frame, uuid, bbox, format_, orientation, layer, aruco=False
+    )
     with flask_app.app_context():
         file = client_flask.select_map_frame(uuid)
         assert isinstance(file, bytes)