Add running with docker compose.

Author: jslvtr

config.py

@@ -0,0 +1,43 @@
+from functools import lru_cache
+from typing import Optional
+
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class BaseConfig(BaseSettings):
+    ENV_STATE: Optional[str] = None
+
+    model_config = SettingsConfigDict(env_file=".env", extra="ignore")
+
+
+class GlobalConfig(BaseConfig):
+    DATABASE_URL: Optional[str] = None
+    DB_FORCE_ROLL_BACK: bool = False
+    MAILGUN_DOMAIN: Optional[str] = None
+    MAILGUN_API_KEY: Optional[str] = None
+    REDIS_URL: Optional[str] = None
+
+
+class DevConfig(GlobalConfig):
+    model_config = SettingsConfigDict(env_prefix="DEV_", extra="ignore")
+
+
+class ProdConfig(GlobalConfig):
+    model_config = SettingsConfigDict(env_prefix="PROD_", extra="ignore")
+
+
+class TestConfig(GlobalConfig):
+    DATABASE_URL: str = "sqlite:///test.db"
+    DB_FORCE_ROLL_BACK: bool = True
+    REDIS_URL: str = ""
+
+    model_config = SettingsConfigDict(env_prefix="TEST_", extra="ignore")
+
+
+@lru_cache()
+def get_config(env_state: str):
+    configs = {"dev": DevConfig, "prod": ProdConfig, "test": TestConfig}
+    return configs[env_state]()
+
+
+config = get_config(BaseConfig().ENV_STATE)

docker-compose.yml

@@ -0,0 +1,38 @@
+services:
+  web:
+    build: .
+    ports:
+      - "5000:80"
+    depends_on:
+      db:
+        condition: service_healthy
+      redis:
+        condition: service_started
+    env_file:
+      - ./.env
+    volumes:
+      - .:/app
+  db:
+    image: postgres
+    environment:
+      - POSTGRES_PASSWORD=password
+      - POSTGRES_DB=myapp
+    volumes:
+      - postgres_data:/var/lib/postgresql/data
+    healthcheck:
+      test: pg_isready -d $${POSTGRES_DB} -U postgres
+      interval: 2s
+      retries: 10
+  redis:
+    image: redis:alpine
+  rq_worker:
+    build: .
+    depends_on:
+      - redis
+    env_file:
+      - ./.env
+    volumes:
+      - .:/app
+    command: /bin/bash -c "cd /app && rq worker -c settings"
+volumes:
+  postgres_data:

requirements-dev.txt

@@ -1,3 +1,4 @@
 pytest
 black
-flake8
+flake8
+fakeredis

requirements.txt

@@ -9,4 +9,6 @@ gunicorn
 redis
 rq
 psycopg2
-requests
+requests
+pydantic
+pydantic_settings

resources/user.py

@@ -1,30 +1,28 @@
-import os
+
 import redis
 from flask.views import MethodView
+from flask_jwt_extended import (create_access_token, create_refresh_token,
+                                get_jwt, get_jwt_identity, jwt_required)
 from flask_smorest import Blueprint, abort
-from flask_jwt_extended import (
-    create_access_token,
-    create_refresh_token,
-    get_jwt_identity,
-    get_jwt,
-    jwt_required,
-)
 from passlib.hash import pbkdf2_sha256
-from sqlalchemy import or_
 from rq import Queue
+from sqlalchemy import or_
 
-from tasks import send_user_registration_email
-
+import config
+from blocklist import BLOCKLIST
 from db import db
 from models import UserModel
-from schemas import UserSchema, UserRegisterSchema
-from blocklist import BLOCKLIST
+from schemas import UserRegisterSchema, UserSchema
+from tasks import send_user_registration_email
 
 blp = Blueprint("Users", "users", description="Operations on users")
-connection = redis.from_url(
-    os.getenv("REDIS_URL")
-)  # Get this from Render.com or run in Docker
-queue = Queue("emails", connection=connection)
+if isinstance(config.config, config.TestConfig):
+    import fakeredis
+    connection = fakeredis.FakeStrictRedis()
+    queue = Queue("emails", is_async=False, connection=connection)
+else:
+    connection = redis.from_url(config.config.REDIS_URL)
+    queue = Queue("emails", connection=connection)
 
 
 @blp.route("/register")

settings.py

@@ -1,7 +1,8 @@
 import os
+import config
 from dotenv import load_dotenv
 
 load_dotenv()
 
-REDIS_URL = os.getenv("REDIS_URL", "redis://localhost:6379")
+REDIS_URL = config.config.REDIS_URL
 QUEUES = ["emails", "default"]