sheets.py

import gspread
from gspread_dataframe import get_as_dataframe

from oauth2client.service_account import ServiceAccountCredentials


class GoogleSheets:
    def __init__(self, credentials):
        scope = ["https://www.googleapis.com/auth/drive"]
        self.gc = gspread.authorize(
            ServiceAccountCredentials.from_json_keyfile_dict(
                credentials, scope
            )
        )

    def open_by_key(self, key, wks=0, df=True):
        sht = self.gc.open_by_key(key)

        if wks is None:
            return sht

        if isinstance(wks, int):
            worksheet = sht.get_worksheet(wks)
        else:
            worksheet = sht.worksheet(wks)

        if df is not True:
            return worksheet

        return get_as_dataframe(worksheet)