day1

Author: rochacbruno

.gitignore

@@ -127,3 +127,4 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+*.db

.vscode/settings.json

@@ -19,5 +19,14 @@
     "coverage.xml": true,
     ".pytest_cache": true
   },
-  "editor.mouseWheelZoom": true
+  "editor.mouseWheelZoom": true,
+  "python.testing.pytestArgs": [
+    "tests"
+  ],
+  "python.testing.unittestEnabled": false,
+  "python.testing.pytestEnabled": true,
+  "sqlite.logLevel": "DEBUG",
+  "python.formatting.provider": "black",
+  "editor.renderWhitespace": "boundary",
+  "workbench.tree.renderIndentGuides": "always"
 }

beerlog/cli.py

@@ -1,5 +1,52 @@
-from .config import settings
+import typer
+from typing import Optional
+from rich.console import Console
+from rich.table import Table
+from rich import print
 
+from beerlog.core import add_beer_to_database, get_beers_from_database
 
-def main():
-    print("Hello from", settings.NAME)
+
+main = typer.Typer(help="Beer Management Application")
+console = Console()
+
+
+@main.command()
+def add(
+    name: str,
+    style: str,
+    flavor: int = typer.Option(...),
+    image: int = typer.Option(...),
+    cost: int = typer.Option(...),
+):
+    """Adds a new beer to the database"""
+    if add_beer_to_database(name, style, flavor, image, cost):
+        print(":beer_mug: Beer added!!!")
+    else:
+        print(":no_entry: - Cannot add beer.")
+
+
+@main.command("list")
+def list_beers(style: Optional[str] = None):
+    """Lists beers from the database"""
+    beers = get_beers_from_database(style)
+    table = Table(
+        title="Beerlog Database" if not style else f"Beerlog {style}"
+    )
+    headers = [
+        "id",
+        "name",
+        "style",
+        "flavor",
+        "image",
+        "cost",
+        "rate",
+        "date",
+    ]
+    for header in headers:
+        table.add_column(header, style="magenta")
+    for beer in beers:
+        beer.date = beer.date.strftime("%Y-%m-%d")
+        values = [str(getattr(beer, header)) for header in headers]
+        table.add_row(*values)
+    console.print(table)

beerlog/core.py

@@ -0,0 +1,27 @@
+from typing import Optional, List
+from sqlmodel import select
+from beerlog.database import get_session
+from beerlog.models import Beer
+
+
+def add_beer_to_database(
+    name: str,
+    style: str,
+    flavor: int,
+    image: int,
+    cost: int,
+) -> bool:
+    with get_session() as session:
+        beer = Beer(**locals())
+        session.add(beer)
+        session.commit()
+
+    return True
+
+
+def get_beers_from_database(style: Optional[str] = None) -> List[Beer]:
+    with get_session() as session:
+        sql = select(Beer)
+        if style:
+            sql = sql.where(Beer.style == style)
+        return list(session.exec(sql))

beerlog/database.py

@@ -0,0 +1,20 @@
+import warnings
+from sqlalchemy.exc import SAWarning
+from sqlmodel.sql.expression import Select, SelectOfScalar
+
+warnings.filterwarnings("ignore", category=SAWarning)
+SelectOfScalar.inherit_cache = True
+Select.inherit_cache = True
+
+
+from sqlmodel import create_engine, Session
+
+from beerlog import models
+from beerlog.config import settings
+
+engine = create_engine(settings.database.url, echo=False)
+models.SQLModel.metadata.create_all(engine)
+
+
+def get_session():
+    return Session(engine)

beerlog/models.py

@@ -0,0 +1,31 @@
+from sqlmodel import SQLModel, Field
+from datetime import datetime
+from typing import Optional
+from pydantic import validator  # NEW
+from statistics import mean  # NEW
+
+
+class Beer(SQLModel, table=True):
+    id: Optional[int] = Field(primary_key=True, default=None, index=True)
+    name: str
+    style: str
+    flavor: int
+    image: int
+    cost: int
+    rate: int = 0
+    date: datetime = Field(default_factory=datetime.now)
+
+    # NEW
+    @validator("image", "flavor", "cost")
+    def validate_ratings(cls, v, field):
+        if v < 1 or v > 10:
+            raise RuntimeError(f"{field.name} must be between 1 and 10")
+        return v
+
+    @validator("rate", always=True)
+    def calculate_rate(cls, v, values):
+        rate = mean([values["flavor"], values["image"], values["cost"]])
+        return int(rate)
+
+
+beer = Beer(name="Lagunitas", style="IPA", flavor=3, image=6, cost=8)

beerlog/settings.toml

@@ -1 +1,4 @@
 name = "beerlog"
+
+[database]
+url = "sqlite:///beerlog.db"