feat: initialize aivideo project
This commit is contained in:
53
backend/app/modules/pricing/service.py
Normal file
53
backend/app/modules/pricing/service.py
Normal file
@@ -0,0 +1,53 @@
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.common.errors.app_error import NotFoundAppError
|
||||
from app.models.entities import PricingRule
|
||||
from app.modules.pricing.repository import PricingRepository
|
||||
|
||||
|
||||
class PricingService:
|
||||
def __init__(self, db: Session) -> None:
|
||||
self.db = db
|
||||
self.repository = PricingRepository(db)
|
||||
|
||||
def list_rules(self) -> list[dict]:
|
||||
return [self.serialize(item) for item in self.repository.list_rules().all()]
|
||||
|
||||
def create_rule(self, payload) -> dict:
|
||||
item = PricingRule(**payload.model_dump())
|
||||
self.db.add(item)
|
||||
self.db.commit()
|
||||
self.db.refresh(item)
|
||||
return self.serialize(item)
|
||||
|
||||
def update_rule(self, rule_id: int, payload) -> dict:
|
||||
item = self.repository.get_rule(rule_id)
|
||||
if not item:
|
||||
raise NotFoundAppError("pricing rule not found", code=60004)
|
||||
for key, value in payload.model_dump().items():
|
||||
setattr(item, key, value)
|
||||
self.db.commit()
|
||||
return self.serialize(item)
|
||||
|
||||
def publish_rule(self, rule_id: int) -> dict:
|
||||
item = self.repository.get_rule(rule_id)
|
||||
if not item:
|
||||
raise NotFoundAppError("pricing rule not found", code=60004)
|
||||
item.status = 1
|
||||
self.db.commit()
|
||||
return self.serialize(item)
|
||||
|
||||
@staticmethod
|
||||
def serialize(item: PricingRule) -> dict:
|
||||
return {
|
||||
"id": item.id,
|
||||
"ruleName": item.rule_name,
|
||||
"videoModelId": item.video_model_id,
|
||||
"pointsPerSecond": item.points_per_second,
|
||||
"minimumPoints": item.minimum_points,
|
||||
"effectiveAt": item.effective_at.isoformat(),
|
||||
"expiredAt": item.expired_at.isoformat() if item.expired_at else None,
|
||||
"versionNo": item.version_no,
|
||||
"status": item.status,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user