Skip to content

prefect-census

PyPI

Welcome!

This collection contains Prefect integrations for working with Census.

Census is an Operational Analytics platform that enables you to sync your trusted analytics data from your hub into operational tools that your business teams use on a daily basis.

Some things you can do with this collection out-of-the-box include: - Trigger a Census sync run and wait for it to finish (see how) - Call a custom endpoint with a CensusClient using your CensusCredentials (see how)

For information on how to get started with Census, refer to the Census docs.

Getting Started

Python setup

Requires an installation of Python 3.7+.

We recommend using a Python virtual environment manager such as pipenv, conda or virtualenv.

These tasks are designed to work with Prefect 2.0. For more information about how to use Prefect, please refer to the Prefect documentation.

Installation

Install prefect-census with pip:

pip install prefect-census

A list of available blocks in prefect-census and their setup instructions can be found here.

Get a Census API Key

You will need a Census API key to be able to use the integrations in this collection.

For directions for how to generate a Census API key, refer to the Getting API Access section of the Census docs.

Once you have a Census API key, you can configure a CensusCredentials block in the Prefect UI for use with the integrations in this collection. For information about how to configure a block in the Prefect UI, refer to the Prefect docs.

💡 NOTE

All Census Syncs are incremental, which means they only capture records that are new or changed data since the previous sync.

Write and run a flow

Use a CensusSync job block to run a sync and wait for completion

from prefect import flow
from prefect_census import (
    CensusCredentials, CensusSync, run_census_sync
)

census_sync = CensusSync(
    credentials=CensusCredentials(api_key="my_api_key"),
    sync_id=42
)

@flow
def my_census_flow():
    # do some setup

    run_census_sync(census_sync)

    # do some cleanup

Get Census sync run info:

from prefect import flow

from prefect_census import CensusCredentials
from prefect_census.runs import get_census_sync_run_info

@flow
def get_sync_run_info_flow():
    credentials = CensusCredentials(api_key="my_api_key")

    return get_census_sync_run_info(
        credentials=credentials,
        run_id=42
    )

get_sync_run_info_flow()

Call a custom endpoint:

from prefect import flow
from prefect_census import CensusCredentials
from prefect_census.client import CensusClient

@flow
def my_flow(sync_id):
    creds_block = CensusCredentials(api_key="my_api_key")

    client = CensusClient(
        api_key=creds_block.api_key.get_secret_value()    
    )
    response = client.call_endpoint(
        http_method="GET", path=f"/syncs/{sync_id}"
    )
    return response

my_flow(42)

For more tips on how to use tasks and flows in a Collection, check out Using Collections!

Resources

If you encounter any bugs while using prefect-census, feel free to open an issue in the prefect-census repository.

If you have any questions or issues while using prefect-census, you can find help in either the Prefect Discourse forum or the Prefect Slack community.

Feel free to star or watch prefect-census for updates too!

Contributing

If you'd like to help contribute to fix an issue or add a feature to prefect-census, please propose changes through a pull request from a fork of the repository.

Here are the steps: 1. Fork the repository 2. Clone the forked repository 3. Install the repository and its dependencies:

pip install -e ".[dev]"
4. Make desired changes 5. Add tests 6. Insert an entry to CHANGELOG.md 7. Install pre-commit to perform quality checks prior to commit:
pre-commit install
8. git commit, git push, and create a pull request