DEV Community

Cover image for Pydantic Settings + AWS the easy way
Rafael de Oliveira Marques
Rafael de Oliveira Marques

Posted on

Pydantic Settings + AWS the easy way

Pydantic Settings is a python library that extends ๐Ÿš€ Pydantic for dealing with settings management.

If you've never heard of pydantic, please, check it's docs page to see how easy and powerfull it is. In my opinion, it is one of the best python libraries I have come across in the last few months/years. Congrats Samuel Colvin!

โš™๏ธ Using Pydantic Settings

To use pydantic-settings, we first need to install it:



pip install pydantic
pip install pydantic-settings


Enter fullscreen mode Exit fullscreen mode

Now let's imagine that we have a FastAPI microservice that needs to load some configuration.

Let's say we depend on some information to work properly:

  • Environment that is available in the environment variables
  • Database host that is available in an AWS Parameter Store
  • Database authentication info that is available in an AWS Secrets Manager

What we would need to do is something like:



import json

import boto3

from pydantic_settings import BaseSettings, SettingsConfigDict


class AppSettings(BaseSettings):
    model_config = SettingsConfigDict(env_prefix="APP_")

    environment: str  # will be loaded by environment variable APP_ENVIRONMENT
    host: str  # need to pass it when creating AppSettings :(
    username: str  # need to pass it when creating AppSettings :(
    password: str  # need to pass it when creating AppSettings :(


ssm_client = boto3.client("boto3")
secrets_client = boto3.client("secretsmanager")

# need to get the secret before creating our settings object
secret_response = secrets_client.get_secret_value(SecretId="my/secrets")
secret_string = secret_response.get("SecretString")
secret = json.loads(secret_string)

# need to get the parameter store before creating our settings object
parameter = ssm_client.get_parameter(Name="/my/ssm/parameter")
host = parameter.get("Parameter").get("Value")


settings = AppSettings(host=host, **secret)


Enter fullscreen mode Exit fullscreen mode

Don't get me wrong, it's still great. But it would be better if we had a pydantic settings source to do that work for us!

Obs: You can check pydantic-settings ๐Ÿ“– docs to see all settings sources that it offers.

โ˜๏ธ Pydantic Settings AWS

Pydantic Settings + AWS

I recently started creating a pydantic-settings extension to deal with settings that lives at AWS. It called Pydantic Settings AWS and you can see the source code at github.com/ceb10n/pydantic-settings-aws

๐Ÿƒ Gettings started

If you still don't have boto3 or pydantic-settings installed:



pip install boto3
pip install pydantic-settings


Enter fullscreen mode Exit fullscreen mode

And then you can install pydantic-settings-aws



pip install pydantic-settings-aws


Enter fullscreen mode Exit fullscreen mode

Now let's get back to our example.

We have a basic settings class that needs to load data from Environmet variables, parameter store and secrets manager.

pydantic-settings-aws offers a class called AWSBaseSettings, that gives you the ability to deal with all that data sources.



from typing import Annotated

from pydantic_settings import SettingsConfigDict
from pydantic_settings_aws import AWSBaseSettings


class AppSettings(AWSBaseSettings):
    model_config = SettingsConfigDict(
        secrets_name="my/secrets",
        env_prefix="APP_"
    )

    environment: str
    host: Annotated[str, {"service": "ssm", "ssm": "/my/ssm/parameter"}]
    username: Annotated[str, {"service": "secrets"}]
    password: Annotated[str, {"service": "secrets"}]


settings = AppSettings()


Enter fullscreen mode Exit fullscreen mode

With AWSBaseSettings, all you need to do is using typing.Annotated to inform what service you want to use, and other things, like the secret and parameter store names.

๐Ÿ™Š Settings with Secrets Manager only

If you simply want to use Secrets Manager, you can use the SecretsManagerBaseSettings.

Since we are using only Secrets Manager as a data source, we don't need to specify anything, except the secrets name:



from pydantic_settings import SettingsConfigDict
from pydantic_settings_aws import SecretsManagerBaseSettings


class AppSettings(SecretsManagerBaseSettings):
    model_config = SettingsConfigDict(
        secrets_name="my/secrets"
    )

    username: str
    password: str

settings = AppSettings()
# username='my-username' password='my-super-secret-password'


Enter fullscreen mode Exit fullscreen mode

๐Ÿ”’ Settings with Parameter Store only

Let's say you use Parameter Store to save a lot of you apps configuration, like queue names, urls, etc.

You can leverage the ParameterStoreBaseSettings class to load all data from different parameter stores.




from pydantic_settings_aws import ParameterStoreBaseSettings


class AppSettings(ParameterStoreBaseSettings):

    dev_base_endpoint: str
    database_host: str
    database_port: str


Enter fullscreen mode Exit fullscreen mode

And thats it. ParameterStoreBaseSettings will try to create a boto3 client with your local configuration, and then will try to get the value from a parameter store with the same name as your field.

But it's not uncommon for the names of our parameter stores to not match the names of our variables. In theses cases, you can use typing.Annotated:



from typing import Annotated

from pydantic_settings_aws import ParameterStoreBaseSettings

class AppSettings(ParameterStoreBaseSettings):

<span class="n">dev_base_endpoint</span><span class="p">:</span> <span class="n">Annotated</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="sh">"</span><span class="s">/payments/endpoint</span><span class="sh">"</span><span class="p">]</span>
<span class="n">database_host</span><span class="p">:</span> <span class="n">Annotated</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="sh">"</span><span class="s">/databases/mongodb/payments/dbhost</span><span class="sh">"</span><span class="p">]</span>
<span class="n">database_port</span><span class="p">:</span> <span class="n">Annotated</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="sh">"</span><span class="s">/databases/mongodb/payments/dbport</span><span class="sh">"</span><span class="p">]</span>
Enter fullscreen mode Exit fullscreen mode
Enter fullscreen mode Exit fullscreen mode




๐Ÿ“œ Docs and source code

You can take a look at pydantic-settings-aws documentations at [ceb10n.github.io/pydantic-settings-aws](https://ceb10n.github.io/pydantic-settings-aws/].

The project is hosted at github: github.com/ceb10n/pydantic-settings-aws. It's still an ongoing project. Feel free to open an issue, make a PR, etc. ๐Ÿค—

Top comments (1)

Collapse
 
gersom_ramos_756348c1f752 profile image
Gersom Ramos

What a brilliant read! I thoroughly enjoyed the insights shared in this piece. Itโ€™s refreshing to see such thoughtful content, and I couldn't help but feel a sense of joy as I read through it. I've also noticed others sharing their experiences, and itโ€™s clear this community is thriving! I wanted to chime in about how AI Remote Hire has been a game-changer for smarter remote staffing. Their services truly make a difference. If you're considering enhancing your remote work strategy, I highly recommend checking out their

Remote Staffing

solutions!

Some comments may only be visible to logged-in visitors. Sign in to view all comments.