Measure time for query execution

xql/main.py

@@ -15,7 +15,11 @@
 
 if __name__ == '__main__':
     try:
-        main()
+        while True:
+            query = input("xql> ")
+            if query == ".exit":
+                break
+            main(query)
     except ImportError as e:
         raise ImportError('main function is not imported please try again.') from e
 

xql/src/xql/apply.py

@@ -23,6 +23,7 @@
 from sqlglot import parse_one, exp
 from xarray.core.groupby import DatasetGroupBy
 
+from .utils import timing
 from .where import apply_where
 
 command_info = {
@@ -443,31 +444,24 @@ def run_query(query: str) -> None:
     result = parse_query(query)
     display_result(result)
 
-
-def main():
+@timing
+def main(query: str):
     """
     Main function for runnning this file.
     """
-    while True:
-
-        query = input("xql> ")
-
-        if query == ".exit":
-            break
+    if ".help" in query:
+        display_help(query)
 
-        elif ".help" in query:
-            display_help(query)
+    elif ".set" in query:
+        set_dataset_table(query)
 
-        elif ".set" in query:
-            set_dataset_table(query)
+    elif ".show" in query:
+        display_table_dataset_map(query)
 
-        elif ".show" in query:
-            display_table_dataset_map(query)
-
-        else:
-            try:
-                result = parse_query(query)
-            except Exception as e:
-                result = f"ERROR: {type(e).__name__}: {e.__str__()}."
+    else:
+        try:
+            result = parse_query(query)
+        except Exception as e:
+            result = f"ERROR: {type(e).__name__}: {e.__str__()}."
 
-            display_result(result)
+        display_result(result)

xql/src/xql/utils.py

@@ -0,0 +1,13 @@
+from functools import wraps
+from time import gmtime, strftime, time
+
+def timing(f):
+    """Measure a time for any function execution."""
+    @wraps(f)
+    def wrap(*args, **kw):
+        ts = time()
+        result = f(*args, **kw)
+        te = time()
+        print(f"Query took: { strftime('%H:%M:%S', gmtime(te - ts)) }")
+        return result
+    return wrap