Merge pull request #3 from datafold/dev

Added main with repl command

Author: erezsh

sqeleton/__main__.py

@@ -0,0 +1,15 @@
+import click
+from .repl import repl as repl_main
+
+@click.group(no_args_is_help=True)
+def cli():
+    pass
+
+@cli.command(no_args_is_help=True)
+@click.argument("database", required=True)
+def repl(database):
+    return repl_main(database)
+
+
+if __name__ == '__main__':
+    cli()

sqeleton/repl.py

@@ -27,8 +27,7 @@ def help():
     rich.print("Otherwise, runs regular SQL query")
 
 
-def main():
-    uri = sys.argv[1]
+def repl(uri):
     db = connect(uri)
     db_name = db.name
 
@@ -60,6 +59,9 @@ def main():
                 if res:
                     print_table(res, [str(i) for i in range(len(res[0]))], q)
 
+def main():
+    uri = sys.argv[1]
+    return repl(uri)
 
 if __name__ == "__main__":
     main()