o XJh.@sdZddlZddlmZddlmZmZmZmZddlm Z ddl Z e GdddZ e Gdd d Z e Gd d d Z e Gd d d ZGdddZdS)zb Database models for the MAB Recommendation System. Defines the structure of all database tables. N)datetime)OptionalListDictAny) dataclassc@sTeZdZUdZeed<eed<eed<eed<dZe ed<de ee ffd d Z d S) FeedbackDBz)User interaction feedback database model.user_id product_id event_typesessionT active_userreturncCs(|j|j|j|jr|jnd|jdS)N)r r r r r )r r r r isoformatr selfrJ/Users/divyeshpatel/Desktop/sahana/Recommender/recommender_rl/db/models.pyto_dicts zFeedbackDB.to_dictN) __name__ __module__ __qualname____doc__int__annotations__strrr boolrrrrrrrr s  rc@speZdZUdZeed<eed<eed<eed<eed<dZe eed<dZ e eed <d e ee ffd d Z dS) ProductCatalogzProduct catalog database model.r product_name brand_namecategorypriceN description image_urlrcCs"|j|j|j|j|j|j|jdS)Nr rrr r!r"r#r$rrrrr+szProductCatalog.to_dict)rrrrrrrfloatr"rr#rrrrrrrr s rc@seZdZUdZeed<eed<dZeeed<dZ eeed<dZ eeed<dZ ee ed<dZ ee ed <d eeeffd d ZdS) UserDemographicsz-User demographics and profile database model.r nameNgenderageuser_cluster_id created_at last_activercCsB|j|j|j|j|j|jr|jnd|jr|jdSddS)N)r r'r(r)r*r+r,)r r'r(r)r*r+rr,rrrrrBszUserDemographics.to_dict)rrrrrrrr(rr)r*r+rr,rrrrrrrr&7s r&c@s`eZdZUdZeed<eed<eed<dZeeed<dZ ee ed<de ee ffd d Z dS) SearchHistoryz#User search history database model.r query session_idNr timestamprcCs,|j|j|j|j|jr|jdSddS)N)r r.r r/r0)r r.r r/r0rrrrrrWszSearchHistory.to_dict)rrrrrrrr rr0rrrrrrrrr-Ns r-c@sXeZdZdZddefddZdejfddZd ejfd d Z d d Z ddZ ddZ dS)DatabaseManagerz7Manages SQLite database connections and initialization.recommender_system.dbdb_pathcCs ||_dSN)r3)rr3rrr__init__ds zDatabaseManager.__init__rcCst|j}tj|_|S)z3Create a new database connection for thread safety.)sqlite3connectr3ZRowZ row_factoryr connectionrrrr7gs zDatabaseManager.connectr9cCs|r|dSdS)z%Close a specific database connection.N)closer8rrrr:ms zDatabaseManager.closecCs|}|}|d|d|d|d|d|d|d|d|d |d |d |d |d |tddS)zInitialize database tables.a8 CREATE TABLE IF NOT EXISTS feedback_db ( id INTEGER PRIMARY KEY AUTOINCREMENT, user_id INTEGER NOT NULL, product_id TEXT NOT NULL, event_type TEXT NOT NULL, session DATETIME NOT NULL, active_user BOOLEAN DEFAULT 1, created_at DATETIME DEFAULT CURRENT_TIMESTAMP, FOREIGN KEY (user_id) REFERENCES user_demographics (user_id), FOREIGN KEY (product_id) REFERENCES product_catalog (product_id) ) a CREATE TABLE IF NOT EXISTS product_catalog ( product_id INTEGER PRIMARY KEY, product_name TEXT NOT NULL, brand_name TEXT NOT NULL, category TEXT NOT NULL, price REAL NOT NULL, description TEXT, image_url TEXT, created_at DATETIME DEFAULT CURRENT_TIMESTAMP ) a CREATE TABLE IF NOT EXISTS user_demographics ( user_id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT NOT NULL, gender TEXT, age TEXT, user_cluster_id INTEGER, created_at DATETIME DEFAULT CURRENT_TIMESTAMP, last_active DATETIME DEFAULT CURRENT_TIMESTAMP ) a CREATE TABLE IF NOT EXISTS search_history ( id INTEGER PRIMARY KEY AUTOINCREMENT, user_id INTEGER NOT NULL, query TEXT NOT NULL, product_id INTEGER, session_id INTEGER NOT NULL, timestamp DATETIME DEFAULT CURRENT_TIMESTAMP, FOREIGN KEY (user_id) REFERENCES user_demographics (user_id), FOREIGN KEY (product_id) REFERENCES product_catalog (product_id) ) zHCREATE INDEX IF NOT EXISTS idx_feedback_user_id ON feedback_db (user_id)zNCREATE INDEX IF NOT EXISTS idx_feedback_product_id ON feedback_db (product_id)zNCREATE INDEX IF NOT EXISTS idx_feedback_event_type ON feedback_db (event_type)zHCREATE INDEX IF NOT EXISTS idx_feedback_session ON feedback_db (session)zMCREATE INDEX IF NOT EXISTS idx_product_category ON product_catalog (category)zLCREATE INDEX IF NOT EXISTS idx_product_brand ON product_catalog (brand_name)zRCREATE INDEX IF NOT EXISTS idx_user_cluster ON user_demographics (user_cluster_id)zICREATE INDEX IF NOT EXISTS idx_search_user_id ON search_history (user_id)zOCREATE INDEX IF NOT EXISTS idx_search_session_id ON search_history (session_id)u,✅ Database tables initialized successfullyN)r7cursorexecutecommitprint)rconnr;rrr init_databasers"             zDatabaseManager.init_databasecCs||_|jSr4)r7_current_connectionrrrr __enter__s zDatabaseManager.__enter__cCst|dr ||jdSdS)NrA)hasattrr:rA)rexc_typeexc_valexc_tbrrr__exit__s zDatabaseManager.__exit__N)r2) rrrrrr5r6 Connectionr7r:r@rBrGrrrrr1asK r1)rr6rtypingrrrr dataclassesrjsonrrr&r-r1rrrrs