def main():
args = parser.parse_args()
client = Client(server_url=args.url, ca_crt=args.ca_crt)
for sql in args.sql:
print("executing: {}".format(sql))
print(client.execute(sql))
class SqlFlowMagic(Magics):
"""
Provides the `%%sqlflow` magic
"""
def __init__(self, shell):
super(SqlFlowMagic, self).__init__(shell)
self.client = Client()
@cell_magic('sqlflow')
def execute(self, line, cell):
"""Runs SQL statement
:param line: The line magic
:type line: str.
:param cell: The cell magic
:type cell: str.
Example:
>>> %%sqlflow SELECT *
... FROM mytable
>>> %%sqlflow SELECT *
... FROM iris.iris limit 1
... TRAIN DNNClassifier
... WITH
... n_classes = 3,
... hidden_units = [10, 10]
... COLUMN sepal_length, sepal_width, petal_length, petal_width
... LABEL class
... INTO my_dnn_model;
"""
return self.client.execute('\n'.join([line, cell]))
def main():
args = parser.parse_args()
client = Client(server_url=args.url)
for sql in args.sql:
print("executing: {}".format(sql))
for res in client.execute(sql):
print(res)