Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Source Facebook Marketing (no-singer) #1443

Closed
Closed
Show file tree
Hide file tree
Changes from 18 commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
dc5f72e
source Facebook Marketing (no-singer)
eugene-kulak Dec 24, 2020
9cef6e3
update sample config
eugene-kulak Dec 24, 2020
bd81d25
fix source import
eugene-kulak Dec 24, 2020
17884b2
add fixme
eugene-kulak Dec 24, 2020
0f7e9c9
fix tests, account fetch and stream reading
eugene-kulak Dec 28, 2020
6343bfc
implement schema resolver
eugene-kulak Dec 29, 2020
b59d50d
add CI creds
eugene-kulak Dec 29, 2020
a422faf
register as standard source
eugene-kulak Dec 29, 2020
7a21b16
fix tests
eugene-kulak Dec 29, 2020
0b5d914
format
eugene-kulak Dec 29, 2020
1ff5878
fix Dockerfile.test
eugene-kulak Dec 29, 2020
c909a4b
fix python3.7 compatibility
eugene-kulak Dec 29, 2020
555038e
clean up
eugene-kulak Dec 29, 2020
5530fa7
address comments from @arhip11
eugene-kulak Dec 29, 2020
a842673
address comments from @arhip11
eugene-kulak Dec 29, 2020
ced47e8
Merge remote-tracking branch 'origin/master' into source-facebook-mar…
eugene-kulak Dec 29, 2020
e285d71
clean up
eugene-kulak Dec 29, 2020
c99e121
"fix" logger usage, better fix possible in #1279
eugene-kulak Dec 29, 2020
3731637
fix incorrect logger usage
eugene-kulak Dec 29, 2020
14c9086
fix incorrect logger usage (2)
eugene-kulak Dec 29, 2020
19b2de3
Update airbyte-integrations/connectors/source-facebook-marketing/buil…
eugene-kulak Dec 30, 2020
929d2ac
Update docs/integrations/sources/facebook-marketing.md
eugene-kulak Dec 30, 2020
79a33f3
Update airbyte-integrations/connectors/source-facebook-marketing/unit…
eugene-kulak Dec 30, 2020
0d1017b
address CR changes
eugene-kulak Dec 30, 2020
54b1953
drop accidentally added .dockerinore file
eugene-kulak Dec 30, 2020
dee3d88
fix logger usage (3)
eugene-kulak Dec 31, 2020
13c2eee
format
eugene-kulak Dec 31, 2020
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .github/workflows/test-command.yml
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ jobs:
BIGQUERY_INTEGRATION_TEST_CREDS: ${{ secrets.BIGQUERY_INTEGRATION_TEST_CREDS }}
BRAINTREE_TEST_CREDS: ${{ secrets.BRAINTREE_TEST_CREDS }}
DRIFT_INTEGRATION_TEST_CREDS: ${{ secrets.DRIFT_INTEGRATION_TEST_CREDS }}
FACEBOOK_MARKETING_TEST_INTEGRATION_CREDS: ${{ secrets.FACEBOOK_MARKETING_TEST_INTEGRATION_CREDS }}
FACEBOOK_MARKETING_API_TEST_INTEGRATION_CREDS: ${{ secrets.FACEBOOK_MARKETING_API_TEST_INTEGRATION_CREDS }}
FRESHDESK_TEST_CREDS: ${{ secrets.FRESHDESK_TEST_CREDS }}
GH_INTEGRATION_TEST_CREDS: ${{ secrets.GH_INTEGRATION_TEST_CREDS }}
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
{
"sourceDefinitionId": "e7778cfc-e97c-4458-9ecb-b4f2bba8946c",
"name": "Facebook Marketing",
"dockerRepository": "airbyte/source-facebook-marketing",
"dockerImageTag": "0.1.0",
"documentationUrl": "https://hub.docker.com/r/airbyte/source-facebook-marketing"
}
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,11 @@
dockerRepository: airbyte/source-googleanalytics-singer
dockerImageTag: 0.1.5
documentationUrl: https://hub.docker.com/r/airbyte/source-googleanalytics-singer
- sourceDefinitionId: e7778cfc-e97c-4458-9ecb-b4f2bba8946c
name: Facebook Marketing
dockerRepository: airbyte/source-facebook-marketing
dockerImageTag: 0.1.0
documentationUrl: https://hub.docker.com/r/airbyte/source-facebook-marketing
- sourceDefinitionId: 74d47f79-8d01-44ac-9755-f5eb0d7caacb
name: Facebook Marketing APIs
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

