batch_main.py

from dotenv import load_dotenv

from gmail_api.gmail_service import GmailService
from pipelines.pipeline import pipeline
from utils.configuration import Config
from utils.db_utils import authenticate_gmail, fetch_users, insert_report
from utils.token_usage_counter import TokenUsageCounter


def main():
    load_dotenv()
    Config.load()

    # 유저 테이블 불러오기
    users = fetch_users()

    # access token, refresh token 가져와서 service 객체 선언하기
    for user in users:
        try:
            # if user["id"] != 11:
            #     continue

            service = authenticate_gmail(user)
            Config.user_upstage_api_key = user["upstage_api_key"]
            # GmailService 인스턴스 생성
            gmail_service = GmailService(service)

            json_checklist, report = pipeline(gmail_service)
            print(f"============ FINAL REPORT of {user['id']} =============")
            print(report)
            print("=======================================================")

            if Config.config["token_tracking"]:
                TokenUsageCounter.plot_token_cost()

            insert_report(user["id"], report, json_checklist)

        except Exception as e:
            print(e)


if __name__ == "__main__":
    main()