24 lines
650 B
Python
24 lines
650 B
Python
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from models import Base
|
|
import os
|
|
|
|
# Get the absolute path to the database file
|
|
DATABASE_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "automotive_docs.db")
|
|
DATABASE_URL = f"sqlite:///{DATABASE_FILE}"
|
|
|
|
def init_db():
|
|
# Create new engine
|
|
engine = create_engine(DATABASE_URL)
|
|
|
|
# Create tables if they don't exist
|
|
if not os.path.exists(DATABASE_FILE):
|
|
Base.metadata.create_all(engine)
|
|
|
|
return engine
|
|
|
|
def get_session():
|
|
engine = create_engine(DATABASE_URL)
|
|
Session = sessionmaker(bind=engine)
|
|
return Session()
|