Field name

Author: brunovcosta

json_sql/apply.py

@@ -1,5 +1,6 @@
 from typing import List, Dict, Optional
 from .tables import ITablesSnapshot
+from .field_name import field_name
 from .ast import (
     Expression,
     StringExpression,
@@ -195,7 +196,9 @@ def apply_limit(limit: Limit, data: List[dict], ctx: dict):
 def apply_select_fields(fields: List[SelectField], data: List[dict], ctx: dict):
     return [
         {
-            field.alias: apply_expression(field.expression, {**ctx, **row})
+            field_name(field) or field.expression: apply_expression(
+                field.expression, {**ctx, **row}
+            )
             for field in fields
         }
         for row in data

json_sql/cli.py

@@ -1,8 +1,10 @@
-from typing import List
 from pathlib import Path
-from json_sql.tables import FileSystemTables, Table
-from json import loads
-from eval import eval_sql
+from .tables import FileSystemTables
+from .eval import eval_sql
+from argparse import ArgumentParser
+from csv import DictWriter
+from sys import stdout
+from json import dumps
 
 
 def query(code: str, workdir: Path, ctx: dict):
@@ -11,4 +13,40 @@ def query(code: str, workdir: Path, ctx: dict):
     return result
 
 
-def help(): ...
+if __name__ == "__main__":
+    parser = ArgumentParser(description="Run SQL queries on JSON files.")
+    parser.add_argument("code", type=str, help="SQL query to execute")
+    parser.add_argument(
+        "--workdir",
+        type=Path,
+        default=Path.cwd(),
+        help="Directory containing JSON files",
+    )
+    parser.add_argument(
+        "--ctx",
+        type=str,
+        default="{}",
+        help="Context for the query (default: empty dictionary)",
+    )
+    parser.add_argument(
+        "--format",
+        type=str,
+        default="json",
+        choices=["json", "csv"],
+        help="Output format (default: json)",
+    )
+    args = parser.parse_args()
+
+    result = query(code=args.code, workdir=args.workdir, ctx={})
+
+    if args.format == "json":
+        print(dumps(result, indent=2))
+    elif args.format == "csv":
+        if len(result) == 0:
+            print("No results to write to CSV.")
+        else:
+            keys = result[0].keys()
+            writer = DictWriter(stdout, fieldnames=keys)
+            writer.writeheader()
+            for row in result:
+                writer.writerow(row)

json_sql/field_name.py

@@ -0,0 +1,13 @@
+from .ast import SelectField, NameExpression
+
+
+def field_name(field: SelectField) -> str:
+    """
+    Get the field name from a SelectField object.
+    """
+    if field.alias:
+        return field.alias
+    elif isinstance(field.expression, NameExpression):
+        return field.expression.name
+    else:
+        return "?column?"