Merge pull request #138 from ilyarolf/feature/sqladmin

Feature/sqladmin

Author: ilyarolf

.env

@@ -31,4 +31,8 @@ REFERRER_BONUS_PERCENT = "3"
 REFERRER_BONUS_DEPOSIT_LIMIT = "5"
 REFERRAL_BONUS_CAP_PERCENT = "7"
 REFERRER_BONUS_CAP_PERCENT = "7"
-TOTAL_BONUS_CAP_PERCENT = "12"
+TOTAL_BONUS_CAP_PERCENT = "12"
+SQLADMIN_RAW_PASSWORD = ""
+JWT_EXPIRE_MINUTES = "30"
+JWT_ALGORITHM = "HS256"
+JWT_SECRET_KEY = ""

admin.py

@@ -0,0 +1,44 @@
+from sqladmin.authentication import AuthenticationBackend
+from starlette.requests import Request
+
+import config
+from config import SQLADMIN_HASHED_PASSWORD
+from enums.user_role import UserRole
+from utils.utils import create_access_token, decode_token, verify_password
+
+
+class AdminAuth(AuthenticationBackend):
+    async def login(self, request: Request) -> bool:
+        form = await request.form()
+        username = form.get("username")
+        password = form.get("password")
+
+        if username == "admin" and verify_password(password, SQLADMIN_HASHED_PASSWORD):
+            token = create_access_token(
+                {
+                    "sub": UserRole.ADMIN.name,
+                }
+            )
+
+            request.session.update({"token": token})
+            return True
+        else:
+            return False
+
+    async def logout(self, request: Request) -> bool:
+        request.session.clear()
+        return True
+
+    async def authenticate(self, request: Request) -> bool:
+        token = request.session.get("token")
+        if not token:
+            return False
+
+        payload = decode_token(token)
+        if not payload:
+            return False
+
+        return True
+
+
+authentication_backend = AdminAuth(secret_key=config.JWT_SECRET_KEY)

bot.py

@@ -6,14 +6,32 @@
 from aiogram.fsm.storage.redis import RedisStorage
 from aiogram.types import BufferedInputFile
 from redis.asyncio import Redis
+from sqladmin import Admin
+
 import config
 from aiogram import Bot, Dispatcher
 from aiogram.enums import ParseMode
 from fastapi import FastAPI, Request, status, HTTPException
-from db import create_db_and_tables
+
+from admin import authentication_backend
+from db import create_db_and_tables, engine
 import uvicorn
 from fastapi.responses import JSONResponse
 from enums.cryptocurrency import Cryptocurrency
+from models.buy import BuyAdmin
+from models.buyItem import BuyItemAdmin
+from models.cart import CartAdmin
+from models.cartItem import CartItemAdmin
+from models.category import CategoryAdmin
+from models.coupon import CouponAdmin
+from models.deposit import DepositAdmin
+from models.item import ItemAdmin
+from models.payment import PaymentAdmin
+from models.referral import ReferralBonusAdmin
+from models.review import ReviewAdmin
+from models.shipping_option import ShippingOptionAdmin
+from models.subcategory import SubcategoryAdmin
+from models.user import UserAdmin
 from processing.processing import processing_router
 from repositories.button_media import ButtonMediaRepository
 from services.notification import NotificationService
