Add OpenSearch refresh CLI command

ElliottKasoar · Mar 13, 2024 · b119b3c · b119b3c
1 parent 63a49b6
commit b119b3c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/abcd/backends/atoms_opensearch.py b/abcd/backends/atoms_opensearch.py
@@ -330,7 +330,7 @@ def info(self):
         else:
             host, port = None, None
 
-        self.client.indices.refresh(index=self.index_name)
+        self.refresh()
         return {
             "host": host,
             "port": port,

diff --git a/abcd/frontends/commandline/commands.py b/abcd/frontends/commandline/commands.py
@@ -269,6 +269,12 @@ def server(*, abcd_url, url, api_only, **kwargs):
     app.run(host=o.hostname, port=o.port)
 
 
+@init_config
+@init_db
+def refresh(*, db, **kwargs):
+    db.refresh()
+
+
 class Formater(object):
     partialBlocks = ["▏", "▎", "▍", "▌", "▋", "▊", "▉", "█"]  # char=pb
 

diff --git a/abcd/frontends/commandline/parser.py b/abcd/frontends/commandline/parser.py
@@ -203,6 +203,8 @@
     "-u", "--url", help="Url to run the server.", default="http://localhost:5000"
 )
 
+refresh_parser = subparsers.add_parser("refresh", help="refresh database")
+refresh_parser.set_defaults(callback_func=commands.refresh)
 
 def main(args=None):
     kwargs = parser.parse_args(args).__dict__