Awesome
IMDB Dataform Project
About
An example Dataform project to load and transform the publicly available dataset from IMDB.
Prerequisites
Google Cloud Project
Google Cloud projects form the basis for creating, enabling, and using all Google Cloud services, such as Dataform and BigQuery.
If you do not already have a Google Cloud project for which you want to load the IMDB dataset into, then you will need to create a new Google Cloud project. The documentation on how to do this can be found here.
Once you have a Google Cloud project, remember to take note of the Project Number and Project ID. These can be found on the Google Cloud project console welcome page, which you can find here.
Google Cloud Storage Bucket
Now you have a Google Cloud project, you need to create a Google Cloud Storage Bucket for which the IMDB dataset will be uploaded into and Dataform will use to source the data in which to load data into BigQuery. The documentation on how to create a new storage bucket can be found here.
Remeber to take note of the bucket name as this will be required for one of the Dataform config variables.
Enable Dataform Service
Next, you will need to enable the Dataform service within the Google Cloud project just created. This can be achieved by clicking the "Enable" button here.
Create a Dataform Repository
After the Dataform Service has been enabled, you will be redirected to the BigQuery Dataform page within the Google Cloud console. For reference, this can be found here.
Go ahead and create a repository. For more information on how to do this, go to the documentation page found here.
Grant Permissions to Dataform Service Account
When you create your first Dataform repository, Dataform automatically generates a service account. Dataform uses the service account to interact with BigQuery on your behalf.
Your Dataform service account ID is in the following format:
service-YOUR_PROJECT_NUMBER@gcp-sa-dataform.iam.gserviceaccount.com
Replace YOUR_PROJECT_NUMBER with the Project Number of your Google Cloud project, which you previously took note of.
The Dataform service account requires a number of IAM roles with which to be able to execute the workflows in BigQuery and load data from the Google Cloud Storage Bucket. This can be achieved by following these steps:
- In the Google Cloud console, go to the IAM page.
- Click Add.
- In the New principals field, enter your Dataform service account ID.
- In the Select a role drop-down list, select the BigQuery Job User role.
- Click Add another role, and then in the Select a role drop-down list, select the BigQuery Data Editor role.
- Click Add another role, and then in the Select a role drop-down list, select the BigQuery Data Viewer role.
- Click Add another role, and then in the Select a role drop-down list, select the Storage Object Viewer role.
- Click Save.
IMDB Public Dataset
Even though the public datasets can be accessed without any user credentials, it is recommended that you create an IMDB account, which you can do here. Once signed in, you can go to the Alternate Interfaces page for general data access, which will provide you details on licensing and metadata for each of the data files.
The data files can be downloaded directly from here, or individually listed below:
- name.basics.tsv.gz
- title.akas.tsv.gz
- title.basics.tsv.gz
- title.crew.tsv.gz
- title.episode.tsv.gz
- title.principals.tsv.gz
- title.ratings.tsv.gz
Download each of these files, before uploading them to the Google Cloud Storage Bucket you created
DataForm Workflow Settings
The workflow_settings.yaml
contains the following parameters
defaultProject
: The Project ID of your Google Cloud project, which you previously took note ofdefaultLocation
: Target BigQuery LocationdefaultDataset
: Name of the BigQuery Dataset for which the IMDB tables are to be createddefaultAssertionDataset
: Name of the BigQuery Dataset for which any Dataform Assertions are to be created and executed againstLOAD_GCS_BUCKET
: Name of the Google Cloud Storage Bucket, which you previously took note ofRAW_DATA
: Name of the BigQuery Dataset for which the IMDB data files are to be loaded intoTARGET_DATA
: Name of the BigQuery Dataset for which the final transformed IMDB tables are to be located
Here is what an example configuration looks like
dataformCoreVersion: 3.0.0-beta.4
defaultProject: winter-dataform
defaultLocation: australia-southeast1
defaultDataset: imdb
defaultAssertionDataset: imdb_assertions
vars:
LOAD_GCS_BUCKET: winter-data/imdb
RAW_DATA: imdb_staging
TARGET_DATA: imdb