from google.analytics.data_v1beta import BetaAnalyticsDataClient
from google.analytics.data_v1beta.types import (
    DateRange,
    Dimension,
    Metric,
    RunReportRequest,
    AudienceDimension,
    AudienceExport
)
import pandas as pd


def sample_run_report(property_id=GA4_PROPERTY_ID):
    property_id = GA4_PROPERTY_ID

    client = BetaAnalyticsDataClient()

    request = RunReportRequest(
        property=f"properties/{property_id}",
        dimensions=[Dimension(name='city'), Dimension(name='country'), AudienceDimension(name="userId")],
        metrics=[Metric(name="activeUsers"), Metric(name='userKeyEventRate'), Metric(name='screenPageViews')],
        date_ranges=[DateRange(start_date="2020-03-31", end_date="today")],
    )
    
    response = client.run_report(request)

    data = {'city': [], 'country': [], 'activeUsers': [], 'userKeyEventRate': [], 'screenPageViews':[]}

    for row in response.rows:
        data['city'].append(row.dimension_values[0].value)
        data['country'].append(row.dimension_values[1].value)
        data['activeUsers'].append(row.metric_values[0].value)
        data['userKeyEventRate'].append(row.metric_values[1].value)
        data['screenPageViews'].append(row.metric_values[2].value)


    df = pd.DataFrame(data=data)

    print(df.head(30))