commentposter.py

import os
import time
import json
import sys
import googleapiclient.errors
from utilities import Utilities
import google_auth_oauthlib.flow as google_auth
from googleapiclient.discovery import build as get_youtube_api
from itertools import cycle
from structlog import get_logger

spinner = cycle("\\|/-")
log = get_logger()


class CommentPoster:
    utils = None

    def __init__(self):
        self.utils = self.utils = Utilities.get_instance()
        scopes = ["https://www.googleapis.com/auth/youtube.force-ssl"]
        api_service_name = "youtube"
        api_version = "v3"
        self.class_name = self.__class__.__name__

        # Get credentials and create an API client
        flow = google_auth.InstalledAppFlow.from_client_secrets_file(os.getenv("CLIENT_SECRET_PATH"), scopes)
        credentials = flow.run_console()
        self.youtube = get_youtube_api(api_service_name, api_version, credentials=credentials)

    def post_comment(self, comment_body):
        request = self.youtube.comments().insert(part="snippet", body=comment_body)

        try:
            response = request.execute()
        except googleapiclient.errors.HttpError as e:
            log.error(self.class_name, error=e)
            return

        log.info(self.class_name, response_msg=response, response_type=type(response))
        return response

    def run(self):
        while True:
            time.sleep(1)
            with open("database/topost.json") as post_file:
                try:
                    responses_to_post = json.load(post_file)
                except json.decoder.JSONDecodeError:
                    responses_to_post = []

            if responses_to_post:
                log.info(self.class_name, responses_to_post=responses_to_post)
            else:
                print("\b" + next(spinner), end="")
                sys.stdout.flush()

            if responses_to_post:
                body = responses_to_post.pop()

                self.post_comment(body)

            with open("database/topost.json", "w") as post_file:
                # we modified the queue, put the rest back, if any
                json.dump(responses_to_post, post_file, indent="\t")


if __name__ == "__main__":
    commentPoster = CommentPoster()
    commentPoster.run()