データベースのマイグレーション方法

Question

前提・実現したいこと

fastapiを利用して、apiに関して学習を進めようと思っております。
参考として、下記記事を参考にさせていただきながら環境構築を実施しております。
https://nmomos.com/tips/2021/01/23/fastapi-docker-1/
ただし、本記事ではPostgreSQLを使用していますが、学習にはMySQLを使用したいと考えております。

発生している問題・エラーメッセージ

記事のその2に記載してある、データベースのマイグレートの部分で「alembic upgrade head」を打つと、
以下エラーが出てしまっている状態です。
MySQLのコンテナは立ち上がっていることを確認できており、
「alembic revision -m ""」にてマイグレート用のスクリプトファイルも作成できていることを確認しております。

/backend # alembic upgrade head
Traceback (most recent call last):
  File "/usr/local/bin/alembic", line 8, in <module>
    sys.exit(main())
  File "/usr/local/lib/python3.9/site-packages/alembic/config.py", line 559, in main
    CommandLine(prog=prog).main(argv=argv)
  File "/usr/local/lib/python3.9/site-packages/alembic/config.py", line 553, in main
    self.run_cmd(cfg, options)
  File "/usr/local/lib/python3.9/site-packages/alembic/config.py", line 530, in run_cmd
    fn(
  File "/usr/local/lib/python3.9/site-packages/alembic/command.py", line 294, in upgrade
    script.run_env()
  File "/usr/local/lib/python3.9/site-packages/alembic/script/base.py", line 481, in run_env
    util.load_python_file(self.dir, "env.py")
  File "/usr/local/lib/python3.9/site-packages/alembic/util/pyfiles.py", line 97, in load_python_file
    module = load_module_py(module_id, path)
  File "/usr/local/lib/python3.9/site-packages/alembic/util/compat.py", line 182, in load_module_py
    spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 850, in exec_module
  File "<frozen importlib._bootstrap>", line 228, in _call_with_frames_removed
  File "./app/db/migrations/env.py", line 33, in <module>
    run_migrations_online()
  File "./app/db/migrations/env.py", line 18, in run_migrations_online
    connectable = engine_from_config(
  File "/usr/local/lib/python3.9/site-packages/sqlalchemy/engine/__init__.py", line 563, in engine_from_config
    return create_engine(url, **options)
  File "/usr/local/lib/python3.9/site-packages/sqlalchemy/engine/__init__.py", line 520, in create_engine
    return strategy.create(*args, **kwargs)
  File "/usr/local/lib/python3.9/site-packages/sqlalchemy/engine/strategies.py", line 87, in create
    dbapi = dialect_cls.dbapi(**dbapi_args)
  File "/usr/local/lib/python3.9/site-packages/sqlalchemy/dialects/mysql/mysqldb.py", line 118, in dbapi
    return __import__("MySQLdb")
ModuleNotFoundError: No module named 'MySQLdb'

該当のソースコード

以下MySQLに接続するためにコードを書き換えていますが、
この変更した部分が影響しているのかが分かりません。

config.py(変更前)

from databases import DatabaseURL
from starlette.config import Config
from starlette.datastructures import Secret


config = Config(".env")

PROJECT_NAME = "Hedgehog Reservation"
VERSION = "1.0.0"
API_PREFIX = "/api"

SECRET_KEY = config("SECRET_KEY", cast=Secret, default="CHANGEME")
POSTGRES_USER = config("POSTGRES_USER", cast=str)
POSTGRES_PASSWORD = config("POSTGRES_PASSWORD", cast=Secret)
POSTGRES_SERVER = config("POSTGRES_SERVER", cast=str, default="db")
POSTGRES_PORT = config("POSTGRES_PORT", cast=str, default="5432")
POSTGRES_DB = config("POSTGRES_DB", cast=str)

DATABASE_URL = config(
    "DATABASE_URL",
    cast=DatabaseURL,
    default=f"postgresql://{POSTGRES_USER}:{POSTGRES_PASSWORD}@{POSTGRES_SERVER}:{POSTGRES_PORT}/{POSTGRES_DB}"

config.py(変更後)

from databases import DatabaseURL
from starlette.config import Config
from starlette.datastructures import Secret

config = Config(".env")

PROJECT_NAME = "SAMPLE FAST API"
VERSION = "1.0.0"
API_PREFIX = "/api"

SECRET_KEY = config("SECRET_KEY", cast=Secret, default="CHANGEME")
MYSQL_USER = config("MYSQL_USER", cast=str)
MYSQL_PASSWORD = config("MYSQL_PASSWORD", cast=Secret)
MYSQL_SERVER = config("MYSQL_SERVER", cast=str, default="db")
MYSQL_PORT = config("MYSQL_PORT", cast=str, default="5432")
MYSQL_DB = config("MYSQL_DB", cast=str)

DATABASE_URL = config(
    "DATABASE_URL",
    cast=DatabaseURL,
    default=f"mysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_SERVER}:{MYSQL_PORT}/{MYSQL_DB}"
)

試したこと

別記事で本エラーメッセージが出た場合の対処法として以下コマンドを打つと治るとあったので、
実行してみましたが、現状変わらずでした。

pip3 install mysqlclient

Answer 1

自己解決しました。

default=f"mysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_SERVER}:{MYSQL_PORT}/{MYSQL_DB}"

上記URL部分を下記に変更後、問題解消しました。

default=f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_SERVER}:{MYSQL_PORT}/{MYSQL_DB}"