Create app/services/suggestion_service.py

app/services/suggestion_service.py

@@ -0,0 +1,40 @@
+from typing import List
+from sqlalchemy.orm import Session
+
+from app.models.suggestion import Suggestion
+from app.schemas.suggestion import SuggestionCreate
+
+
+def create_suggestion(db: Session, data: SuggestionCreate) -> Suggestion:
+    translations_str = ",".join(data.translations) if data.translations else ""
+    suggestion = Suggestion(
+        headword_zomi=data.headwordZomi,
+        headword_english=data.headwordEnglish,
+        translations=translations_str,
+        part_of_speech=data.partOfSpeech,
+        definition=data.definition,
+        contributor_id=data.contributorId,
+    )
+    db.add(suggestion)
+    db.commit()
+    db.refresh(suggestion)
+    return suggestion
+
+
+def get_suggestions_by_user(db: Session, user_id: str) -> List[Suggestion]:
+    return db.query(Suggestion).filter(Suggestion.contributor_id == user_id).all()
+
+
+def vote_on_suggestion(db: Session, suggestion_id: int, is_upvote: bool) -> Suggestion | None:
+    suggestion = db.query(Suggestion).filter(Suggestion.id == suggestion_id).first()
+    if not suggestion:
+        return None
+
+    if is_upvote:
+        suggestion.upvotes += 1
+    else:
+        suggestion.downvotes += 1
+
+    db.commit()
+    db.refresh(suggestion)
+    return suggestion