85 lines
4.5 KiB
Python
85 lines
4.5 KiB
Python
"""Business Information Service for ALwrity backend."""
|
|
from sqlalchemy.orm import Session
|
|
from models.user_business_info import UserBusinessInfo
|
|
from models.business_info_request import BusinessInfoRequest, BusinessInfoResponse
|
|
from services.database import get_db
|
|
from loguru import logger
|
|
from typing import Optional
|
|
|
|
logger.info("🔄 Loading BusinessInfoService...")
|
|
|
|
class BusinessInfoService:
|
|
def __init__(self):
|
|
logger.info("🆕 Initializing BusinessInfoService...")
|
|
|
|
def save_business_info(self, business_info: BusinessInfoRequest) -> BusinessInfoResponse:
|
|
db: Session = next(get_db())
|
|
logger.debug(f"Attempting to save business info for user_id: {business_info.user_id}")
|
|
|
|
# Check if business info already exists for this user
|
|
existing_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.user_id == business_info.user_id).first()
|
|
|
|
if existing_info:
|
|
logger.info(f"Existing business info found for user_id {business_info.user_id}, updating it.")
|
|
existing_info.business_description = business_info.business_description
|
|
existing_info.industry = business_info.industry
|
|
existing_info.target_audience = business_info.target_audience
|
|
existing_info.business_goals = business_info.business_goals
|
|
db.commit()
|
|
db.refresh(existing_info)
|
|
logger.success(f"Updated business info for user_id {business_info.user_id}, ID: {existing_info.id}")
|
|
return BusinessInfoResponse(**existing_info.to_dict())
|
|
else:
|
|
logger.info(f"No existing business info for user_id {business_info.user_id}, creating new entry.")
|
|
db_business_info = UserBusinessInfo(
|
|
user_id=business_info.user_id,
|
|
business_description=business_info.business_description,
|
|
industry=business_info.industry,
|
|
target_audience=business_info.target_audience,
|
|
business_goals=business_info.business_goals
|
|
)
|
|
db.add(db_business_info)
|
|
db.commit()
|
|
db.refresh(db_business_info)
|
|
logger.success(f"Saved new business info for user_id {business_info.user_id}, ID: {db_business_info.id}")
|
|
return BusinessInfoResponse(**db_business_info.to_dict())
|
|
|
|
def get_business_info(self, business_info_id: int) -> Optional[BusinessInfoResponse]:
|
|
db: Session = next(get_db())
|
|
logger.debug(f"Retrieving business info by ID: {business_info_id}")
|
|
business_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.id == business_info_id).first()
|
|
if business_info:
|
|
logger.debug(f"Found business info for ID: {business_info_id}")
|
|
return BusinessInfoResponse(**business_info.to_dict())
|
|
logger.warning(f"No business info found for ID: {business_info_id}")
|
|
return None
|
|
|
|
def get_business_info_by_user(self, user_id: int) -> Optional[BusinessInfoResponse]:
|
|
db: Session = next(get_db())
|
|
logger.debug(f"Retrieving business info by user ID: {user_id}")
|
|
business_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.user_id == user_id).first()
|
|
if business_info:
|
|
logger.debug(f"Found business info for user ID: {user_id}")
|
|
return BusinessInfoResponse(**business_info.to_dict())
|
|
logger.warning(f"No business info found for user ID: {user_id}")
|
|
return None
|
|
|
|
def update_business_info(self, business_info_id: int, business_info: BusinessInfoRequest) -> Optional[BusinessInfoResponse]:
|
|
db: Session = next(get_db())
|
|
logger.debug(f"Updating business info for ID: {business_info_id}")
|
|
db_business_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.id == business_info_id).first()
|
|
if db_business_info:
|
|
db_business_info.business_description = business_info.business_description
|
|
db_business_info.industry = business_info.industry
|
|
db_business_info.target_audience = business_info.target_audience
|
|
db_business_info.business_goals = business_info.business_goals
|
|
db.commit()
|
|
db.refresh(db_business_info)
|
|
logger.success(f"Updated business info for ID: {business_info_id}")
|
|
return BusinessInfoResponse(**db_business_info.to_dict())
|
|
logger.warning(f"No business info found to update for ID: {business_info_id}")
|
|
return None
|
|
|
|
business_info_service = BusinessInfoService()
|
|
logger.info("✅ BusinessInfoService loaded successfully!")
|