turn query structure into sql text

llama_test.ipynb

@@ -160,7 +160,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [
     {
@@ -172,27 +172,46 @@
       "1-1000181-1\n",
       "['State/territory', 'Text/background colour', 'Format', 'Current slogan', 'Current series', 'Notes']\n",
       "\n",
-      "{'id': '1-1000181-1', 'header': ['State/territory', 'Text/background colour', 'Format', 'Current slogan', 'Current series', 'Notes'], 'types': ['text', 'text', 'text', 'text', 'text', 'text'], 'rows': [['Australian Capital Territory', 'blue/white', 'Yaa·nna', 'ACT · CELEBRATION OF A CENTURY 2013', 'YIL·00A', 'Slogan screenprinted on plate'], ['New South Wales', 'black/yellow', 'aa·nn·aa', 'NEW SOUTH WALES', 'BX·99·HI', 'No slogan on current series'], ['New South Wales', 'black/white', 'aaa·nna', 'NSW', 'CPX·12A', 'Optional white slimline series'], ['Northern Territory', 'ochre/white', 'Ca·nn·aa', 'NT · OUTBACK AUSTRALIA', 'CB·06·ZZ', 'New series began in June 2011'], ['Queensland', 'maroon/white', 'nnn·aaa', 'QUEENSLAND · SUNSHINE STATE', '999·TLG', 'Slogan embossed on plate'], ['South Australia', 'black/white', 'Snnn·aaa', 'SOUTH AUSTRALIA', 'S000·AZD', 'No slogan on current series'], ['Victoria', 'blue/white', 'aaa·nnn', 'VICTORIA - THE PLACE TO BE', 'ZZZ·562', 'Current series will be exhausted this year']], 'name': 'table_1000181_1'}\n"
+      "{'id': '1-1000181-1', 'header': ['State/territory', 'Text/background colour', 'Format', 'Current slogan', 'Current series', 'Notes'], 'types': ['text', 'text', 'text', 'text', 'text', 'text'], 'rows': [['Australian Capital Territory', 'blue/white', 'Yaa·nna', 'ACT · CELEBRATION OF A CENTURY 2013', 'YIL·00A', 'Slogan screenprinted on plate'], ['New South Wales', 'black/yellow', 'aa·nn·aa', 'NEW SOUTH WALES', 'BX·99·HI', 'No slogan on current series'], ['New South Wales', 'black/white', 'aaa·nna', 'NSW', 'CPX·12A', 'Optional white slimline series'], ['Northern Territory', 'ochre/white', 'Ca·nn·aa', 'NT · OUTBACK AUSTRALIA', 'CB·06·ZZ', 'New series began in June 2011'], ['Queensland', 'maroon/white', 'nnn·aaa', 'QUEENSLAND · SUNSHINE STATE', '999·TLG', 'Slogan embossed on plate'], ['South Australia', 'black/white', 'Snnn·aaa', 'SOUTH AUSTRALIA', 'S000·AZD', 'No slogan on current series'], ['Victoria', 'blue/white', 'aaa·nnn', 'VICTORIA - THE PLACE TO BE', 'ZZZ·562', 'Current series will be exhausted this year']], 'name': 'table_1000181_1'}\n",
+      "SELECT  col5 FROM table WHERE col3 = SOUTH AUSTRALIA\n",
+      "SELECT  Notes FROM table WHERE Current slogan = SOUTH AUSTRALIA\n"
      ]
     }
    ],
    "source": [
     "import json\n",
+    "import sys\n",
+    "import re\n",
+    "sys.path.append('./WikiSQL')\n",
+    "from WikiSQL.lib.query import Query\n",
+    "\n",
+    "def replace_cols(qs,cmap):\n",
+    "    for k,v in cmap.items():\n",
+    "        qs = re.sub(f' {k} ',f' {v} ',qs)\n",
+    "    return qs\n",
+    "\n",
     "with open('data/train.jsonl') as f:\n",
     "    j = f.readline()\n",
     "    js = json.loads(j)\n",
     "    print(js)\n",
-    "\n",
+    "    \n",
     "print()\n",
     "\n",
     "with open('data/train.tables.jsonl') as f:\n",
     "    j = f.readline()\n",
-    "    js = json.loads(j)\n",
+    "    js2 = json.loads(j)\n",
     "    #print(js)\n",
-    "    print(js['id'])\n",
-    "    print(js['header'])\n",
+    "    print(js2['id'])\n",
+    "    print(js2['header'])\n",
     "    print()\n",
-    "    print(js)"
+    "    print(js2)\n",
+    "\n",
+    "sql = js['sql']\n",
+    "q = Query.from_dict(sql)\n",
+    "print(q)\n",
+    "cm = {f'col{i}':js2['header'][i] for i in range(len(js2['header']))}\n",
+    "qs = replace_cols(str(q),cm)\n",
+    "print(qs)"
    ]
   }
  ],