alias

Author: liquidcarbon

qstrings/Q.py

@@ -68,6 +68,7 @@ def __new__(
         qstr.id = int(f"{datetime.now():%y%m%d%H%M%S%f}")
         qstr.refs = refs  # references used to create the Q string
         qstr.file = _path if file else None
+        qstr.alias = kwargs.get("alias")
         qstr.exec_id = 0
         qstr.duration = 0.0
         qstr._quiet = kwargs.get("quiet", False)

tests/test_Q.py

@@ -42,6 +42,11 @@ def test_from_file():
     assert q.file and q.file.name == "test_format.sql"
 
 
+def test_alias():
+    q = Q("SELECT 42 AS answer", alias="blah")
+    assert q.alias == "blah"
+
+
 def test_parse_error():
     q = Q("SELE 42")
     assert q.ast_errors