From 5405eb6929294a9fab3b06d0f54f74355f489695 Mon Sep 17 00:00:00 2001
From: Andrey <andrey@simiotics.com>
Date: Fri, 31 May 2024 18:35:48 +0300
Subject: [PATCH] Add temp changes.

---
 moonstreamapi/moonstreamapi/admin/cli.py   | 49 +++++++++++++++++++++-
 moonstreamdb-v3/moonstreamdbv3/db.py       | 12 ++++--
 moonstreamdb-v3/moonstreamdbv3/version.txt |  2 +-
 3 files changed, 57 insertions(+), 6 deletions(-)

diff --git a/moonstreamapi/moonstreamapi/admin/cli.py b/moonstreamapi/moonstreamapi/admin/cli.py
index 1c82c600..3c544905 100644
--- a/moonstreamapi/moonstreamapi/admin/cli.py
+++ b/moonstreamapi/moonstreamapi/admin/cli.py
@@ -29,6 +29,7 @@ from .migrations import (
     generate_entity_subscriptions,
     update_dashboard_subscription_key,
 )
+from .databases import databases_v2_to_v3_labels_migration
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -302,6 +303,14 @@ def moonworm_tasks_v3_migrate(args: argparse.Namespace) -> None:
     moonworm_tasks.migrate_v3_tasks(user_id=args.user_id, customer_id=args.customer_id)
 
 
+def databases_v2_to_v3_labels_migration_handler(args: argparse.Namespace) -> None:
+    """
+    Migrate labels in database
+    """
+
+    databases_v2_to_v3_labels_migration(args.user_id, args.blockchain)
+
+
 def main() -> None:
     cli_description = f"""Moonstream Admin CLI
 
@@ -557,7 +566,7 @@ This CLI is configured to work with the following API URLs:
     parser_moonworm_tasks_add.set_defaults(func=moonworm_tasks_add_subscription_handler)
 
     parser_moonworm_tasks_migrate = subcommands_moonworm_tasks.add_parser(
-        "migrate-v3",
+        "migrate-v2-tasks",
         description="Migrate moonworm tasks to abi_jobs of moonstream index",
     )
 
@@ -648,6 +657,44 @@ This CLI is configured to work with the following API URLs:
 
     generate_usage_parser.set_defaults(func=generate_usage_handler)
 
+    ### databases commands
+    databases_parser = subcommands.add_parser(
+        "databases", description="Manage Moonstream databases"
+    )
+
+    databases_parser.set_defaults(func=lambda _: databases_parser.print_help())
+
+    databases_subcommands = databases_parser.add_subparsers(
+        description="Database commands"
+    )
+
+    database_labels_migration_parser = databases_subcommands.add_parser(
+        "v2-to-v3-labels-migration",
+        description="Migrate labels in database",
+    )
+
+    database_labels_migration_parser.add_argument(
+        "--user-id",
+        type=uuid_type,
+        help="User ID for which to migrate labels",
+    )
+
+    database_labels_migration_parser.add_argument(
+        "--customer-id",
+        type=uuid_type,
+        help="Customer ID for which to migrate labels",
+    )
+
+    database_labels_migration_parser.add_argument(
+        "--blockchain",
+        type=str,
+        help="Blockchain for which to migrate labels",
+    )
+
+    database_labels_migration_parser.set_defaults(
+        func=lambda args: print("Not implemented yet")
+    )
+
     args = parser.parse_args()
     args.func(args)
 
diff --git a/moonstreamdb-v3/moonstreamdbv3/db.py b/moonstreamdb-v3/moonstreamdbv3/db.py
index 1c983952..04f3f2bf 100644
--- a/moonstreamdb-v3/moonstreamdbv3/db.py
+++ b/moonstreamdb-v3/moonstreamdbv3/db.py
@@ -14,13 +14,17 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 try:
-    MOONSTREAM_DB_URI = os.environ.get("MOONSTREAM_DB_URI")
+    MOONSTREAM_DB_URI = os.environ.get("MOONSTREAM_DB_V3_INDEXES_URI")
     if MOONSTREAM_DB_URI is None:
-        raise Warning("MOONSTREAM_DB_URI environment variable must be set")
+        raise Warning("MOONSTREAM_DB_V3_INDEXES_URI environment variable must be set")
 
-    MOONSTREAM_DB_URI_READ_ONLY = os.environ.get("MOONSTREAM_DB_URI_READ_ONLY")
+    MOONSTREAM_DB_URI_READ_ONLY = os.environ.get(
+        "MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY"
+    )
     if MOONSTREAM_DB_URI_READ_ONLY is None:
-        raise Warning("MOONSTREAM_DB_URI_READ_ONLY environment variable must be set")
+        raise Warning(
+            "MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY environment variable must be set"
+        )
 except ValueError as e:
     raise ValueError(e)
 except Warning:
diff --git a/moonstreamdb-v3/moonstreamdbv3/version.txt b/moonstreamdb-v3/moonstreamdbv3/version.txt
index 99d85ecd..5c4511c3 100644
--- a/moonstreamdb-v3/moonstreamdbv3/version.txt
+++ b/moonstreamdb-v3/moonstreamdbv3/version.txt
@@ -1 +1 @@
-0.0.6
\ No newline at end of file
+0.0.7
\ No newline at end of file