import json

from peopledatalabs import PDLPY

client = PDLPY(
    api_key="YOUR API KEY",
)

params = {"profile": ["http://linkedin.com/in/seanthorne"], "min_likelihood": 6}

json_response = client.person.enrichment(**params).json()

if json_response["status"] == 200:
    record = json_response["data"]

    # Print selected fields
    print(
        record["work_email"],
        record["full_name"],
        record["job_title"],
        record["job_company_name"],
    )

    print(f"successfully enriched profile with pdl data")

    with open("my_pdl_enrichment.jsonl", "w") as out:
        out.write(json.dumps(record) + "/n")

else:
    print("Enrichment unsuccessful. See error and try again.")
    print("error:", json_response)