parsing log w/pydantic, typer cli

Author: jackharrhy

.gitignore

@@ -0,0 +1 @@
+__pycache__

cli.py

@@ -0,0 +1,18 @@
+from pathlib import Path
+
+import typer
+from devtools import debug
+
+from library.parse import parse_log
+
+app = typer.Typer()
+
+
+@app.command()
+def cmd_parse_log():
+    log_path = Path("./log.toml")
+    debug(parse_log(log_path))
+
+
+if __name__ == "__main__":
+    app()

library/models.py

@@ -0,0 +1,46 @@
+from typing import Annotated, List, Literal, Union
+
+from pydantic import BaseModel, Field
+
+
+class User(BaseModel):
+    name: str
+    github_username: str | None
+
+
+class Book(BaseModel):
+    id: str
+    isbn: int
+    url: str | None
+    cover_image: str | None
+
+
+class BaseAction(BaseModel):
+    pass
+
+
+class AddUser(BaseAction, User):
+    action_type: Literal["AddUser"]
+
+
+class AddBook(BaseAction):
+    action_type: Literal["AddBook"]
+    user: str
+    book: Book
+
+
+class LendBook(BaseAction):
+    action_type: Literal["LendBook"]
+    from_user: str
+    to_user: str
+    book_id: str
+
+
+class Action(BaseModel):
+    __root__: Annotated[
+        Union[AddUser, AddBook, LendBook], Field(discriminator="action_type")
+    ]
+
+
+class Log(BaseModel):
+    actions: List[Action]

library/parse.py

@@ -0,0 +1,9 @@
+from pathlib import Path
+
+import toml
+
+from library.models import Log
+
+
+def parse_log(log_path: Path) -> Log:
+    return Log(**toml.loads(log_path.read_text()))

log.toml

@@ -0,0 +1,25 @@
+[[actions]]
+action_type = 'AddUser'
+name = 'jack'
+github_username = 'jackharrhy'
+
+[[actions]]
+action_type = 'AddUser'
+name = 'marty'
+github_username = 'lilmert'
+
+[[actions]]
+action_type = 'AddBook'
+user = 'jack'
+book = { id = 'dop', isbn = 9781617298578, url = 'https://www.manning.com/books/data-oriented-programming', cover_image_url = 'https://images.manning.com/360/480/resize/book/0/76cfc61-95fd-4c0d-b221-0e5c3ca90583/Sharvit-DOP-HI.png' }
+
+[[actions]]
+action_type = 'AddBook'
+user = 'marty'
+book = { id = 'ddd', isbn = 9781521822807, cover_image_url = 'https://images-na.ssl-images-amazon.com/images/I/41gvS8EAHoL._SX404_BO1,204,203,200_.jpg' }
+
+[[actions]]
+action_type = 'LendBook'
+from_user = 'marty'
+to_user = 'jack'
+book_id = 'ddd'

main.py

@@ -6,6 +6,10 @@ authors = ["Jack Arthur Harrhy <[email protected]>"]
 
 [tool.poetry.dependencies]
 python = "^3.10"
+pydantic = "^1.10.2"
+toml = "^0.10.2"
+typer = "^0.6.1"
+devtools = "^0.9.0"
 
 [tool.poetry.dev-dependencies]