parent
c46ce578d4
commit
ae51d8c622
@ -1,144 +1,144 @@
|
||||
from concurrent.futures import thread
|
||||
import re
|
||||
import time
|
||||
import aquery_parser as parser
|
||||
import engine
|
||||
import subprocess
|
||||
import mmap
|
||||
import sys
|
||||
import os
|
||||
from engine.utils import base62uuid
|
||||
import atexit
|
||||
try:
|
||||
os.remove('server.bin')
|
||||
except Exception as e:
|
||||
print(type(e), e)
|
||||
|
||||
subprocess.call(['make', 'server.bin'])
|
||||
cleanup = True
|
||||
def rm():
|
||||
if cleanup:
|
||||
mm.seek(0,os.SEEK_SET)
|
||||
mm.write(b'\x00\x00')
|
||||
mm.flush()
|
||||
time.sleep(.001)
|
||||
server.kill()
|
||||
files = os.listdir('.')
|
||||
for f in files:
|
||||
if f.endswith('.shm'):
|
||||
os.remove(f)
|
||||
mm.close()
|
||||
cleanup = False
|
||||
|
||||
def proc_alive(pid):
|
||||
try:
|
||||
os.kill(pid, 0)
|
||||
except OSError:
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
|
||||
atexit.register(rm)
|
||||
|
||||
shm = base62uuid()
|
||||
if sys.platform != 'win32':
|
||||
import readline
|
||||
shm += '.shm'
|
||||
basecmd = ['bash', '-c', 'rlwrap k']
|
||||
mm = None
|
||||
if not os.path.isfile(shm):
|
||||
# create initial file
|
||||
with open(shm, "w+b") as handle:
|
||||
handle.write(b'\x01\x00') # [running, new job]
|
||||
handle.flush()
|
||||
mm = mmap.mmap(handle.fileno(), 2, access=mmap.ACCESS_WRITE, offset=0)
|
||||
if mm is None:
|
||||
exit(1)
|
||||
else:
|
||||
basecmd = ['bash.exe', '-c', 'rlwrap ./k']
|
||||
mm = mmap.mmap(0, 2, shm)
|
||||
mm.write(b'\x01\x00')
|
||||
mm.flush()
|
||||
server = subprocess.Popen(["./server.bin", shm])
|
||||
|
||||
test_parser = True
|
||||
|
||||
# code to test parser
|
||||
ws = re.compile(r'\s+')
|
||||
|
||||
q = 'SELECT p.Name, v.Name FROM Production.Product p JOIN Purchasing.ProductVendor pv ON p.ProductID = pv.ProductID JOIN Purchasing.Vendor v ON pv.BusinessEntityID = v.BusinessEntityID WHERE ProductSubcategoryID = 15 ORDER BY v.Name;'
|
||||
|
||||
res = parser.parse(q)
|
||||
|
||||
print(res)
|
||||
|
||||
|
||||
# else:f
|
||||
# if subprocess.call(['make', 'snippet']) == 0:
|
||||
# mm.seek(0)
|
||||
# mm.write(b'\x01\x01')
|
||||
# time.sleep(.1)
|
||||
# mm.seek(0)
|
||||
# print(mm.read(2))
|
||||
|
||||
# mm.close()
|
||||
# handle.close()
|
||||
# os.remove(shm)
|
||||
# exit()
|
||||
keep = False
|
||||
cxt = None
|
||||
while test_parser:
|
||||
try:
|
||||
if not proc_alive(server.pid):
|
||||
server = subprocess.Popen(["./server.bin", shm])
|
||||
q = input()
|
||||
if q == 'exec':
|
||||
if not keep or cxt is None:
|
||||
cxt = engine.initialize()
|
||||
stmts_stmts = stmts['stmts']
|
||||
if type(stmts_stmts) is list:
|
||||
for s in stmts_stmts:
|
||||
engine.generate(s, cxt)
|
||||
else:
|
||||
engine.generate(stmts_stmts, cxt)
|
||||
print(cxt.ccode)
|
||||
with open('out.cpp', 'wb') as outfile:
|
||||
outfile.write((cxt.finalize()).encode('utf-8'))
|
||||
if subprocess.call(['make', 'snippet']) == 0:
|
||||
mm.seek(0,os.SEEK_SET)
|
||||
mm.write(b'\x01\x01')
|
||||
continue
|
||||
elif q == 'k':
|
||||
subprocess.call(basecmd)
|
||||
continue
|
||||
elif q == 'print':
|
||||
print(stmts)
|
||||
continue
|
||||
elif q == 'keep':
|
||||
keep = not keep
|
||||
continue
|
||||
elif q=='format' or q == 'fmt':
|
||||
subprocess.call(['clang-format', 'out.cpp'])
|
||||
elif q == 'exit':
|
||||
break
|
||||
elif q == 'r':
|
||||
if subprocess.call(['make', 'snippet']) == 0:
|
||||
mm.seek(0,os.SEEK_SET)
|
||||
mm.write(b'\x01\x01')
|
||||
continue
|
||||
trimed = ws.sub(' ', q.lower()).split(' ')
|
||||
if trimed[0].startswith('f'):
|
||||
fn = 'stock.a' if len(trimed) <= 1 or len(trimed[1]) == 0 \
|
||||
else trimed[1]
|
||||
|
||||
with open(fn, 'r') as file:
|
||||
contents = file.read()
|
||||
stmts = parser.parse(contents)
|
||||
continue
|
||||
stmts = parser.parse(q)
|
||||
print(stmts)
|
||||
except (ValueError, FileNotFoundError) as e:
|
||||
rm()
|
||||
print(type(e), e)
|
||||
|
||||
from concurrent.futures import thread
|
||||
import re
|
||||
import time
|
||||
import aquery_parser as parser
|
||||
import engine
|
||||
import subprocess
|
||||
import mmap
|
||||
import sys
|
||||
import os
|
||||
from engine.utils import base62uuid
|
||||
import atexit
|
||||
try:
|
||||
os.remove('server.bin')
|
||||
except Exception as e:
|
||||
print(type(e), e)
|
||||
|
||||
subprocess.call(['make', 'server.bin'])
|
||||
cleanup = True
|
||||
|
||||
def rm():
|
||||
global cleanup
|
||||
if cleanup:
|
||||
mm.seek(0,os.SEEK_SET)
|
||||
mm.write(b'\x00\x00')
|
||||
mm.flush()
|
||||
|
||||
try:
|
||||
time.sleep(.001)
|
||||
server.kill()
|
||||
time.sleep(.001)
|
||||
server.terminate()
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
files = os.listdir('.')
|
||||
for f in files:
|
||||
if f.endswith('.shm'):
|
||||
os.remove(f)
|
||||
mm.close()
|
||||
cleanup = False
|
||||
|
||||
atexit.register(rm)
|
||||
|
||||
shm = base62uuid()
|
||||
if sys.platform != 'win32':
|
||||
import readline
|
||||
shm += '.shm'
|
||||
basecmd = ['bash', '-c', 'rlwrap k']
|
||||
mm = None
|
||||
if not os.path.isfile(shm):
|
||||
# create initial file
|
||||
with open(shm, "w+b") as handle:
|
||||
handle.write(b'\x01\x00') # [running, new job]
|
||||
handle.flush()
|
||||
mm = mmap.mmap(handle.fileno(), 2, access=mmap.ACCESS_WRITE, offset=0)
|
||||
if mm is None:
|
||||
exit(1)
|
||||
else:
|
||||
basecmd = ['bash.exe', '-c', 'rlwrap ./k']
|
||||
mm = mmap.mmap(0, 2, shm)
|
||||
mm.write(b'\x01\x00')
|
||||
mm.flush()
|
||||
server = subprocess.Popen(["./server.bin", shm])
|
||||
|
||||
test_parser = True
|
||||
|
||||
# code to test parser
|
||||
ws = re.compile(r'\s+')
|
||||
|
||||
q = 'SELECT p.Name, v.Name FROM Production.Product p JOIN Purchasing.ProductVendor pv ON p.ProductID = pv.ProductID JOIN Purchasing.Vendor v ON pv.BusinessEntityID = v.BusinessEntityID WHERE ProductSubcategoryID = 15 ORDER BY v.Name;'
|
||||
|
||||
res = parser.parse(q)
|
||||
|
||||
print(res)
|
||||
|
||||
# else:f
|
||||
# if subprocess.call(['make', 'snippet']) == 0:
|
||||
# mm.seek(0)
|
||||
# mm.write(b'\x01\x01')
|
||||
# time.sleep(.1)
|
||||
# mm.seek(0)
|
||||
# print(mm.read(2))
|
||||
|
||||
# mm.close()
|
||||
# handle.close()
|
||||
# os.remove(shm)
|
||||
# exit()
|
||||
keep = False
|
||||
cxt = None
|
||||
while test_parser:
|
||||
try:
|
||||
if server.poll() is not None:
|
||||
server = subprocess.Popen(["./server.bin", shm])
|
||||
q = input()
|
||||
if q == 'exec':
|
||||
if not keep or cxt is None:
|
||||
cxt = engine.initialize()
|
||||
stmts_stmts = stmts['stmts']
|
||||
if type(stmts_stmts) is list:
|
||||
for s in stmts_stmts:
|
||||
engine.generate(s, cxt)
|
||||
else:
|
||||
engine.generate(stmts_stmts, cxt)
|
||||
print(cxt.ccode)
|
||||
with open('out.cpp', 'wb') as outfile:
|
||||
outfile.write((cxt.finalize()).encode('utf-8'))
|
||||
if subprocess.call(['make', 'snippet']) == 0:
|
||||
mm.seek(0,os.SEEK_SET)
|
||||
mm.write(b'\x01\x01')
|
||||
continue
|
||||
elif q == 'k':
|
||||
subprocess.call(basecmd)
|
||||
continue
|
||||
elif q == 'print':
|
||||
print(stmts)
|
||||
continue
|
||||
elif q == 'keep':
|
||||
keep = not keep
|
||||
continue
|
||||
elif q=='format' or q == 'fmt':
|
||||
subprocess.call(['clang-format', 'out.cpp'])
|
||||
elif q == 'exit':
|
||||
break
|
||||
elif q == 'r':
|
||||
if subprocess.call(['make', 'snippet']) == 0:
|
||||
mm.seek(0,os.SEEK_SET)
|
||||
mm.write(b'\x01\x01')
|
||||
continue
|
||||
trimed = ws.sub(' ', q.lower()).split(' ')
|
||||
if trimed[0].startswith('f'):
|
||||
fn = 'stock.a' if len(trimed) <= 1 or len(trimed[1]) == 0 \
|
||||
else trimed[1]
|
||||
|
||||
with open(fn, 'r') as file:
|
||||
contents = file.read()
|
||||
stmts = parser.parse(contents)
|
||||
continue
|
||||
stmts = parser.parse(q)
|
||||
print(stmts)
|
||||
except (ValueError, FileNotFoundError) as e:
|
||||
rm()
|
||||
print(type(e), e)
|
||||
|
||||
rm()
|
||||
@ -1,10 +1,10 @@
|
||||
CREATE TABLE test(a INT, b INT, c INT, d INT)
|
||||
|
||||
LOAD DATA INFILE "test.csv"
|
||||
INTO TABLE test
|
||||
FIELDS TERMINATED BY ","
|
||||
|
||||
SELECT sum(c), b, d
|
||||
FROM test
|
||||
group by a,b,d
|
||||
-- order by d DESC, b ASC
|
||||
CREATE TABLE test(a INT, b INT, c INT, d INT)
|
||||
|
||||
LOAD DATA INFILE "test.csv"
|
||||
INTO TABLE test
|
||||
FIELDS TERMINATED BY ","
|
||||
|
||||
SELECT sum(c), b, d
|
||||
FROM test
|
||||
group by a,b,d
|
||||
-- order by d DESC, b ASC
|
||||
|
||||
Loading…
Reference in new issue