Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/mtnbarreto/flask-base-api

A complete boilerplate Flask RESTful API code.
https://github.com/mtnbarreto/flask-base-api

flask flask-backend rest-api restful-api

Last synced: 3 months ago
JSON representation

A complete boilerplate Flask RESTful API code.

Awesome Lists containing this project

README

        

[![Docker Image CI](https://github.com/mtnbarreto/flask-base-api/actions/workflows/docker-image.yml/badge.svg)](https://github.com/mtnbarreto/flask-base-api/actions/workflows/docker-image.yml)

# Flask Base API

This repository aims to create a starting point to develop a REST API using Python and Flask framework as main technologies.

**Features:**
* Development environment with Docker that supports Test-Driven Development (TDD).
* Staging, Testing, and Production environments.
* RESTful API powered by Python, Flask web framework, Postgres DB, rabbitmq, and other technologies.
* Unit tests covering the REST API services.
* Code coverage.
* RESTful API documentation via Swagger.
* Easily visualize and consume RESTful API via Swagger UI.
* RabbitMQ message broker and RabbitMQ management plugin integration.
* Supports RESTful API versioning.
* JWT authentication.
* Google authentication.
* Facebook login.
* Firebase Cloud Messaging integration to send push notifications.
* SQLAlchemy ORM integration and modeling of base db entities.
* pgAdmin db administration and development platform for PostgreSQL.
* Nginx reverse proxy and load balancer.
## Contents

* [Quick start guide](#quick-start-guide)
* [Commands](#commands)
* [Dependencies](#dependencies)
* [RESTful endpoints](#restful-endpoints)
* [FAQ](#faq)

## Quick start guide


#### 1 - Create a folder to clone all projects.

```bash
mkdir && cd
```

#### 2 - Clone the project from folder.

```bash
git clone [email protected]:mtnbarreto/flask-base-api.git
```

#### 3 - Set up environment variables

Before putting up and running the app containers, we need to set up some environment variables, mainly user services accounts like Firebase and Twilio. To do so, create a file named `set_local_env_vars.sh` and add the following content replacing the values.

```bash
#!/usr/bin/env bash

export APP_SETTINGS="project.config.DevelopmentConfig"
export FLASK_APP=project/__init__.py
export SECRET_KEY="mysecret"
export MAIL_SERVER="smtp.googlemail.com"
export MAIL_PORT="465"
export MAIL_USERNAME="my_email@my_email_domain.com"
export MAIL_PASSWORD="my_email_password"
export MAIL_DEFAULT_SENDER="my_email@my_email_domain.com"
export TWILIO_ACCOUNT_SID="1234qwer"
export TWILIO_AUTH_TOKEN="qwer1234"
export TWILIO_FROM_NUMBER="+123456789"
export CELLPHONE_VALIDATION_CODE_EXP_SECS="600"
export MAIL_USE_TLS="False"
export MAIL_USE_SSL="True"
export FCM_SERVER_KEY="9876oiuy"
export GOOGLE_CLIENT_ID="my-google-client-id.apps.googleusercontent.com"
```

To set up the env variables, execute:

```bash
source ../set_local_env_vars.sh
```
#### 4 - Move to `flask-main`.

```bash
cd flask-main
```

#### 5 - Now let's build the images and run the containers. `flask-base-api` requires docker 20.10.1 or newer.

```bash
docker-compose build --no-cache
docker-compose up -d
```

> `docker-compose build` build the images. `--no-cache` arg indicates the cache should not be used. Docker caches the result of the build and use it in the subsequent builds. Remove this arg to build the images faster.
> `docker-compose up` fires up the containers. The `-d` flag is used to run the containers in the background.

`docker-compose up -d` should output something like ...

```bash
Creating network "flask-main_default" with the default driver
Creating pgadmin ... done
Creating redis ... done
Creating postgres-db ... done
Creating rabbitmq ... done
Creating celery-worker ... done
Creating flask-api ... done
Creating swagger ... done
Creating nginx ... done
```

After running the previous commands, let see the list of container by running:

```bash
docker-compose ps
```

```bash
Name Command State Ports
-------------------------------------------------------------------------------------------------------------------------------------------------------------------
flask-api /bin/sh -c bash -c 'while ... Up 0.0.0.0:5001->5000/tcp, 0.0.0.0:5678->5678/tcp
celery-worker /bin/sh -c celery -A proje ... Up
postgres-db docker-entrypoint.sh postgres Up (healthy) 0.0.0.0:5435->5432/tcp
nginx /docker-entrypoint.sh ngin ... Up 0.0.0.0:80->80/tcp
pgadmin /entrypoint.sh Up 443/tcp, 0.0.0.0:5050->80/tcp
rabbitmq docker-entrypoint.sh rabbi ... Up 15671/tcp, 0.0.0.0:15675->15672/tcp, 15691/tcp, 15692/tcp, 25672/tcp, 4369/tcp, 5671/tcp,
0.0.0.0:5675->5672/tcp
redis docker-entrypoint.sh redis ... Up 0.0.0.0:6375->6379/tcp
swagger sh /usr/share/nginx/docker ... Up 0.0.0.0:8080->8080/tcp
```

Make sure all services started properly. `State` should be `Up`.

#### 6 - Set up database

By running the following command, we recreate all development db tables:

```bash
docker-compose exec flask-api python manage.py recreate_db
```

then you can populate the db by executing the following command:

```bash
docker-compose exec flask-api python manage.py seed_db
```

Finally, test that everything works by executing the following curl command that tries to logged in using a user created by the seed_db command:

```bash
curl -X POST "http://0.0.0.0/v1/auth/login" -H "accept: application/json" -H "Content-Type: application/json" -d "{\"email\":\"[email protected]\",\"password\":\"password\"}"
```

it should output something like this:

```json
{
"status":"success",
"message":"Successfully logged in.",
"auth_token":"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2ODk5NjQyNDgsImlhdCI6MTY4NzM3MjI0OCwic3ViIjoxfQ.RyYsShoPiDAXyb0q72TOWMFZFVfTpWrHQmxIwTIU1Y8",
"email":"[email protected]",
"username":null,
"given_name":null,
"family_name":null
}
```

## Commands

### Debugging

| Command | Result |
|:---|---|
|`docker-compose logs`| Shows logs of all docker-compose related containers.|
|`docker exec -ti postgres-db psql -U postgres`| Runs psql.|
|`docker-compose exec bash`| Runs bash in container_name container. See the example below.|

> `docker-compose logs -f ` shows only the logs of the container. For example, `docker-compose logs -f flask-api` shows flask web service logs while `docker-compose logs -f postgres-db` shows PostgreSQL db container logs.

```bash
barreto$ docker-compose exec postgres-db bash
root@ceeb60f9aea8:/# psql -U postgres
psql (10.0)
Type "help" for help.

postgres=# \c db_dev
You are now connected to database "db_dev" as user "postgres".

db_dev=#
```

### Base commands

| Command | Result |
|:---|---|
| `docker-compose exec flask-api flask routes` | Shows the app's endpoints list |
| `docker-compose exec flask-api flask shell` | Runs a shell in the app context |

### DB Creation

| Command | Result |
|:---|---|
| `docker-compose exec flask-api python manage.py recreate_db` | Recreates database by dropping and creating tables.|
| `docker-compose exec flask-api python manage.py seed_db` | Seeds the database |

### DB Migrations

| Command | Result |
|:---|---|
|`docker-compose exec flask-api flask db [OPTIONS] COMMAND [ARGS]...`| Perform database migrations. |

| COMMAND | Result |
|:---|---|
|`branches [OPTIONS]` |Show current branch points.|
|`current [OPTIONS]` |Display the current revision for each database.|
|`downgrade [OPTIONS] [REVISION]` |Revert to a previous version.|
|`edit [OPTIONS] [REVISION]` |Edit a revision file.|
|`heads [OPTIONS]` |Show current available heads in the script directory.|
|`history [OPTIONS]` |List changeset scripts in chronological order.|
|`init [OPTIONS]` |Creates a new migration repository.|
|`merge [OPTIONS] [REVISIONS]...` |Merge two revisions together, creating a new revision file|
|`migrate [OPTIONS]` |Autogenerate a new revision file (Alias for `revision --autogenerate`)|
|`revision [OPTIONS]` |Create a new revision file.|
|`show [OPTIONS] [REVISION]` |Show the revision denoted by the given symbol|
|`stamp [OPTIONS] [REVISION]` |'stamp' the revision table with the given revision; don't run any migrations|
|`upgrade [OPTIONS] [REVISION]` |Upgrade to a later version|

> you can see all DB migration commands documentation by executing `docker-compose run flask-api flask db --help`
> For a particular command documentation you can execute `docker-compose run flask-api flask db [COMMAND] --help`

### Run Tests

| Command | Result |
|:---|---|
|`docker-compose exec flask-api python manage.py test`| Runs all integration tests|
|`docker-compose exec flask-api python manage.py test `| Runs all integration tests that matches the file-name or pattern |

> Make sure you run tests on the `development` target. It does not work on `debug` target, which uses debugpy library.
> There are more than 54 integration test covering the REST endpoints. Check out test implementations inside [test](/flask-api/tests/) folder.

The most convenient way to run tests on demand is by using VSCode UI.

![vs-code UI testing](.github/files/ui-tests.gif)

## Dependencies

* Python 3.11.3
* Docker 20.10.11+
* Docker Compose 1.29.2+
* Flask v2.3.2

For a complete list of Python dependencies, check out [requiremets.txt](/flask-api/requirements.txt) file.

## RESTful endpoints

Run `docker-compose exec flask-api flask routes` to see the REST API endpoints list.

```bash
Endpoint Methods Rule
------------------------------------------ ------- -------------------------------------
auth.facebook_login POST /v1/auth/facebook/login
auth.get_user_status GET /v1/auth/status
auth.login_user POST /v1/auth/login
auth.logout_user GET /v1/auth/logout
auth.password_change PUT /v1/auth/password_change
auth.password_recovery POST /v1/auth/password_recovery
auth.password_reset PUT /v1/auth/password
auth.register_user POST /v1/auth/register
auth.set_standalone_user PUT /v1/auth/facebook/set_standalone_user
devices.connect_device_with_logged_in_user PUT /v1/devices/
devices.register_device POST /v1/devices
email_validation.email_verification PUT /v1/email_verification
email_validation.verify_email GET /v1/email_verification/
phone_validation.register_user_cellphone POST /v1/cellphone
phone_validation.verify_user_cellphone PUT /v1/cellphone/verify
static GET /static/
users.add_user POST /v1/users
users.get_all_users GET /v1/users
users.get_single_user GET /v1/users/
users.ping_pong GET /v1/ping
users.push_echo POST /v1/push_echo
```

### Sanity Check

| Endpoint | HTTP Method | Result |
|:---|:---:|---|
| `/ping` | `GET` | Sanity check |

### Authentication

| Endpoint | HTTP Method | Result |
|:---|:---:|---|
| `/auth/register` | `POST` | Registers a new user |
| `/auth/login` | `POST` | Login the user |
| `/auth/logout` | `GET` | User logout |
| `/auth/status` | `GET` | Returns the logged in user's status |
| `/auth/password_recovery` | `POST` | Creates a password_recovery_hash and sends email to user |
| `/auth/password` | `PUT` | Reset user password |
| `/auth/password_change` | `PUT` | Changes user password |
| `/auth/facebook/login` | `POST` | Logs in user using fb_access_token returning the corresponding JWT. if the user does not exist registers/creates a new one |
| `/auth/facebook/set_standalone_user` | `PUT` | Sets username and password to work directly on the system without Facebook |

> Endpoints implementation can be found under [/project/api/v1/auth.py](project/api/v1/auth.py).

### Cell phone number validation

| Endpoint | HTTP Method | Result |
|:---|:---:|---|
| `/cellphone` | `POST` | Generates cellphone_validation_code, idempotent (could be used for resend cellphone_validation_code) allows just 1 user per cellphone validation! |
| `/cellphone/verify` | `PUT` | Verifies cellphone_validation_code, idempotent (could be used many times) |

> Endpoints implementation can be found under [/project/api/v1/phone_validation.py](project/api/v1/phone_validation.py).

### Devices (Push notifications support)

| Endpoint | HTTP Method | Result |
|:---|:---:|---|
| `/devices` | `POST` | Creates or updates the device in the system |
| `/devices/` | `PUT` | creates/updates and associates the device device_id to the user logged_in_user_id |

> Endpoints implementation can be found under [/project/api/v1/devices.py](project/api/v1/devices.py).

### Email validation

|Endpoint| HTTP Method | Result |
|:---|:---:|---|
| `/email_verification` | `PUT` | Creates a email_token_hash and sends email with token to user (assumes login=email), idempotent (could be use for resend) |
| `/email_verification/` | `GET` | Sets email verified date |

> Endpoints implementation can be found under [/project/api/v1/email_validation.py](project/api/v1/email_validation.py).

## FAQ

### How do I add a new API endpoint version..?

`flask-base-api` supports backwards compatibility, which is crucial for mobile api development. API services are added under [/v1](/flask-api/project/api/v1/) url prefix by default.

In order to update endpoints and still support v1, simply regist blueprints with url-prefix `/v2` and do not change `/v1` endpoints. Whenever an endpoint needs to be updated, copy `v1` version and paste it in `v2` project folder, then make necesarry changes over `v2`.

```python
app.register_blueprint(auth_blueprint, url_prefix='/v1')
app.register_blueprint(auth_blueprint2, url_prefix='/v2')
...
..
.
```

### How do I change/add app configs?

`flask-base-api` support multiple environments. Check out config values at [/flask-api/project/config.py](/flask-api/project/config.py).

### How does error handling works?

`flask-base-api` registers multiple error handlers for most common exceptions types.

```python
# register error handlers
from project.api.common.utils import exceptions
from project.api.common import error_handlers
InvalidPayload, error_handlers.handle_exception)
app.register_error_handler(exceptions.BusinessException, error_handlers.handle_exception)
app.register_error_handler(exceptions.UnauthorizedException, error_handlers.handle_exception)
app.register_error_handler(exceptions.ForbiddenException, error_handlers.handle_exception)
app.register_error_handler(exceptions.NotFoundException, error_handlers.handle_exception)
app.register_error_handler(exceptions.ServerErrorException, error_handlers.handle_exception)
app.register_error_handler(Exception, error_handlers.handle_general_exception)
```

You can raise `InvalidPayload`, `BusinessException`, `UnauthorizedException`, `ForbiddenException`, `NotFoundException`, `ServerErrorException` from within endpoint implementations, which is formatted into JSON format by error handlers. For instance:

```python
raise UnauthorizedException(message='Something went wrong. Please contact us.')
```

and the error handler returns

> 401 Unauthorized

```json
{"message":"Something went wrong. Please contact us.","status":"error"}
```

You can easily implement a custom exception by extending the `APIException` type, the same type that `flask-base-api` exceptions extend.

```python
class APIException(Exception):

def __init__(self, message, status_code, payload):
super().__init__()
self.message = message
self.status_code = status_code
self.payload = payload

def to_dict(self):
rv = dict(self.payload or ())
rv['message'] = self.message
rv['status'] = 'error'
return rv
```

We can return a response message for the new exception type by registering a handler like:

```python
app.register_error_handler(MyNewException, error_handlers.handle_general_exception)
```

### How do I implement an authenticated service?

`flask-base-api` use JWT authentication.

By using `@authenticate` decorator you can easily force JWT check and authentication for the endpoint like the following method indicates. If authentication goes well, `user_id` is passed as argument to the endpoint function. If something goes wrong, it raises `UnauthorizedException()` and the server responds with a 401 Unauthorized response.

```python
@phone_validation_blueprint.route('/cellphone/verify', methods=['PUT'])
@accept('application/json')
@authenticate
def verify_user_cellphone(user_id: int):
''' verifies cellphone_validation_code, idempotent (could be used many times) '''
post_data = request.get_json()
if not post_data:
raise InvalidPayload()
validation_code = post_data.get('validation_code')
user = User.get(user_id)
...
..
.
```

Curious about how the `@authenticate` decorator is implemented?

```python
def authenticate(f):
@wraps(f)
def decorated_function(*args, **kwargs):
auth_header = request.headers.get('Authorization')
if not auth_header:
raise UnauthorizedException()
auth_token = auth_header.split(" ")[1]
user_id = User.decode_auth_token(auth_token)
user = User.get(user_id)
if not user or not user.active:
raise UnauthorizedException(message='Something went wrong. Please contact us.')
return f(user_id, *args, **kwargs)
return decorated_function
```

### How do I test an endpoint?

Tests are implemented using [Flask-Testing](https://github.com/jarus/flask-testing) library. All testing files are placed under [/tests](tests). There are more than 50 test cases already implemented.

You can create a new test case class in a new file which must start with `test_`, and ideally, it should extend to the `BaseTestCase` class which loads testing configuration and clean up db state after each test case execution. Test case method name must be prefixed with `test`.

After implementing the new test case, you can run it. Go to [here](#run-tests) to see how.

### How do I use Swagger to play with the API?

Now you can check swagger RESTful API documentation by visiting http://localhost:8080.

![swagger](.github/files/swagger.png)

RESTful API is available under http://localhost:5001/v1. You can also use port 80 since Nginx was also set up. [http://localhost/v1/ping](http://localhost/v1/ping) should return `{"status":"success","message":"pong!"}`.

### How do I use pgAdmin?

First, get the Postgres db ip address by executing

```bash
docker inspect postgres-db | grep "IPAddress"
```

which outputs the `IPAddress`

```bash
"SecondaryIPAddresses": null,
"IPAddress": "",
"IPAddress": "192.168.224.3",
```

Then open `pgAdmin` in your browser `http://0.0.0.0:5050/`, and log in by using email: `[email protected]`, password: `password` credentials.

Then connect to db server using the ip address retrieved above and using the following Postgres credentials: user: `postgres`, password: `postgres`.

![pgAdmin](.github/files/pgAdmin.png)