import os
from segment_config_api.api import SegmentConfigApi

access_token = os.getenv('SEGMENT_CONFIG_API_ACCESS_TOKEN') # Or hard-coded or whichever way you want to access it
api = SegmentConfigApi(access_token)

# List all workspaces
api.workspaces.list()

# Get a workspace
api.workspace('myworkspace').get()

# List all sources
catalog = api.integrations_catalog

catalog.sources.list()

# Get a specific source
catalog.source('customerio').get()

# List all destinations
catalog = api.integrations_catalog

catalog.destinations.list()

# Get a specific destination
catalog.destination('customerio').get()

# Sources are scoped on a workspace level.
workspace = api.workspace('myworkspace')
sources = workspace.sources

# List all sources
sources.list()

# Create a new source
payload = {...}
sources.create(payload)

# Get a specific source by name
workspace.source('js').get()

# Get schema config
sc = workspace.source('js').get_schema_config

#Update schema config
payload = {...}
workspace.source('js').update_schema_config(payload)

# Delete a source
workspace.source('js').delete()

# Destinations are scoped on the source level
workspace = api.workspace('myworkspace')
destinations = api.source('js').destinations

# List all destinations at this source
destinations.list()

# Create a new destination for this source
payload = {}
destinations.create(payload)

# Get a specific destination by name
ga = workspace.source('js').destination('google-analytics')
ga.get()

# Update a destination
payload = {...}
ga.update(payload)

# Delete a destination
ga.delete()

# Tracking plans are scoped on the workspace level
workspace = api.workspace('myworkspace')
tracking_plans = workspace.tracking_plans

# List all the tracking plans
tracking_plans.list()

# Create a tracking plan
payload = {...}
tracking_plans.create(payload)

# Get a specific tracking plan by plan_id
plan = workspace.tracking_plan('rs_123')
plan.get()

# Update a tracking plan
payload = {...}
plan.update(payload)

# Create tracking plan source connection
plan.create_source_connection('workspaces/myworkspace/sources/js')

# Batch create source connections
source_names = [...]
plan.batch_create_source_connection(source_names)

# List source connections
plan.list_source_connections()

# Delete source connection
plan.delete_source_connection('workspaces/myworkspace/sources/js')

# Delete a tracking plan
plan.delete()

# Event delivery metrics are scoped on a destination level
ga = api.workspace('myworkspace').source('js').destination('google-analytics')

metrics = ga.event_delivery_metrics

# List all the timeseries metrics
payload = {...}
metrics.list_timeseries(payload=payload) #payload is optional


# Get a timeseries metric
metrics.get_timeseries('time_to_success_p95')

# Batch get a timeseries metric, no need to add the full path prefix to metrics names, we'll do it for you
metrics.batch_get_timeseries(['successes', 'retried_502'])

# Get summary metrics
metrics.get_summary()

# Batch get summary metrics (on a workspace level)
ws = api.workspace('myworkspace')
pairs = ['workspaces/myworkspace/sources/js/destinations/google-analytics']
# Or let the api wrapper build your path
pairs = [ws.source('js').destination('google-analytics').model_path]

ws.batch_get_summary_metrics(pairs)

# Destination Filters are scoped on a destination level
ga = api.workspace('myworkspace').source('js').destination('google-analytics')
destination_filters = ga.destination_filters

# List all filters for a destination
destination_filters.list()

# Create a destination filter
payload = {...}
destination_filters.create(payload)

# Get a specific filter
filter = ga.destination_filter('df_123')
filter.get()

# Update a filter
payload = {...}
filter.update(payload)

# Delete a filter
filter.delete()

# Preview a filter - This is scoped on the api level
payload = {
    "filter": {
        "if": "all",
            "actions": [
                {
                    "type": "blacklist_fields",
                    "fields": {
                        "properties": {
                            "fields": ["name", "age"]
                        }
                    }
                }
            ],
            "enabled": True
        },
        "input": "{ \"userId\": \"6592\", \"properties\": { \"name\": \"Bob Smith\", \"age\": \"40\", \"order\": \"129\" } }"
}
api.destination_filters.preview(payload)

# Regulations are scoped on a workspace level
ws = api.workspace('myworkspace')
regulations = ws.regulations

# List all regulations for a workspace
regulations.list()

#List regulations based on status
regulations.list(regulation_types='Suppress_With_Delete')

# Create a regulation
payload = {...}
regulations.create(payload)

# Get a specific regulation on it's regulation_id
regulation = ws.regulation('1KQVncbOPeRGjRcpuOHdnhDwPn7')
regulation.get()

# Delete a regulation (It has to be in an `initialized` state)
regulation.delete()

# Create a regulation on a source
payload = {...}
ws.source('js').regulations.create(payload)

# Delete an Object from a cloud source
payload = {...}
ws.source('js').regulations.delete(payload)

# List suppressed users on a workspace level
ws.suppressed_users.list()

# Roles are scoped on a workspace level
ws = api.workspace('myworkspace')
roles = ws.roles

# List all roles for a workspace
roles.list()

# List role policies
roles.policies.list()

# Get a specific regulation on it's regulation_id
regulation = ws.regulation('1KQVncbOPeRGjRcpuOHdnhDwPn7')
regulation.get()

# Delete a regulation (It has to be in an `initialized` state)
regulation.delete()

# Create a role policy (scoped on the api level)
payload = {...}
api.role_policies(role_name).create(payload)

# Delete role policy
api.role_policy(policy_name).delete()

# Invites are scoped on the workspace level
invites = ws.invites

# List all invites
invites.list()

# Create an invite
payload = {...}
invites.create(payload)

# Delete an invite
ws.invite('123').delete()