feat: added dependencies and created clickhouse connectior and sessio…

…n developments

pyproject.toml

@@ -22,6 +22,7 @@ coverage = "^7.5.0"
 pytest = "^8.2.0"
 sqlalchemy = "^2.0.31"
 sqlalchemy-clickhouse = "^0.1.5.post0"
+alembic = "^1.13.2"
 
 
 [build-system]

src/recommendation_engine/app/core/database/clickhouse.py

@@ -0,0 +1,31 @@
+from loguru import logger
+from typing import Iterator
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, Session
+from sqlalchemy.ext.declarative import declarative_base
+
+
+engine = create_engine("clickhouse+native://default:@localhost/default", future=True)
+
+SessionLocal = sessionmaker(
+ autocommit=False,
+ autoflush=False,
+ bind=engine,
+)
+
+ClickhouseBase = declarative_base()
+
+
+def get_session() -> Iterator[Session]:
+ db = SessionLocal()
+ try:
+ yield db
+ except Exception as e:
+ logger.exception(str(e))
+ db.rollback()
+ finally:
+ db.close()
+
+
+__all__ = ["get_session", "ClickhouseBase"]