⚙️feat: add Conversation Cleanup

Author: teloryfrozy

README.md

@@ -43,7 +43,7 @@ fastapi dev main.py
 
 | Feature                | Description                                              | Status               |
 | ---------------------- | -------------------------------------------------------- | -------------------- |
-| Conversation Cleanup   | Auto-delete conversations after x months of inactivity   | ⚒️ Under Refactoring |
+| Conversation Cleanup   | Auto-delete conversations after x months of inactivity   | ✅ Done |
 | Ad Refresh             | Automatic refreshing of listings                         | ⚒️ Under Refactoring |
 | Sales Analytics        | Export transactions, generate graphs & statistics        | 📝 To Be Done        |
 | Annual Reports         | Yearly compatible reporting system                       | 📝 To Be Done        |

backend/main.py

@@ -1,7 +1,7 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI
 from config.settings import ALLOW_ORIGINS, DEBUG_MODE
-from routers import ads_management, auth, follow_mass
+from routers import ads_management, auth, follow_mass, accounting
 from config.models import lifespan
 
 
@@ -12,6 +12,7 @@
 app.include_router(ads_management.router)
 app.include_router(auth.router)
 app.include_router(follow_mass.router)
+app.include_router(accounting.router)
 
 
 app.add_middleware(

backend/routers/accounting.py

@@ -0,0 +1,56 @@
+import datetime
+import re
+from fastapi import HTTPException, APIRouter, Depends
+from pydantic import BaseModel
+from sqlmodel import Session, select
+from constants import API_URL
+from utils import execute_request, get_vinted_headers
+from config.models import User, get_session
+from fastapi import APIRouter, Depends
+
+router = APIRouter(
+    prefix="/accounting",
+)
+
+
+class CleanConversations(BaseModel):
+    monthsToKeep: int
+
+
+@router.post("/clean-conversations")
+async def clean_conversations(
+    clean_conversations: CleanConversations,
+    session: Session = Depends(get_session),
+    headers: dict = Depends(get_vinted_headers),
+):
+    page = 1
+    nb_conversations_deleted = 0
+
+    response = execute_request(
+        "GET", f"{API_URL}inbox?page={page}?per_page=20", headers, session=session
+    )
+
+    total_pages = response.json()["pagination"]["total_pages"]
+
+    while page < total_pages:
+        response = execute_request(
+            "GET", f"{API_URL}inbox?page={page}?per_page=20", headers, session=session
+        )
+
+        for conversation in response.json()["conversations"]:
+            if (datetime.datetime.now() - datetime.datetime.strptime(conversation["updated_at"].split("+")[0], "%Y-%m-%dT%H:%M:%S")).days > clean_conversations.monthsToKeep * 30:
+                nb_conversations_deleted += 1
+                if not conversation["is_deletion_restricted"]:
+                    execute_request(
+                        "DELETE",
+                        f"{API_URL}conversations/{conversation['id']}",
+                        headers,
+                        session=session,
+                    )
+
+        page += 1
+
+    return {
+        "message": "Conversations cleaned",
+        "nnConversationsDeleted": nb_conversations_deleted,
+    }

backend/routers/ads_management.py

@@ -113,7 +113,7 @@ async def refresh_ads(
     while True:
         user = session.exec(select(User).where(User.id == 1)).first()
         url = f"{API_URL}wardrobe/{user.userId}/items?page={page}&per_page=20&order=revelance"
-        response = execute_request("get", url, headers)
+        response = execute_request("GET", url, headers)
 
         if response.status_code != 200:
             break
@@ -193,7 +193,7 @@ async def delete_sold_items(
     while True:
         user = session.exec(select(User).where(User.id == 1)).first()
         url = f"{API_URL}wardrobe/{user.userId}/items?page={page}&per_page=20&order=revelance"
-        response = execute_request("get", url, headers)
+        response = execute_request("GET", url, headers)
         nb_items_deleted = 0
 
         if response.status_code != 200:
@@ -236,7 +236,7 @@ async def delete_all_ads(
     while True:
         user = session.exec(select(User).where(User.id == 1)).first()
         url = f"{API_URL}wardrobe/{user.userId}/items?page={page}&per_page=20&order=revelance"
-        response = execute_request("get", url, headers)
+        response = execute_request("GET", url, headers)
         nb_items_deleted = 0
 
         if response.status_code != 200:

backend/routers/follow_mass.py

@@ -64,7 +64,7 @@ async def backup_followed_users(
 def toggle_follow(user_id: int, headers: dict) -> int:
     url = f"{API_URL}followed_users/toggle"
     data = {"user_id": user_id}
-    response = execute_request("post", url, headers, data)
+    response = execute_request("POST", url, headers, data)
     print(
         f"[RESULT] user_id: {user_id} CODE: {response.status_code} | REASON: {response.reason} | text: {response.text}"
     )

backend/utils.py

@@ -59,19 +59,19 @@ def execute_request(
     i = 0
 
     while i < MAX_RETRIES:
-        if method == "get":
+        if method == "GET":
             response = requests.get(url, headers=headers)
-        elif method == "post":
+        elif method == "POST":
             response = requests.post(url, headers=headers, json=data)
-        elif method == "put":
-            response = requests.put(url, headers=headers, json=data)
-        elif method == "delete":
+        elif method == "PATCH":
+            response = requests.patch(url, headers=headers, json=data)
+        elif method == "DELETE":
             response = requests.delete(url, headers=headers, json=data)
 
         if response.status_code != 200:
             refresh_access_token(headers, session)
 
-            headers = get_vinted_headers()
+            headers = get_vinted_headers(session)
             print(
                 f"Status code: {response.status_code}",
                 f"Reason: {response.reason}",
@@ -89,10 +89,14 @@ def execute_request(
 
 
 def refresh_access_token(headers: dict, session: Session):
-    response = execute_request(
-        "post", f"https://www.vinted.fr/web/api/auth/refresh", headers
+    response = requests.post(
+        f"https://www.vinted.fr/web/api/auth/refresh", headers=headers
     )
-    print("Token vient d'être rafraîchi", response.json()["access_token"][:-5])
+    if response.status_code != 200:
+        raise HTTPException(
+            status_code=401,
+            detail="Refresh token refused by Vinted server. Please update your token.",
+        )
     auth = session.exec(select(User).order_by(User.id.desc())).first()
     auth.vinted_access_token = response.json()["access_token"]
     session.add(auth)