In a separate PR we should delete the Singer one.

dockerRepository: airbyte/source-facebook-marketing-api-singer
Expand Down
75 changes: 73 additions & 2 deletions airbyte-integrations/bases/base-python/base_python/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,15 @@

import inspect
import json
import os
import pkgutil
from abc import ABC, abstractmethod
from datetime import datetime
from typing import Dict, Generator, Tuple
from typing import Dict, Generator, Tuple, List

import pkg_resources
from airbyte_protocol import AirbyteRecordMessage, AirbyteStream
from jsonschema import RefResolver


def package_name_from_class(cls: object) -> str:
Expand All @@ -38,6 +41,65 @@ def package_name_from_class(cls: object) -> str:
return module.__name__.split(".")[0]


class JsonSchemaResolver:
"""Helper class to expand $ref items in json schema"""

def __init__(self, shared_schemas_path: str):
self._shared_refs = self._load_shared_schema_refs(shared_schemas_path)

@staticmethod
def _load_shared_schema_refs(path: str):
shared_file_names = [f for f in os.listdir(path) if os.path.isfile(os.path.join(path, f))]

shared_schema_refs = {}
for shared_file in shared_file_names:
with open(os.path.join(path, shared_file)) as data_file:
shared_schema_refs[shared_file] = json.load(data_file)

return shared_schema_refs

def _resolve_schema_references(self, schema: dict, resolver: RefResolver) -> dict:
if "$ref" in schema:
reference_path = schema.pop("$ref", None)
resolved = resolver.resolve(reference_path)[1]
schema.update(resolved)
return self._resolve_schema_references(schema, resolver)

if "properties" in schema:
for k, val in schema["properties"].items():
schema["properties"][k] = self._resolve_schema_references(val, resolver)

if "patternProperties" in schema:
for k, val in schema["patternProperties"].items():
schema["patternProperties"][k] = self._resolve_schema_references(val, resolver)

if "items" in schema:
schema["items"] = self._resolve_schema_references(schema["items"], resolver)

if "anyOf" in schema:
for i, element in enumerate(schema["anyOf"]):
schema["anyOf"][i] = self._resolve_schema_references(element, resolver)

return schema

def resolve(self, schema: dict, refs: Dict[str, dict] = None) -> dict:
"""Resolves and replaces json-schema $refs with the appropriate dict.
Recursively walks the given schema dict, converting every instance
of $ref in a 'properties' structure with a resolved dict.
This modifies the input schema and also returns it.
Arguments:
schema:
the schema dict
refs:
a dict of <string, dict> which forms a store of referenced schemata
Returns:
schema
"""
refs = refs or {}
refs = {**self._shared_refs, **refs}
return self._resolve_schema_references(schema, RefResolver("", schema, store=refs))


class ResourceSchemaLoader:
"""JSONSchema loader from package resources"""

Expand All @@ -46,6 +108,9 @@ def __init__(self, package_name: str):

def get_schema(self, name: str) -> dict:
raw_schema = json.loads(pkgutil.get_data(self.package_name, f"schemas/{name}.json"))
shared_schemas_folder = pkg_resources.resource_filename(self.package_name, "schemas/shared/")
if os.path.exists(shared_schemas_folder):
return JsonSchemaResolver(shared_schemas_folder).resolve(raw_schema)
return raw_schema


Expand All @@ -70,13 +135,19 @@ def _enumerate_methods(self) -> Dict[str, callable]:

return mapping

@staticmethod
def _get_fields_from_stream(stream: AirbyteStream) -> List[str]:
return list(stream.json_schema.get("properties", {}).keys())

