Upload 9 files

models/Database_Entity.py

@@ -0,0 +1,79 @@
+from mongoengine import (
+    Document, StringField, ReferenceField, DateTimeField, BooleanField, ObjectIdField,EmbeddedDocumentField
+)
+from datetime import datetime
+import pytz
+from mongoengine import connect
+import os
+from dotenv import load_dotenv
+load_dotenv()
+MONGO_URI = os.getenv("MONGO_URI", "mongodb+srv://kltn20133118:XEYSVzYvEwsp5Cvo@cluster0.nnsw9.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0")
+
+def init_db(db_name="chatbot_hmdrinks"):
+    connect(db_name, host=MONGO_URI)
+
+from mongoengine import Document, StringField, IntField,ListField
+
+
+class User(Document):
+    _id = ObjectIdField(required=True, primary_key=True)
+    user_name = StringField(required=True, max_length=255)
+    user_id = IntField(unique=True, required=True)
+    is_deleted = BooleanField(default=False)
+    date_deleted = DateTimeField(default=None)
+    meta = {'collection': 'users'}
+
+
+class ChatHistory(Document):
+    _id = ObjectIdField(required=True, primary_key=True)
+    user = ReferenceField(User, required=True, reverse_delete_rule=2)
+    name_chat = StringField(required=True, unique=True, max_length=255)
+    is_deleted = BooleanField(default=False)
+    date_deleted = DateTimeField(default=None)
+    meta = {'collection': 'chat_history'}
+
+from bson import ObjectId
+from mongoengine import EmbeddedDocument, StringField, IntField, FloatField
+
+
+
+
+class CartProduct(EmbeddedDocument):
+    name = StringField()
+    quantity = IntField(min_value=1, default=1)
+    size = StringField()
+
+class DetailChat(Document):
+    _id = ObjectIdField(required=True, primary_key=True,default=ObjectId)
+    chat_history = ReferenceField(ChatHistory, required=True, reverse_delete_rule=2)
+    you_message = StringField()
+    ai_message = StringField()
+    data_relevant = StringField()
+    full_input = StringField()
+    source_file = StringField()
+    cart_id = IntField()
+    confirmed_order = BooleanField(default=False)
+    timestamp = DateTimeField(default=lambda: datetime.now(pytz.UTC))  
+    is_deleted = BooleanField(default=False)
+    date_deleted = DateTimeField(default=lambda: datetime.now(pytz.UTC))  
+    meta = {'collection': 'detail_chat',
+            'strict': False
+            }
+                                       
+                                       
+class ChatCart(Document):
+    _id = ObjectIdField(primary_key=True, default=ObjectId)
+    chat_history = ReferenceField(ChatHistory, required=True, reverse_delete_rule=2)
+    created_from_detail_chat = ReferenceField(DetailChat)  # Để biết cart được tạo từ đoạn chat nào
+    cart_products = ListField(EmbeddedDocumentField(CartProduct), default=list)
+    status = StringField(choices=["pending", "confirmed", "failed"], default="pending")
+    confirmed_order = BooleanField(default=False)
+    created_at = DateTimeField(default=lambda: datetime.now(pytz.UTC))
+    confirmed_at = DateTimeField()
+    note = StringField()
+    cart_id = IntField()
+    
+    meta = {
+        'collection': 'chat_cart',
+        'strict': False
+    }

models/Database_MySQL.py

@@ -0,0 +1,74 @@
+from sqlalchemy import Column, String, Text, DateTime, Integer, ForeignKey, Boolean, Enum
+from sqlalchemy.orm import relationship, DeclarativeBase
+from sqlalchemy.sql import func
+import enum
+
+
+class UserRole(enum.Enum):
+    ADMIN = "ADMIN"
+    CUSTOMER = "CUSTOMER"
+    SHIPPER = "SHIPPER"
+
+
+class SexType(enum.Enum):
+    OTHER = "OTHER"
+    MALE = "MALE"
+    FEMALE = "FEMALE"
+
+
+class UserType(enum.Enum):
+    BASIC = "BASIC"
+    BOTH = "BOTH"
+    EMAIL = "EMAIL"
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class User(Base):
+    __tablename__ = 'user'
+    user_id = Column(Integer, primary_key=True, autoincrement=True)
+    avatar = Column(Text, nullable=True)
+    birth_date = Column(DateTime, nullable=True)
+    city = Column(String(255), nullable=True)
+    district = Column(String(255), nullable=True)
+    email = Column(String(255), unique=True, nullable=False)
+    full_name = Column(String(255), nullable=False)
+    is_deleted = Column(Boolean, default=False)
+    password = Column(Text, nullable=False)
+    phone_number = Column(String(20), nullable=True)
+    role = Column(Enum(UserRole), nullable=False)  # 🎯 Enum Role
+    sex = Column(Enum(SexType), nullable=True)  # 🎯 Enum Sex
+    street = Column(String(255), nullable=True)
+    type = Column(Enum(UserType), nullable=True)  # 🎯 Enum Type
+    username = Column(String(255), unique=True, nullable=False)
+    date_created = Column(DateTime, default=func.now())
+    date_updated = Column(DateTime, default=func.now(), onupdate=func.now())
+    date_deleted = Column(DateTime, nullable=True)
+    ward = Column(String(255), nullable=True)
+    tokens = relationship("Token", back_populates="user")
+    user_chats = relationship("UserChat", back_populates="user")
+
+
+class Token(Base):
+    __tablename__ = 'token'
+    token_id = Column(Integer, primary_key=True, autoincrement=True)
+    access_token = Column(Text, nullable=False)
+    expire = Column(DateTime, nullable=False)
+    refresh_token = Column(Text, nullable=False)
+    user_id = Column(Integer, ForeignKey('user.user_id'), nullable=False)
+    user = relationship("User", back_populates="tokens")
+
+
+class UserChat(Base):
+    __tablename__ = 'user_chat'
+    user_chat_id = Column(Integer, primary_key=True, autoincrement=True)
+    chat_name = Column(String(255), nullable=False)
+    id_mongo_db = Column(String(255), nullable=True)
+    user_id = Column(Integer, ForeignKey('user.user_id'), nullable=False)
+    date_created = Column(DateTime, default=func.now())
+    date_updated = Column(DateTime, default=func.now(), onupdate=func.now())
+    date_deleted = Column(DateTime, nullable=True)
+    is_deleted = Column(Boolean, default=False)
+    user = relationship("User", back_populates="user_chats")