add mysql

compose.yaml

@@ -0,0 +1,7 @@
+services:
+  db:
+    image: mysql/mysql-server
+    ports:
+      - 3306:3306
+    volumes:
+      - ./ddl.sql:/docker-entrypoint-initdb.d/ddl.sql

ddl.sql

@@ -0,0 +1,17 @@
+create database foo;
+
+create table if not exists foo.classes (
+  id smallint not null auto_increment,
+  name varchar(255) not null,
+  is_asset bool not null,
+  primary key (id)
+);
+
+create table if not exists foo.amounts (
+  id mediumint not null auto_increment,
+  day date not null,
+  amount decimal(10,2) not null,
+  class_id smallint not null,
+  primary key (id),
+  foreign key (class_id) references classes(id)
+);

seed-db.py

@@ -0,0 +1,49 @@
+import json
+
+import polars as pl
+
+df = (
+    pl.read_csv("balance-sheet.csv")
+    .select(pl.exclude("Notes", "Change", "Total"))
+    .with_columns(pl.exclude("Date").str.replace_all("[,$]", ""))
+    .select(
+        pl.col("Date").str.to_date("%-m/%-d/%Y"), pl.exclude("Date").cast(pl.Decimal)
+    )
+)
+
+assert df.null_count().select(s=pl.sum_horizontal(pl.all())).row(0, named=True) == {
+    "s": 0
+}
+
+print(df)
+
+with open("names.json") as f:
+    asset_indicators = json.load(f)
+
+
+# database = Database('sqlite+aiosqlite:///example.db')  # FIXME
+classes_query = "insert into foo.classes(name, is_asset) values (:name, :is_asset)"
+
+asset_info = [
+    {"name": col, "is_asset": col in asset_indicators["assets"]}
+    for col in df.columns
+    if col != "Date"
+]
+print(asset_info)
+
+
+async def main() -> None:
+    await database.connect()
+
+    await database.execute_many(query=classes_query, values=asset_info)
+
+    rows = await database.fetch_all(query="select * from foo.classes")
+    print(f"{rows=}")
+
+
+for row in df.iter_rows(named=True):
+    # TODO: insert into the various tables
+    # this will be moved into main
+    print(row)
+
+# asyncio.run(main())