def read_stream(self, stream: AirbyteStream) -> Generator[AirbyteRecordMessage, None, None]:
"""Yield records from stream"""
method = self._stream_methods.get(stream.name)
if not method:
raise ValueError(f"Client does not know how to read stream `{stream.name}`")

for message in method():
fields = self._get_fields_from_stream(stream)

for message in method(fields=fields):
now = int(datetime.now().timestamp()) * 1000
yield AirbyteRecordMessage(stream=stream.name, data=message, emitted_at=now)

Expand Down
2 changes: 1 addition & 1 deletion airbyte-integrations/bases/base-python/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
url="https://github.com/airbytehq/airbyte",
packages=setuptools.find_packages(),
package_data={"": ["models/yaml/*.yaml"]},
install_requires=["PyYAML==5.3.1", "pydantic==1.6.1", "airbyte-protocol"],
install_requires=["PyYAML==5.3.1", "pydantic==1.6.1", "airbyte-protocol", "jsonschema"],
entry_points={
"console_scripts": ["base-python=base_python.entrypoint:main"],
},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,13 +35,13 @@ def __init__(self, access_token: str):
super().__init__()
self._client = APIClient(access_token)

def stream__accounts(self) -> Iterator[dict]:
def stream__accounts(self, **kwargs) -> Iterator[dict]:
yield from self._client.accounts.list()

def stream__users(self) -> Iterator[dict]:
def stream__users(self, **kwargs) -> Iterator[dict]:
yield from self._client.users.list()

def stream__conversations(self) -> Iterator[dict]:
def stream__conversations(self, **kwargs) -> Iterator[dict]:
yield from self._client.conversations.list()

def health_check(self) -> Tuple[bool, str]:
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
FROM airbyte/integration-base-python:dev

