XJh.dZddlZddlmZddlmZmZmZmZddlm Z ddl Z e GddZ e Gdd Z e Gd d Z e Gd d ZGddZy)zb Database models for the MAB Recommendation System. Defines the structure of all database tables. N)datetime)OptionalListDictAny) dataclassc^eZdZUdZeed<eed<eed<eed<dZe ed<de ee ffd Z y ) FeedbackDBz)User interaction feedback database model.user_id product_id event_typesessionT active_userreturnc|j|j|j|jr|jj nd|j dS)N)r r r rr)r r r r isoformatrselfs V/Users/divyeshpatel/Desktop/sahana/Recommender/recommender_rl/recommender/db/models.pyto_dictzFeedbackDB.to_dictsB||////37< '''rcTt|dr|j|jyy)NrR)hasattrrIrR)rexc_typeexc_valexc_tbs r__exit__zDatabaseManager.__exit__s$ 4. / JJt// 0 0rN)zrecommender_system.db) rrrrrr@rB ConnectionrCrIrPrSrYr rrr<r<asFA++  2 2 I>V(1rr<)rrBrtypingrrrr dataclassesrjsonr r"r-r6r<r rrr^s ,,!      $     ,     ,     $b1b1r