Get the users that likes a post

Author: Maron09

authentication_middleware.py

@@ -0,0 +1,19 @@
+from fastapi import Request
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.responses import JSONResponse
+
+class AuthenticationMiddleware(BaseHTTPMiddleware):
+    def __init__(self, app, instagram_service):
+        super().__init__(app)
+        self.instagram_service = instagram_service
+
+    async def dispatch(self, request: Request, call_next):
+        if request.url.path == "/v1/api/media_likers":
+            username = request.query_params.get("username")
+            if not username or not self.instagram_service.get_client(username):
+                return JSONResponse(
+                    status_code=401,
+                    content={"message": "Authentication required. Please login first using the /v1/api/login endpoint."}
+                )
+        response = await call_next(request)
+        return response

instagram_service.py

@@ -0,0 +1,33 @@
+from instagrapi import Client
+from instagrapi.types import UserShort
+from typing import Optional
+
+class InstagramService:
+    def __init__(self):
+        self.clients = {}
+
+    def login(self, username: str, password: str) -> str:
+        client = Client()
+        client.login(username, password)
+        self.clients[username] = client
+        return username
+
+    def get_client(self, username: str) -> Optional[Client]:
+        return self.clients.get(username)
+
+    def get_media_likers(self, username: str, media_id: str):
+        client = self.get_client(username)
+        if client:
+            likers = client.media_likers(media_id)
+            return [self.user_short_to_dict(liker) for liker in likers]
+        else:
+            return None
+
+    def user_short_to_dict(self, user_short: UserShort):
+        return {
+            "pk": user_short.pk,
+            "username": user_short.username,
+            "full_name": user_short.full_name,
+        }
+
+instagram_service = InstagramService()

main.py

@@ -1,18 +1,41 @@
 from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import JSONResponse
 from starlette.middleware import Middleware
-from middleware import TimeoutMiddleware
+from time_middleware import TimeoutMiddleware
 from typing import Optional
+from pydantic import BaseModel
 
 # Import the functions from the other scripts
 from profile_data import retrieve_profile_info
 from user_post import retrieve_post_metrics
 from media import get_post_details
 
+# Import the Instagram service
+from instagram_service import instagram_service
+from authentication_middleware import AuthenticationMiddleware
+
+class Credentials(BaseModel):
+    username: str
+    password: str
+
 app = FastAPI(middleware=[
-    Middleware(TimeoutMiddleware, timeout=600)
+    Middleware(TimeoutMiddleware, timeout=600),
+    Middleware(AuthenticationMiddleware, instagram_service=instagram_service)
 ])
 
+
+@app.post('/v1/api/login')
+def login(credentials: Credentials):
+    try:
+        username = instagram_service.login(credentials.username, credentials.password)
+        return JSONResponse(content={"meassege": "Login Successful", "username": username})
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=str(e)
+        )
+
+
 @app.get('/v1/api/profile')
 def profile_info(username: str = Query(..., description="Username of the profile to retrieve")):
     try:
@@ -30,8 +53,8 @@ def profile_info(username: str = Query(..., description="Username of the profile
 @app.get('/v1/api/user_posts')
 def user_posts(
     username: str = Query(..., description="Username to retrieve posts for"),
-    from_date: Optional[str] = Query(None, description="Start date for post retrieval (YYYY-MM-DD)"),
-    to_date: Optional[str] = Query(None, description="End date for post retrieval (YYYY-MM-DD)")
+    from_date: Optional[str] = Query(None, description="Start date for post retrieval (YYYY-MM-DD), Optional"),
+    to_date: Optional[str] = Query(None, description="End date for post retrieval (YYYY-MM-DD), Optional")
 ):
     try:
         posts = retrieve_post_metrics(username, from_date, to_date)
@@ -57,6 +80,29 @@ def media_posts(url: str = Query(..., description="URL of the media to retrieve
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
+
+@app.get('/v1/api/media_likers')
+def media_likers(
+    media_id: str = Query(..., description="media_id = {post_id}_{owner_id} to retrieve likers for the post"),
+    username: str = Query(..., description="Username of the logged-in user")
+):
+    try:
+        likers = instagram_service.get_media_likers(username, media_id)
+        if not likers:
+            raise HTTPException(
+                status_code=404,
+                detail="Likers for this media not found"
+            )
+        return JSONResponse(content=likers)
+    except HTTPException as e:
+        raise e
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=str(e)
+        )
+
 if __name__ == '__main__':
     import uvicorn
     uvicorn.run(app, host="127.0.0.1", port=8000, log_level="debug")

profile_data.py

@@ -24,6 +24,7 @@ def retrieve_profile_info(username: str):
         # Access profile details
         profile_info = {
             'user_info':{
+                'uniqueID': profile.userid,
                 'Username': profile.username,
                 'Full_Name': profile.full_name,
                 'Followers': profile.followers,