Примеры использования #

import requests
import os
import sys

API_KEY = os.getenv('ABLABZ_API_KEY')
BASE_URL = 'https://workbench.ab-labz.com/api/v1'
FILE_PATH = 'dataset.csv'
EXPERIMENT_ID = 'homepage_test'

def upload_dataset(file_path: str, experiment_id: str) -> bool:
    """
    Загружает датасет в AB-Labz Workbench через API.
    
    Args:
        file_path: путь к CSV файлу
        experiment_id: идентификатор эксперимента
        
    Returns:
        True если загрузка успешна
        
    Raises:
        requests.HTTPError: при ошибках API
        FileNotFoundError: если файл не найден
    """
    if not os.path.exists(file_path):
        raise FileNotFoundError(f"Файл не найден: {file_path}")
    
    file_size_mb = os.path.getsize(file_path) / (1024 * 1024)
    print(f"🚀 Загрузка датасета для эксперимента '{experiment_id}'")
    print(f"📦 Размер файла: {file_size_mb:.2f} МБ")
    
    # 1. Получить URL для загрузки
    print("⏳ Получение ссылки для загрузки...")
    response = requests.post(
        f'{BASE_URL}/datasets/initiate/',
        headers={'Authorization': f'Bearer {API_KEY}'},
        json={'experiment_id': experiment_id}
    )
    response.raise_for_status()
    data = response.json()
    
    upload_url = data['upload_url']
    print(f"✅ Ссылка получена (действительна {data['expires_in']} сек)")
    
    # 2. Загрузить файл
    print("⬆️  Загрузка файла в хранилище...")
    with open(file_path, 'rb') as f:
        upload_response = requests.put(
            upload_url,
            headers={'Content-Type': 'text/csv'},
            data=f
        )
        upload_response.raise_for_status()
    
    print("✅ Файл успешно загружен!")
    print(f"📊 Датасет доступен в UI: Подготовка данных → Получить по API")
    return True

if __name__ == '__main__':
    try:
        upload_dataset(FILE_PATH, EXPERIMENT_ID)
    except requests.HTTPError as e:
        print(f"❌ Ошибка API: {e.response.status_code}")
        print(f"   {e.response.text}")
        sys.exit(1)
    except FileNotFoundError as e:
        print(f"❌ {e}")
        sys.exit(1)
    except Exception as e:
        print(f"❌ Неожиданная ошибка: {e}")
        sys.exit(1)

import requests

API_KEY = 'abn_xxxxxxxxxxxxxxxxxxxxx'
BASE_URL = 'https://workbench.ab-labz.com/api/v1'
FILE_PATH = 'dataset.csv'

# 1. Получить URL для загрузки
response = requests.post(
    f'{BASE_URL}/datasets/initiate/',
    headers={'Authorization': f'Bearer {API_KEY}'},
    json={'experiment_id': 'homepage_test'}
)
upload_url = response.json()['upload_url']

# 2. Загрузить файл
with open(FILE_PATH, 'rb') as f:
    requests.put(upload_url, data=f, headers={'Content-Type': 'text/csv'})

print("✅ Датасет загружен!")

#!/usr/bin/env python3
# upload_to_ablabz.py

import os
import requests
import pandas as pd
from datetime import datetime
import logging

# Настройка логирования
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s'
)

API_KEY = os.getenv('ABLABZ_API_KEY')
BASE_URL = 'https://workbench.ab-labz.com/api/v1'
EXPERIMENT_ID = 'homepage_test'

def export_data_from_db():
    """Экспорт данных из БД в CSV"""
    # Пример с PostgreSQL
    import psycopg2
    
    conn = psycopg2.connect(
        host="localhost",
        database="analytics",
        user="user",
        password=os.getenv('DB_PASSWORD')
    )
    
    query = """
        SELECT user_id, variant, date, conversion, revenue
        FROM experiment_data
        WHERE date >= CURRENT_DATE - INTERVAL '14 days'
        AND experiment_id = 'homepage_test'
    """
    
    df = pd.read_sql(query, conn)
    conn.close()
    
    csv_path = f'/tmp/experiment_{datetime.now().strftime("%Y%m%d")}.csv'
    df.to_csv(csv_path, index=False)
    logging.info(f"Exported {len(df)} rows to {csv_path}")
    
    return csv_path

def upload_to_ablabz(file_path, experiment_id):
    """Загрузка файла в AB-Labz Workbench"""
    logging.info(f"Uploading {file_path} for experiment {experiment_id}")
    
    # 1. Получить URL для загрузки
    response = requests.post(
        f'{BASE_URL}/datasets/initiate/',
        headers={'Authorization': f'Bearer {API_KEY}'},
        json={'experiment_id': experiment_id}
    )
    response.raise_for_status()
    upload_url = response.json()['upload_url']
    
    # 2. Загрузить файл
    with open(file_path, 'rb') as f:
        upload_response = requests.put(
            upload_url,
            data=f,
            headers={'Content-Type': 'text/csv'}
        )
        upload_response.raise_for_status()
    
    logging.info("Upload successful!")
    return True

if __name__ == '__main__':
    try:
        csv_path = export_data_from_db()
        upload_to_ablabz(csv_path, EXPERIMENT_ID)
        os.remove(csv_path)  # Очистка
    except Exception as e:
        logging.error(f"Error: {e}")
        raise