journal_impact_factor.py

from logger import print_error, print_warn, print_info
import gspread
from oauth2client.service_account import ServiceAccountCredentials

# Define the scope and authenticate
scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
creds = ServiceAccountCredentials.from_json_keyfile_name("./google-credentials.json", scope)
client = gspread.authorize(creds)

# Open the Google Sheet by URL
sheet_url = "https://docs.google.com/spreadsheets/d/1lP75APkxXAgT8aobV4UjTR51BpX9Ee0wgYA7tTd-zrM/edit?gid=0"
sheet = client.open_by_url(sheet_url).sheet1

def load_impact_factor():
    """
    Load the impact factor data from the Google Sheet and return it as a list of dictionaries.
    """
    # Get all values in columns A and B
    journal_names = sheet.col_values(1) # Column A (Journal names)
    impact_factors = sheet.col_values(2) # Column B (Impact factors)

    # Combine the two lists into a list of dictionaries
    impact_factor_data = []
    for journal_name, impact_factor in zip(journal_names, impact_factors):
        impact_factor_data.append({"journal_name": journal_name, "impact_factor": impact_factor})

    return impact_factor_data

def add_impact_factor(journal_name, impact_factor):
    """
    Add a new journal name and impact factor to the Google Sheet.
    """
    sheet.append_row([journal_name, impact_factor])