feat(wip): prepare to implement commands

Signed-off-by: Gabor Boros <gabor.brs@gmail.com>

Author: gabor-boros

Makefile

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

rethinkdb/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/ast.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/cli/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/cli/_dump.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -20,11 +20,13 @@
 """
 Dump creates an archive of data from a RethinkDB cluster.
 """
+
 import click
 
 
 @click.command
-def cmd_dump():
+@click.pass_context
+def cmd_dump(ctx):
     """
     Dump creates an archive of data from a RethinkDB cluster.
     """

rethinkdb/cli/_export.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -20,11 +20,13 @@
 """
 Export exports data from a RethinkDB cluster into a directory.
 """
+
 import click
 
 
 @click.command
-def cmd_export():
+@click.pass_context
+def cmd_export(ctx):
     """
     Export data from a RethinkDB cluster into a directory.
     """

rethinkdb/cli/_import.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -25,7 +25,8 @@
 
 
 @click.command
-def cmd_import():
+@click.pass_context
+def cmd_import(ctx):
     """
     Import loads data into a RethinkDB cluster.
     """

rethinkdb/cli/_index_rebuild.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -28,7 +28,8 @@
 
 
 @click.command
-def cmd_index_rebuild():
+@click.pass_context
+def cmd_index_rebuild(ctx):
     """
     Rebuild outdated secondary indexes.
     """

rethinkdb/cli/_repl.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -18,24 +18,24 @@
 # Copyright 2010-2016 RethinkDB, all rights reserved.
 
 """
-The main command is a special group that is the root of the command tree.
-
-This command creates a subcommand tree that with the following commands for
-those cases when the rethinkdb binary is not installed:
-    - dump
-    - export
-    - import
-    - index_rebuild
-    - repl
-    - restore
+Start an interactive Python shell (repl) with the RethinkDB driver imported.
 """
 
 import click
 
+BANNER = """
+The RethinkDB driver has been imported as `r`.
+
+A connection to %s:%d has been established as `conn`
+and can be used by calling `run()` on a query without any arguments.
+"""
 
 @click.command
-def cmd_repl():
+@click.pass_context
+def cmd_repl(ctx):
     """
-    Rebuild outdated secondary indexes.
+    Start an interactive Python shell (repl) with the RethinkDB driver imported.
     """
-    click.echo("repl command")
+
+    click.echo(f"repl command: {ctx}")
+

rethinkdb/cli/_restore.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -25,7 +25,8 @@
 
 
 @click.command
-def cmd_restore():
+@click.pass_context
+def cmd_restore(ctx):
     """
     Restore loads data into a RethinkDB cluster from an archive.
     """

rethinkdb/cli/main.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.
@@ -43,7 +43,38 @@
 
 
 @click.group
-def cmd_main():
+@click.option(
+    "--debug",
+    default=False,
+    help="Print debug information.",
+    envvar="RETHINKDB_DEBUG",
+    type=click.BOOL,
+)
+@click.option(
+    "--user",
+    "-u",
+    default="admin",
+    help="The RethinkDB user to connect with.",
+    envvar="RETHINKDB_USER",
+    type=click.STRING,
+)
+@click.option(
+    "--host",
+    "-h",
+    default="localhost",
+    help="The RethinkDB host to connect to.",
+    envvar="RETHINKDB_HOSTNAME",
+    type=click.STRING,
+)
+@click.option(
+    "--port",
+    "-p",
+    default=28015,
+    help="The RethinkDB driver port to connect to.",
+    envvar="RETHINKDB_DRIVER_PORT",
+    type=click.INT,
+)
+def cmd_main(*args, **kwargs):
     """
     Group of commands for the RethinkDB database.
     """
@@ -55,3 +86,6 @@ def cmd_main():
 cmd_main.add_command(cmd_index_rebuild, "index_rebuild")
 cmd_main.add_command(cmd_repl, "repl")
 cmd_main.add_command(cmd_restore, "restore")
+
+if __name__ == "__main__":
+    cmd_main(auto_envvar_prefix="RETHINKDB")

rethinkdb/encoder.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/errors.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/handshake.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/query.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/repl.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

rethinkdb/utilities.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

scripts/convert_protofile.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the "License"); you may not use
 # this file except in compliance with the License. You may obtain a copy of the

tests/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/helpers.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/integration/test_ast.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/integration/test_queries.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/integration/test_repl.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_date_and_time.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_encoder.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_errors.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_handshake.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_net.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_query.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_repl.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.

tests/test_utilities.py

@@ -1,4 +1,4 @@
-# Copyright 2022 RethinkDB
+# Copyright 2022 - present RethinkDB
 #
 # Licensed under the Apache License, Version 2.0 (the 'License');
 # you may not use this file except in compliance with the License.