@@ -24,6 +42,22 @@
 bot = Bot(config.TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
 dp = Dispatcher(storage=RedisStorage(redis))
 app = FastAPI()
+admin = Admin(app=app, engine=engine, authentication_backend=authentication_backend)
+admin.add_model_view(UserAdmin)
+admin.add_model_view(BuyAdmin)
+admin.add_model_view(ShippingOptionAdmin)
+admin.add_model_view(CouponAdmin)
+admin.add_model_view(CategoryAdmin)
+admin.add_model_view(SubcategoryAdmin)
+admin.add_model_view(ItemAdmin)
+admin.add_model_view(DepositAdmin)
+admin.add_model_view(BuyItemAdmin)
+admin.add_model_view(PaymentAdmin)
+admin.add_model_view(CartAdmin)
+admin.add_model_view(CartItemAdmin)
+admin.add_model_view(ReferralBonusAdmin)
+admin.add_model_view(ReviewAdmin)
+
 app.include_router(processing_router)
 
 

config.py

@@ -4,7 +4,7 @@
 
 from enums.currency import Currency
 from enums.runtime_environment import RuntimeEnvironment
-from utils.utils import get_sslipio_external_url, start_ngrok
+from utils.utils import get_sslipio_external_url, start_ngrok, hash_password
 
 load_dotenv(".env")
 RUNTIME_ENVIRONMENT = RuntimeEnvironment(os.environ.get("RUNTIME_ENVIRONMENT"))
@@ -48,3 +48,9 @@
 REFERRAL_BONUS_CAP_PERCENT = float(os.environ.get("REFERRAL_BONUS_CAP_PERCENT", "7"))
 REFERRER_BONUS_CAP_PERCENT = float(os.environ.get("REFERRER_BONUS_CAP_PERCENT", "7"))
 TOTAL_BONUS_CAP_PERCENT = float(os.environ.get("TOTAL_BONUS_CAP_PERCENT", "12"))
+# SQLADMIN
+SQLADMIN_RAW_PASSWORD = os.environ.get("SQLADMIN_RAW_PASSWORD")
+SQLADMIN_HASHED_PASSWORD = hash_password(SQLADMIN_RAW_PASSWORD)
+JWT_EXPIRE_MINUTES = int(os.environ.get("JWT_EXPIRE_MINUTES", "30"))
+JWT_ALGORITHM = os.environ.get("JWT_ALGORITHM", "HS256")
+JWT_SECRET_KEY = os.environ.get("JWT_SECRET_KEY")

models/buy.py

@@ -1,6 +1,7 @@
 from datetime import datetime
 
 from pydantic import BaseModel
+from sqladmin import ModelView
 from sqlalchemy import Column, Integer, Float, DateTime, ForeignKey, func, CheckConstraint, Enum, String
 from sqlalchemy.orm import relationship
 
@@ -17,21 +18,34 @@ class Buy(Base):
 
     id = Column(Integer, primary_key=True)
     buyer_id = Column(Integer, ForeignKey('users.id'), nullable=True)
-    buyer = relationship('User', backref='buys')
+    buyer = relationship("User", back_populates="buys")
     total_price = Column(Float, nullable=False)
     buy_datetime = Column(DateTime, default=func.now())
     status = Column(Enum(BuyStatus), nullable=False)
     coupon_id = Column(Integer, ForeignKey('coupons.id'), nullable=True)
+    coupon = relationship("Coupon", back_populates="buys")
     discount = Column(Float, nullable=False, default=0.0)
     shipping_address = Column(String, nullable=True)
     track_number = Column(String, nullable=True)
     shipping_option_id = Column(Integer, ForeignKey('shipping_options.id'), nullable=True)
+    shipping_option = relationship("ShippingOption", back_populates="buys")
+    buy_items = relationship(
+        "BuyItem",
+        back_populates="buy",
+        cascade="all, delete-orphan"
+    )
 
     __table_args__ = (
-        CheckConstraint('quantity > 0', name='check_quantity_positive'),
         CheckConstraint('total_price > 0', name='check_total_price_positive'),
     )
 
+    def __repr__(self):
+        return get_text(Language.EN, BotEntity.USER, "purchase_history_item").format(
+            buy_id=self.id,
+            total_price=self.total_price,
+            currency_sym=config.CURRENCY.get_localized_symbol()
+        )
+
 
 class BuyDTO(BaseModel):
     id: int | None = None
@@ -60,3 +74,20 @@ class RefundDTO(BaseModel):
     item_ids: list[int] | None = None
     buy_id: int | None = None
     language: Language
+
+
+class BuyAdmin(ModelView, model=Buy):
+    column_exclude_list = [Buy.shipping_option_id,
+                           Buy.coupon_id,
+                           Buy.buyer_id]
+    column_sortable_list = [Buy.id,
+                            Buy.total_price,
+                            Buy.buy_datetime,
+                            Buy.status,
+                            Buy.discount,
+                            Buy.shipping_address,
+                            Buy.track_number]
+    can_create = False
+    can_delete = False
+    can_edit = False
+    can_export = False

models/buyItem.py

@@ -1,5 +1,7 @@
 from pydantic import BaseModel
+from sqladmin import ModelView
 from sqlalchemy import Column, Integer, ForeignKey, JSON
+from sqlalchemy.orm import relationship
 
 from models.base import Base
 
@@ -9,12 +11,34 @@ class BuyItem(Base):
 
     id = Column(Integer, primary_key=True)
     buy_id = Column(Integer, ForeignKey("buys.id", ondelete="CASCADE"), nullable=False)
+    buy = relationship(
+        "Buy",
+        back_populates="buy_items"
+    )
     # ARRAY CRUTCH FOR SQLALCHEMY+SQLITE 🩼
     # item_ids = Column(ARRAY, nullable=False)
     item_ids = Column(JSON, nullable=False)
+    review = relationship(
+        "Review",
+        back_populates="buy_item",
+        uselist=False,
+        cascade="all, delete-orphan",
+        passive_deletes=True
+    )
+
+    def __repr__(self):
+        return f"BuyItem ID:{self.id}"
 
 
 class BuyItemDTO(BaseModel):
     id: int | None = None
     buy_id: int | None = None
     item_ids: list[int] | None = None
+
+
+class BuyItemAdmin(ModelView, model=BuyItem):
+    column_exclude_list = [BuyItem.buy_id]
+    can_create = False
+    can_delete = False
+    can_edit = False
+    can_export = False

models/cart.py

@@ -5,7 +5,10 @@
 # note that the item is NOT reserved or blocked so that the availability of the item
 # needs to be checked again during checkout
 from pydantic import BaseModel
+from sqladmin import ModelView
 from sqlalchemy import Column, Integer, ForeignKey
+from sqlalchemy.orm import relationship
+
 from models.base import Base
 
 
@@ -14,8 +17,24 @@ class Cart(Base):
 
     id = Column(Integer, primary_key=True)
     user_id = Column(Integer, ForeignKey('users.id'), nullable=False)
+    user = relationship("User", back_populates="cart", uselist=False)
+    cart_items = relationship("CartItem", back_populates="cart", cascade="all, delete-orphan")
+
+    def __repr__(self):
+        return f"Cart ID:{self.id}"
 
 
 class CartDTO(BaseModel):
     id: int | None = None
     user_id: int | None = None
+
+
+class CartAdmin(ModelView, model=Cart):
+    name = "Cart"
+    name_plural = "Carts"
+    column_sortable_list = [Cart.id]
+    column_exclude_list = [Cart.user_id, Cart.cart_items]
+    can_delete = False
+    can_edit = False
+    can_create = False
+    can_export = False

models/cartItem.py

@@ -1,5 +1,7 @@
 from pydantic import BaseModel
+from sqladmin import ModelView
 from sqlalchemy import Column, Integer, ForeignKey, CheckConstraint, Enum
+from sqlalchemy.orm import relationship
 
 from enums.item_type import ItemType
 from models.base import Base
@@ -10,15 +12,21 @@ class CartItem(Base):
 
     id = Column(Integer, primary_key=True)
     cart_id = Column(Integer, ForeignKey("carts.id"), nullable=False)
+    cart = relationship("Cart", back_populates="cart_items")
     item_type = Column(Enum(ItemType), nullable=False)
     category_id = Column(Integer, ForeignKey('categories.id'), nullable=False)
+    category = relationship("Category", back_populates="cart_items")
     subcategory_id = Column(Integer, ForeignKey('subcategories.id'), nullable=False)
+    subcategory = relationship("Subcategory", back_populates="cart_items")
     quantity = Column(Integer, nullable=False)
 
     __table_args__ = (
         CheckConstraint('quantity > 0', name='check_quantity_positive'),
     )
 
+    def __repr__(self):
+        return f"CartItem ID: {self.id}"
+
 
 class CartItemDTO(BaseModel):
     id: int | None = None
@@ -27,3 +35,17 @@ class CartItemDTO(BaseModel):
     category_id: int | None = None
     subcategory_id: int | None = None
     quantity: int | None = None
+
+
+class CartItemAdmin(ModelView, model=CartItem):
+    column_sortable_list = [CartItem.id,
+                            CartItem.item_type,
+                            CartItem.quantity,
+                            CartItem.item_type]
+    column_exclude_list = [CartItem.category_id,
+                           CartItem.subcategory_id,
+                           CartItem.cart_id]
+    can_create = False
+    can_delete = False
+    can_edit = False
+    can_export = False

models/category.py

@@ -1,5 +1,7 @@
 from pydantic import BaseModel
+from sqladmin import ModelView
 from sqlalchemy import Integer, Column, String
+from sqlalchemy.orm import relationship
 
 from models.base import Base
 
@@ -10,9 +12,28 @@ class Category(Base):
     id = Column(Integer, primary_key=True)
     name = Column(String, nullable=False, unique=True, index=True)
     media_id = Column(String, nullable=False)
+    items = relationship("Item", back_populates="category")
+    cart_items = relationship("CartItem", back_populates="category")
+
+    def __repr__(self):
+        return self.name
 
 
 class CategoryDTO(BaseModel):
     id: int | None = None
     name: str | None = None
     media_id: str | None = None
+
+
+class CategoryAdmin(ModelView, model=Category):
+    name = "Category"
+    name_plural = "Categories"
+    column_searchable_list = [Category.name]
+    column_sortable_list = [Category.id, Category.name]
+    column_exclude_list = [Category.items,
+                           Category.media_id,
+                           Category.cart_items]
+    can_delete = False
+    can_edit = True
+    can_create = True
+    can_export = False