Upload 3 files

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3.11.5-slim
+
+WORKDIR /app
+
+COPY . /app
+
+RUN pip install -r requirements.txt

requirements.txt

@@ -0,0 +1,39 @@
+annotated-types==0.6.0
+anyio==3.7.1
+cachetools==5.3.2
+certifi==2023.11.17
+charset-normalizer==3.3.2
+click==8.1.7
+ecdsa==0.18.0
+fastapi==0.104.1
+google-api-core==2.14.0
+google-api-python-client==2.109.0
+google-auth==2.25.1
+google-auth-httplib2==0.1.1
+google-auth-oauthlib==1.1.0
+googleapis-common-protos==1.61.0
+gunicorn==21.2.0
+h11==0.14.0
+httplib2==0.22.0
+idna==3.6
+jose==1.0.0
+oauthlib==3.2.2
+packaging==23.2
+protobuf==4.25.1
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
+pydantic==2.5.2
+pydantic_core==2.14.5
+pyparsing==3.1.1
+python-jose==3.3.0
+python-multipart==0.0.6
+requests==2.31.0
+requests-oauthlib==1.3.1
+rsa==4.9
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+typing_extensions==4.8.0
+uritemplate==4.1.1
+urllib3==2.1.0
+uvicorn==0.24.0.post1

test.py

@@ -0,0 +1,105 @@
+from fastapi import FastAPI, Depends
+from fastapi.security import OAuth2PasswordBearer
+import requests
+from jose import jwt
+import webbrowser
+import base64
+
+app = FastAPI()
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
+
+
+
+
+
+# Replace these with your own values from the Google Developer Console
+GOOGLE_CLIENT_ID = ""
+GOOGLE_CLIENT_SECRET = ""
+GOOGLE_REDIRECT_URI = ""
+
+
+
+@app.get("/login/google")
+async def login_google():
+    # oauth_url = f"https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={GOOGLE_CLIENT_ID}&redirect_uri={GOOGLE_REDIRECT_URI}&scope=openid%20profile%20email&access_type=offline"
+    
+    #Below is the URL to prompt the user to login to his specified gmail account and also give a readonly access to his gmail
+    oauth_url = f"https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={GOOGLE_CLIENT_ID}&redirect_uri={GOOGLE_REDIRECT_URI}&scope=openid%20profile%20email%20https://www.googleapis.com/auth/gmail.readonly&access_type=offline"
+
+    # webbrowser.open(oauth_url)
+    return {
+        "url": f"https://accounts.google.com/o/oauth2/auth?response_type=code&client_id={GOOGLE_CLIENT_ID}&redirect_uri={GOOGLE_REDIRECT_URI}&scope=openid%20profile%20email&access_type=offline"
+    }
+
+@app.get("/auth/google")
+async def auth_google(code: str):
+    #This code is basically the authorization code and this authorization code helps us to get the access token with the required scopes that we have set .
+    #We require the gmail.readonly scopes that requires verification of our application and all.
+    token_url = "https://accounts.google.com/o/oauth2/token"
+    print(code)
+    data = {
+        "code": code,
+        "client_id": GOOGLE_CLIENT_ID,
+        "client_secret": GOOGLE_CLIENT_SECRET,
+        "redirect_uri": GOOGLE_REDIRECT_URI,
+        "grant_type": "authorization_code",
+    }
+    response = requests.post(token_url, data=data)
+    access_token = response.json().get("access_token")
+    print(response.json())
+    user_info = requests.get("https://www.googleapis.com/oauth2/v1/userinfo", headers={"Authorization": f"Bearer {access_token}"})
+    jobs_query = "subject:New application:ios"
+    gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={jobs_query}&maxResults=110"
+    gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token}"})
+    messages = gmail_response.json().get("messages", [])
+    print(messages)
+    print("Printing gmail response")
+    print(gmail_response.json())
+    # Fetch attachments from the first message
+    attachments = []
+    for i,message in enumerate(messages) :
+        print(i)
+        print(message)
+
+        if message:
+            message_id = message.get("id")
+            print(message_id)
+            if message_id:
+                message_url = f"https://www.googleapis.com/gmail/v1/users/me/messages/{message_id}"
+                message_response = requests.get(message_url, headers={"Authorization": f"Bearer {access_token}"})
+                message_data = message_response.json()
+
+                # Check for parts in the message payload
+                if "payload" in message_data and "parts" in message_data["payload"]:
+                    for part in message_data["payload"]["parts"]:
+                        if "body" in part and "attachmentId" in part["body"]:
+                            attachment_id = part["body"]["attachmentId"]
+                            attachment_url = f"https://www.googleapis.com/gmail/v1/users/me/messages/{message_id}/attachments/{attachment_id}"
+                            attachment_response = requests.get(attachment_url, headers={"Authorization": f"Bearer {access_token}"})
+                            attachment_data = attachment_response.json()
+                            data = attachment_data.get("data")
+                            filename = part.get("filename", "untitled.txt")
+
+                            if data:
+                                # Decode base64-encoded attachment data
+                                attachment_content = base64.urlsafe_b64decode(data.encode("UTF-8"))
+
+                                # Save the attachment to a file
+                                save_path = f"/Users/omkarmalpure/Documents/Gmail_API/attachments/{filename}"
+                                with open(save_path, "wb") as file:
+                                    file.write(attachment_content)
+
+                                attachments.append(save_path)
+
+    print(attachments)
+    print(len(attachments))
+    return user_info.json()
+
+@app.get("/token")
+async def get_token(token: str = Depends(oauth2_scheme)):
+    return jwt.decode(token, GOOGLE_CLIENT_SECRET, algorithms=["HS256"])
+
+# if __name__ == "__main__":
+#     import uvicorn
+
+#     uvicorn.run(app, host="0.0.0.0", port=8000)