retrieve_companies_hki.py

import requests
import datetime

# Function to query the PRH API for companies in a given city within the last month
def query_companies(city_name):
    print(f"Querying for companies in {city_name} within the last month")

    # Define the base URL for the PRH API
    base_url = "https://avoindata.prh.fi/bis/v1"
    
    # Calculate the date one month ago from today
    one_month_ago = datetime.datetime.now() - datetime.timedelta(days=30)
    formatted_date = one_month_ago.strftime('%Y-%m-%d')

    # Define your specific parameters
    params = {
        'totalResults': 'true',
        'maxResults': '1000',
        'resultsFrom': '0',
        'registeredOffice': city_name,
        'companyForm': 'OY',
        'companyRegistrationFrom': formatted_date
    }

    # Make the API request
    response = requests.get(base_url, params=params)

    # Check if the request was successful
    if response.status_code == 200:
        data = response.json()
        total_results = data['totalResults']
        print(f"Total companies in {city_name}: {total_results}")

        # Extract email addresses if available
        for company in data['results']:
            # Assuming email is a field in the company data
            email = company.get('email', 'No email available')
            print(f"Company: {company['name']}, Email: {email}")

    else:
        print(f"Failed to retrieve data. Status Code: {response.status_code}, Response: {response.text}")

# Example usage
query_companies('Helsinki')