Initial commit

backend/app/services/post_service.py

@@ -0,0 +1,88 @@
+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()