mirror of
https://gitverse.ru/kpa39l/telepost.git
synced 2026-09-29 18:05:08 +00:00
88 lines
3.3 KiB
Python
88 lines
3.3 KiB
Python
from typing import List, Optional
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy import and_, or_
|
|
from ..models.post import Post
|
|
from ..schemas.post import PostCreate, PostUpdate
|
|
from datetime import datetime
|
|
|
|
|
|
class PostService:
|
|
def __init__(self, db_session: Session):
|
|
self.db = db_session
|
|
|
|
def get_post(self, post_id: int, owner_id: int) -> Optional[Post]:
|
|
"""Получить пост по ID и владельцу"""
|
|
return self.db.query(Post).filter(
|
|
and_(Post.id == post_id, Post.owner_id == owner_id)
|
|
).first()
|
|
|
|
def get_posts(self, owner_id: int, skip: int = 0, limit: int = 100) -> List[Post]:
|
|
"""Получить список постов пользователя"""
|
|
return self.db.query(Post).filter(Post.owner_id == owner_id)\
|
|
.order_by(Post.created_at.desc())\
|
|
.offset(skip)\
|
|
.limit(limit)\
|
|
.all()
|
|
|
|
def create_post(self, post_data: PostCreate, owner_id: int) -> Post:
|
|
"""Создать новый пост"""
|
|
db_post = Post(
|
|
title=post_data.title,
|
|
content=post_data.content,
|
|
scheduled_time=post_data.scheduled_time,
|
|
status=post_data.status,
|
|
owner_id=owner_id
|
|
)
|
|
self.db.add(db_post)
|
|
self.db.commit()
|
|
self.db.refresh(db_post)
|
|
return db_post
|
|
|
|
def update_post(self, post_id: int, post_data: PostUpdate, owner_id: int) -> Optional[Post]:
|
|
"""Обновить пост"""
|
|
db_post = self.get_post(post_id, owner_id)
|
|
if not db_post:
|
|
return None
|
|
|
|
# Обновляем только те поля, которые предоставлены
|
|
update_data = post_data.dict(exclude_unset=True)
|
|
|
|
for field, value in update_data.items():
|
|
setattr(db_post, field, value)
|
|
|
|
# Обновляем время последнего изменения
|
|
db_post.updated_at = datetime.utcnow()
|
|
|
|
# Если статус меняется на published и published_at еще не установлен, устанавливаем его
|
|
if post_data.status == "published" and not db_post.published_at:
|
|
db_post.published_at = datetime.utcnow()
|
|
|
|
self.db.commit()
|
|
self.db.refresh(db_post)
|
|
return db_post
|
|
|
|
def delete_post(self, post_id: int, owner_id: int) -> bool:
|
|
"""Удалить пост"""
|
|
db_post = self.get_post(post_id, owner_id)
|
|
if not db_post:
|
|
return False
|
|
|
|
self.db.delete(db_post)
|
|
self.db.commit()
|
|
return True
|
|
|
|
def get_scheduled_posts(self) -> List[Post]:
|
|
"""Получить посты, готовые к публикации (статус scheduled и время наступило)"""
|
|
current_time = datetime.utcnow()
|
|
return self.db.query(Post).filter(
|
|
and_(
|
|
Post.status == "scheduled",
|
|
Post.scheduled_time <= current_time
|
|
)
|
|
).all()
|
|
|
|
def get_posts_by_status(self, owner_id: int, status: str) -> List[Post]:
|
|
"""Получить посты по статусу"""
|
|
return self.db.query(Post).filter(
|
|
and_(Post.owner_id == owner_id, Post.status == status)
|
|
).all() |