From 2b9ff0a7bffe64bb4c69bc812e2dfd27db30395a Mon Sep 17 00:00:00 2001
From: Geoffrey Yu <geoffxy@mit.edu>
Date: Thu, 26 Sep 2024 10:44:09 -0400
Subject: [PATCH] Remove utils

---
 setup.py                          |  1 -
 src/brad/admin/bootstrap_vdbes.py | 27 ---------------------------
 src/brad/blueprint/vdbe.py        | 25 -------------------------
 src/brad/exec/admin.py            |  2 --
 4 files changed, 55 deletions(-)
 delete mode 100644 src/brad/admin/bootstrap_vdbes.py
 delete mode 100644 src/brad/blueprint/vdbe.py

diff --git a/setup.py b/setup.py
index 34653397..d6a4cd2b 100644
--- a/setup.py
+++ b/setup.py
@@ -45,7 +45,6 @@
     "ddsketch",
     "tqdm",
     "psycopg[binary]",
-    "pydantic",
 ]
 
 DEV_REQUIRES = [
diff --git a/src/brad/admin/bootstrap_vdbes.py b/src/brad/admin/bootstrap_vdbes.py
deleted file mode 100644
index f709da62..00000000
--- a/src/brad/admin/bootstrap_vdbes.py
+++ /dev/null
@@ -1,27 +0,0 @@
-import logging
-
-logger = logging.getLogger(__name__)
-
-
-def register_admin_action(subparser) -> None:
-    parser = subparser.add_parser(
-        "bootstrap_vdbes", help="Set up a new virtual infrastructure on BRAD."
-    )
-    parser.add_argument(
-        "--physical-config-file",
-        type=str,
-        required=True,
-        help="Path to BRAD's physical configuration file.",
-    )
-    parser.add_argument(
-        "--vdbe-file",
-        type=str,
-        required=True,
-        help="Path to the virtual infrastructure definition to boostrap.",
-    )
-    parser.set_defaults(admin_action=bootstrap_vdbes)
-
-
-# This method is called by `brad.exec.admin.main`.
-def bootstrap_vdbes(_args) -> None:
-    logger.info("Running VDBE boostrap.")
diff --git a/src/brad/blueprint/vdbe.py b/src/brad/blueprint/vdbe.py
deleted file mode 100644
index e2d1a235..00000000
--- a/src/brad/blueprint/vdbe.py
+++ /dev/null
@@ -1,25 +0,0 @@
-import enum
-
-from pydantic import BaseModel
-from typing import Set
-
-# This class contains Python definitions
-
-
-class QueryInterface(enum.Enum):
-    SqlPostgresql = "SQL_POSTGRESQL"
-    SqlMysql = "SQL_MYSQL"
-    SqlAwsRedshift = "AWS_REDSHIFT"
-    SqlAwsAthena = "AWS_ATHENA"
-
-
-class VirtualTable(BaseModel):
-    name: str
-    writable: bool
-
-
-class VirtualEngine(BaseModel):
-    name: str
-    query_interface: QueryInterface
-    max_staleness_ms: int
-    tables: Set[VirtualTable]
diff --git a/src/brad/exec/admin.py b/src/brad/exec/admin.py
index dde894b2..d70e7d7d 100644
--- a/src/brad/exec/admin.py
+++ b/src/brad/exec/admin.py
@@ -16,7 +16,6 @@
 import brad.admin.clean_dataset as clean_dataset
 import brad.admin.alter_schema as alter_schema
 import brad.admin.table_adjustments as table_adjustments
-import brad.admin.bootstrap_vdbes as bootstrap_vdbes
 
 logger = logging.getLogger(__name__)
 
@@ -32,7 +31,6 @@ def register_command(subparsers) -> None:
         help="Set to enable debug logging.",
     )
     admin_subparsers = parser.add_subparsers(title="Administrative Actions")
-    bootstrap_vdbes.register_admin_action(admin_subparsers)
     bootstrap_schema.register_admin_action(admin_subparsers)
     drop_schema.register_admin_action(admin_subparsers)
     bulk_load.register_admin_action(admin_subparsers)