From be8b011c9c4c44f8de99d5cf19ebd1dae5b7dfe9 Mon Sep 17 00:00:00 2001 From: Blaise Thompson Date: Tue, 12 Nov 2019 08:32:41 -0600 Subject: coloring --- shopdb/_base.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) (limited to 'shopdb/_base.py') diff --git a/shopdb/_base.py b/shopdb/_base.py index 8033d28..5e8c3bc 100644 --- a/shopdb/_base.py +++ b/shopdb/_base.py @@ -1,8 +1,18 @@ -__all__ = ["engine", "Base"] +__all__ = ["engine", "Base", "Session"] +import appdirs +import pathlib import sqlalchemy +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import sessionmaker -engine = sqlalchemy.create_engine("sqlite:///:memory:", echo=True) -Base = sqlalchemy.ext.declarative.declarative_base + +path = pathlib.Path(appdirs.user_data_dir(appname="shopdb")) / "sqlite3.db" +path.parent.mkdir(parents=True, exist_ok=True) +engine = sqlalchemy.create_engine(f"sqlite:///{path}", echo=False) + +Base = declarative_base() + +Session = sessionmaker(bind=engine) -- cgit v1.2.3