File size: 2,252 Bytes
0af0679 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
# tools.py
import os
import csv
import json
import base64
from dotenv import load_dotenv
from datetime import datetime
try:
import gspread
from google.oauth2.service_account import Credentials
GOOGLE_SHEETS_AVAILABLE = True
except ImportError:
GOOGLE_SHEETS_AVAILABLE = False
CSV_FILE = "user_interest.csv"
SHEET_NAME = "UserInterest"
def _get_google_credentials():
"""
Loads Google credentials either from local file or HF Spaces secret.
Returns a ServiceAccountCredentials object.
"""
load_dotenv(override=True)
scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
google_creds_json = os.getenv("GOOGLE_CREDENTIALS_JSON")
if google_creds_json:
json_str = base64.b64decode(google_creds_json).decode('utf-8')
creds_dict = json.loads(json_str)
creds = Credentials.from_service_account_info(creds_dict, scopes=scope)
print("[info] Loaded Google credentials from environment.")
return creds
raise RuntimeError("Google credentials not found.")
def _save_to_google_sheets(email, name, notes):
creds = _get_google_credentials()
client = gspread.authorize(creds)
sheet = client.open(SHEET_NAME).sheet1
row = [datetime.today().strftime('%Y-%m-%d %H:%M'), email, name, notes]
sheet.append_row(row)
print(f"[Google Sheets] Recorded: {email}, {name}")
def _save_to_csv(email, name, notes):
file_exists = os.path.isfile(CSV_FILE)
with open(CSV_FILE, mode='a', newline='', encoding='utf-8') as f:
writer = csv.writer(f)
if not file_exists:
writer.writerow(["Timestamp", "Email", "Name", "Notes"])
writer.writerow([datetime.today().strftime('%Y-%m-%d %H:%M'), email, name, notes])
print(f"[CSV] Recorded: {email}, {name}")
def _record_user_details(email, name="Name not provided", notes="Not provided"):
try:
if GOOGLE_SHEETS_AVAILABLE:
_save_to_google_sheets(email, name, notes)
else:
raise ImportError("gspread not installed.")
except Exception as e:
print(f"[Warning] Google Sheets write failed, using CSV. Reason: {e}")
_save_to_csv(email, name, notes)
return {"recorded": "ok"}
|