diff --git a/src/hellocomputer/analytics.py b/src/hellocomputer/analytics.py index 560214f..a841e40 100644 --- a/src/hellocomputer/analytics.py +++ b/src/hellocomputer/analytics.py @@ -92,6 +92,18 @@ class DDB: .fetchall()[0][0] .split(";") ) + + # Load all the tables into the database + for sheet in self.sheets: + self.db.query(f""" + create table {sheet} as ( + select + * + from + read_csv_auto('{path}/{sheet}.csv') + ) + """) + return self def load_folder_gcs(self, path: str) -> Self: diff --git a/test/data/TestExcelHelloComputer.xlsx b/test/data/TestExcelHelloComputer.xlsx index 87c760f..c9e1c31 100644 Binary files a/test/data/TestExcelHelloComputer.xlsx and b/test/data/TestExcelHelloComputer.xlsx differ diff --git a/test/test_data.py b/test/test_data.py index ddeb3bc..6f228d8 100644 --- a/test/test_data.py +++ b/test/test_data.py @@ -19,4 +19,8 @@ def test_dump(): def test_load(): db = DDB().load_folder_local(TEST_OUTPUT_FOLDER) + + results = db.query("select * from answers").fetchall() + assert db.sheets == ("answers",) + assert len(results) == 2