Refactor post_blog.py to improve image handling and update OAuth credentials management

get_refresh_token.py

@@ -0,0 +1,30 @@
+import httplib2
+from oauth2client.client import flow_from_clientsecrets
+from oauth2client.file import Storage
+from oauth2client.tools import run_flow
+from googleapiclient import discovery
+
+# Start the OAuth flow to retrieve credentials
+def authorize_credentials():
+    CLIENT_SECRET = 'client_secret_6.json'
+    SCOPE = 'https://www.googleapis.com/auth/blogger'
+    STORAGE = Storage('credentials.storage.json')
+    credentials = STORAGE.get()
+    if credentials is None or credentials.invalid:
+        flow = flow_from_clientsecrets(CLIENT_SECRET, scope=SCOPE)
+        http = httplib2.Http()
+        credentials = run_flow(flow, STORAGE, http=http)
+    return credentials
+
+def getBloggerService():
+    credentials = authorize_credentials()
+    http = credentials.authorize(httplib2.Http())
+    discoveryUrl = ('https://blogger.googleapis.com/$discovery/rest?version=v3')
+    service = discovery.build('blogger', 'v3', http=http, discoveryServiceUrl=discoveryUrl)
+    return service
+
+if __name__ == '__main__':
+    try:
+        getBloggerService()
+    except Exception as e:
+        print(str(e))

main.py

@@ -8,7 +8,6 @@ from post_blog import post_blog
 # Load environment variables
 dotenv.load_dotenv()
 summarizer_api_key = os.getenv("SUMMARIZER_API_KEY")
-mail_api = os.getenv("MAIL_API")
 access_key = os.getenv("ACCESS_KEY")
 
 def paper_data(paper_data):

post_blog.py

@@ -8,27 +8,24 @@ import mistune
 
 dotenv.load_dotenv()
 access_key = os.getenv('ACCESS_KEY')
-client_id = os.getenv('CLIENT_ID')
-client_secret = os.getenv('CLIENT_SECRET')
-refresh_token = os.getenv('REFRESH_TOKEN')
+client_id = os.getenv('CLIENT_ID_1')
+client_secret = os.getenv('CLIENT_SECRET_1')
+refresh_token = os.getenv('REFRESH_TOKEN_1')
 blog_id = os.getenv('BLOG_ID')
 
 def extract_summary(text):
-    match = re.search(r"Summary\n(.*?)\nHighlights", text, re.DOTALL)
+    text = text.replace("#", "").strip().lower()
+    match = re.search(r"summary(.*?)highlights", text.replace("#", ""), re.DOTALL)
     if match:
         return match.group(1).replace("#", "").replace("\n", "").strip()
     return None
 
 def generate_image(title, summary, category):
     extracted_summary = extract_summary(summary)
-    print(extracted_summary)
+    if extracted_summary is None:
+        extracted_summary = title
     url = f"https://image.pollinations.ai/prompt/(({category}))%20{title}%20%3A%20{extracted_summary}?width=1280&height=720&seed=623862700&nologo=true&model=turbo"
-    data = requests.get(url).content
-    with open("image.png", "wb") as file:
-        file.write(data)
-    encoded_image = base64.b64encode(data)
-    image =  f"data:image/png;base64,{encoded_image.decode('utf-8')}"
-    return image
+    return url
 
 def generate_post_html(title, summary, mindmap, category, citation):
     image = generate_image(title, summary, category)
@@ -67,7 +64,7 @@ def generate_post_html(title, summary, mindmap, category, citation):
         </p>
     </div>
     """
-    return post
+    return post, image
 
 def sanitize_citation(citation):
     pattern = r"(https://doi\.org/\S+)"
@@ -82,18 +79,18 @@ def sanitize_citation(citation):
 def create_post(title, category, summary, mindmap, citation):
     post_title = f"{title}"
     post_category = f"{category}"
-    post_body = generate_post_html(title, summary, mindmap, category, sanitize_citation(citation))
-    return post_title, post_category, post_body
+    post_body, post_image = generate_post_html(title, summary, mindmap, category, sanitize_citation(citation))
+    return post_title, post_category, post_body, post_image
 
-def post_post(title, category, body):
+def post_post(title, category, body, image, i):
     try:
         data = requests.post(
                 url='https://oauth2.googleapis.com/token',
                 data={
                     'grant_type': 'refresh_token',
-                    'client_secret': client_secret,
-                    'refresh_token': refresh_token,
-                    'client_id': client_id,
+                    'client_secret': os.getenv(f'CLIENT_SECRET_{i}'),
+                    'refresh_token': os.getenv(f'REFRESH_TOKEN_{i}'),
+                    'client_id': os.getenv(f'CLIENT_ID_{i}'),
                 },
             ).json()
         url = f"https://blogger.googleapis.com/v3/blogs/{blog_id}/posts"
@@ -106,14 +103,21 @@ def post_post(title, category, body):
             "blog": {
                 "id": blog_id
             },
+            "images": [
+                {
+                "url": image
+                }
+            ],
             "title": title,
             "content": body,
             "labels": [category, "recent"]
         }
         data = requests.post(url, headers=headers, json=post_data).json()
-        print(data)
-        time.sleep(2 * 60)
-        return True
+        print(data['status'])
+        if data['status'] == 'LIVE':
+            return True
+        else:
+            return False
     except Exception as e:
         print('An error occurred:', str(e))
         return False
@@ -122,8 +126,13 @@ def post_blog(title, category, summary, mindmap, citation, uaccess_key):
     if uaccess_key != access_key:
         return False
     try:
-        post_title, post_category, post_body = create_post(title, category, summary, mindmap, citation)
-        status = post_post(post_title, post_category, post_body)
+        post_title, post_category, post_body, post_image = create_post(title, category, summary, mindmap, citation)
+        i = 1
+        status = False
+        while (i <= 6 and not status):
+            print(f'Posting... Tries: {i}')
+            status = post_post(post_title, post_category, post_body, post_image, i)
+            i += 1
         if status:
             print('Post created successfully')
             return True