# Bash is installed for more convenient debugging.
RUN apt-get update && apt-get install -y bash && rm -rf /var/lib/apt/lists/*

ENV CODE_PATH="source_facebook_marketing"
ENV AIRBYTE_IMPL_MODULE="source_facebook_marketing"
ENV AIRBYTE_IMPL_PATH="SourceFacebookMarketing"

WORKDIR /airbyte/integration_code
COPY $CODE_PATH ./$CODE_PATH
COPY setup.py ./
RUN pip install .

LABEL io.airbyte.version=0.1.0
LABEL io.airbyte.name=airbyte/source-facebook-marketing
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
FROM airbyte/base-python-test:dev

RUN apt-get update && apt-get install -y bash && rm -rf /var/lib/apt/lists/*

ENV CODE_PATH="integration_tests"
ENV AIRBYTE_TEST_MODULE="integration_tests"
ENV AIRBYTE_TEST_PATH="SourceFacebookMarketingStandardTest"

LABEL io.airbyte.version=0.1.0
LABEL io.airbyte.name=airbyte/source-facebook-marketing-standard-test

WORKDIR /airbyte/integration_code
COPY source_facebook_marketing source_facebook_marketing
COPY $CODE_PATH $CODE_PATH
COPY sample_files/*.json $CODE_PATH/
COPY secrets/* $CODE_PATH
COPY source_facebook_marketing/*.json $CODE_PATH
COPY setup.py ./

RUN pip install ".[tests]"

WORKDIR /airbyte
Original file line number Diff line number Diff line change
@@ -0,0 +1,73 @@
# Facebook Marketing Source

This is the repository for the Facebook Marketing source connector, written in Python.
For information about how to use this connector within Airbyte, see [the documentation](https://docs.airbyte.io/integrations/sources/facebook-marketing).

## Local development

### Prerequisites
**To iterate on this connector, make sure to complete this prerequisites section.**

#### Build & Activate Virtual Environment
First, build the module by running the following from the `airbyte` project root directory:
```
./gradlew :airbyte-integrations:connectors:source-facebook-marketing:build
```

This will generate a virtualenv for this module in `source-facebook-marketing/.venv`. Make sure this venv is active in your
development environment of choice. To activate the venv from the terminal, run:
```
cd airbyte-integrations/connectors/source-facebook-marketing # cd into the connector directory
source .venv/bin/activate
```
If you are in an IDE, follow your IDE's instructions to activate the virtualenv.

#### Create credentials
**If you are a community contributor**, follow the instructions in the [documentation](https://docs.airbyte.io/integrations/sources/facebook-marketing)
to generate the necessary credentials. Then create a file `secrets/config.json` conforming to the `source_facebook_marketing/spec.json` file.
See `sample_files/sample_config.json` for a sample config file.

**If you are an Airbyte core member**, copy the credentials in RPass under the secret name `source-facebook-marketing-integration-test-config`
and place them into `secrets/config.json`.


### Locally running the connector
```
python main_dev.py spec
python main_dev.py check --config secrets/config.json
python main_dev.py discover --config secrets/config.json
python main_dev.py read --config secrets/config.json --catalog sample_files/configured_catalog.json
```

### Unit Tests
To run unit tests locally, from the connector directory run:
```
pytest unit_tests
```

### Locally running the connector docker image
```
# in airbyte root directory
./gradlew :airbyte-integrations:connectors:source-facebook-marketing:airbyteDocker
docker run --rm airbyte/source-facebook-marketing:dev spec
docker run --rm -v $(pwd)/airbyte-integrations/connectors/source-facebook-marketing/secrets:/secrets airbyte/source-facebook-marketing:dev check --config /secrets/config.json
docker run --rm -v $(pwd)/airbyte-integrations/connectors/source-facebook-marketing/secrets:/secrets airbyte/source-facebook-marketing:dev discover --config /secrets/config.json
docker run --rm -v $(pwd)/airbyte-integrations/connectors/source-facebook-marketing/secrets:/secrets -v $(pwd)/airbyte-integrations/connectors/source-facebook-marketing/sample_files:/sample_files airbyte/source-facebook-marketing:dev read --config /secrets/config.json --catalog /sample_files/configured_catalog.json
```

### Integration Tests
1. From the airbyte project root, run `./gradlew :airbyte-integrations:connectors:source-facebook-marketing:standardSourceTestPython` to run the standard integration test suite.
1. To run additional integration tests, place your integration tests in the `integration_tests` directory and run them with `pytest integration_tests`.
Make sure to familiarize yourself with [pytest test discovery](https://docs.pytest.org/en/latest/goodpractices.html#test-discovery) to know how your test files and methods should be named.

## Dependency Management
All of your dependencies should go in `setup.py`, NOT `requirements.txt`. The requirements file is only used to connect internal Airbyte dependencies in the monorepo for local development.

## Populating account with the data

The following will create 120 accounts and conversations
```bash
export DRIFT_TOKEN=<PUT_YOUR_TOKEN_HERE>
cd airbyte-integrations/connectors/source-facebook-marketing/source_facebook_marketing/
python -m client.fixture
```
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
plugins {
id 'airbyte-python'
id 'airbyte-docker'
id 'airbyte-source-test'
id 'airbyte-integration-test-java'
eugene-kulak marked this conversation as resolved.
Show resolved Hide resolved
}

airbytePython {
moduleDirectory 'source_facebook_marketing'
}

dependencies {
implementation files(project(':airbyte-integrations:bases:base-python').airbyteDocker.outputs)
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
"""
MIT License

Copyright (c) 2020 Airbyte

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""

from .standard_source_test import SourceFacebookMarketingStandardTest

__all__ = ["SourceFacebookMarketingStandardTest"]
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
"""
MIT License

Copyright (c) 2020 Airbyte

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""

from base_python_test import DefaultStandardSourceTest


class SourceFacebookMarketingStandardTest(DefaultStandardSourceTest):
pass
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
"""
MIT License

Copyright (c) 2020 Airbyte

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
"""

import sys

from base_python.entrypoint import launch
from source_facebook_marketing import SourceFacebookMarketing

if __name__ == "__main__":
source = SourceFacebookMarketing()
launch(source, sys.argv[1:])
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
-e ../../bases/airbyte-protocol
-e ../../bases/base-python
-e ../../bases/base-python-test
-e .
Loading