Merge pull request #4 from OZ-Coding-School/day8

Day8

Author: Meoyoug

.env

@@ -0,0 +1,6 @@
+SECRET_KEY=secretkey123123124133123
+MYSQL_HOST=127.0.0.1
+MYSQL_PORT=3306
+MYSQL_USER=root
+MYSQL_PASSWORD=1234
+MYSQL_DB=fastapi_assignment_hard

.gitignore

@@ -4,3 +4,4 @@ __pycache__/
 .venv/
 .coverage
 .dmypy.json
+media/

conftest.py

@@ -1,12 +1,43 @@
+import asyncio
+from typing import Any, Generator
+from unittest.mock import Mock, patch
+
 import pytest
+from pytest import FixtureRequest
+from tortoise.backends.base.config_generator import generate_config
+from tortoise.contrib.test import finalizer, initializer
 
-from src.models.movies import MovieModel
-from src.models.users import UserModel
+from src.configs import config
+from src.configs.database import TORTOISE_APP_MODELS
 
 TEST_BASE_URL = "http://test"
+TEST_DB_LABEL = "models"
+TEST_DB_TZ = "Asia/Seoul"
+
+
+def get_test_db_config() -> dict[str, Any]:
+    tortoise_config = generate_config(
+        db_url=f"mysql://{config.MYSQL_USER}:{config.MYSQL_PASSWORD}@{config.MYSQL_HOST}:{config.MYSQL_PORT}/test",
+        app_modules={TEST_DB_LABEL: TORTOISE_APP_MODELS},
+        connection_label=TEST_DB_LABEL,
+        testing=True,
+    )
+    tortoise_config["timezone"] = TEST_DB_TZ
+
+    return tortoise_config
+
+
+@pytest.fixture(scope="session", autouse=True)
+def initialize(request: FixtureRequest) -> Generator[None, None, None]:
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    with patch("tortoise.contrib.test.getDBConfig", Mock(return_value=get_test_db_config())):
+        initializer(modules=TORTOISE_APP_MODELS)
+    yield
+    finalizer()
+    loop.close()
 
 
-@pytest.fixture(scope="function", autouse=True)
-def user_model_clear() -> None:
-    UserModel.clear()
-    MovieModel.clear()
+@pytest.fixture(scope="session", autouse=True)
+def event_loop() -> None:
+    pass

main.py

@@ -1,22 +1,25 @@
 from fastapi import FastAPI
 
+from src.configs.database import initialize_tortoise
 from src.middleware.auth import AuthMiddleware
-from src.models.movies import MovieModel
-from src.models.users import UserModel
+from src.routers.like_router import like_router
 from src.routers.movie_router import movie_router
+from src.routers.review_router import review_router
 from src.routers.user_router import user_router
 
 app = FastAPI()
 
 # include custom middleware
 app.add_middleware(AuthMiddleware)
+
 # include router in app
 app.include_router(user_router)
 app.include_router(movie_router)
+app.include_router(review_router)
+app.include_router(like_router)
 
-# create dummy for test
-UserModel.create_dummy()
-MovieModel.create_dummy()
+# initialize_tortoise-orm
+initialize_tortoise(app=app)
 
 if __name__ == "__main__":
     import uvicorn

migrations/models/0_20250326144351_init.py

@@ -0,0 +1,35 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        CREATE TABLE IF NOT EXISTS `users` (
+    `id` BIGINT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `created_at` DATETIME(6) NOT NULL  DEFAULT CURRENT_TIMESTAMP(6),
+    `username` VARCHAR(50) NOT NULL UNIQUE,
+    `hashed_password` VARCHAR(128) NOT NULL,
+    `age` INT NOT NULL,
+    `gender` VARCHAR(6) NOT NULL  COMMENT 'MALE: male\nFEMALE: female',
+    `last_login` DATETIME(6),
+    KEY `idx_users_usernam_266d85` (`username`)
+) CHARACTER SET utf8mb4;
+CREATE TABLE IF NOT EXISTS `movies` (
+    `id` BIGINT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `created_at` DATETIME(6) NOT NULL  DEFAULT CURRENT_TIMESTAMP(6),
+    `title` VARCHAR(255) NOT NULL,
+    `plot` LONGTEXT NOT NULL,
+    `cast` JSON NOT NULL,
+    `playtime` INT NOT NULL,
+    `genre` VARCHAR(9) NOT NULL  COMMENT 'SF: SF\nADVENTURE: Adventure\nROMANCE: Romance\nCOMIC: Comic\nFANTASY: Fantasy\nSCIENCE: Science\nMYSTERY: Mystery\nACTION: Action\nHORROR: Horror'
+) CHARACTER SET utf8mb4;
+CREATE TABLE IF NOT EXISTS `aerich` (
+    `id` INT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `version` VARCHAR(255) NOT NULL,
+    `app` VARCHAR(100) NOT NULL,
+    `content` JSON NOT NULL
+) CHARACTER SET utf8mb4;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        """

migrations/models/1_20250326144407_update.py

@@ -0,0 +1,13 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `users` ADD `profile_image_url` VARCHAR(255);
+        ALTER TABLE `movies` ADD `poster_image_url` VARCHAR(255);"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `users` DROP COLUMN `profile_image_url`;
+        ALTER TABLE `movies` DROP COLUMN `poster_image_url`;"""

migrations/models/2_20250331101901_update.py

@@ -0,0 +1,33 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        CREATE TABLE IF NOT EXISTS `reviews` (
+    `id` BIGINT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `created_at` DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+    `title` VARCHAR(50) NOT NULL,
+    `content` VARCHAR(255) NOT NULL,
+    `review_image_url` VARCHAR(255),
+    `movie_id` BIGINT NOT NULL,
+    `user_id` BIGINT NOT NULL,
+    UNIQUE KEY `uid_reviews_user_id_44b823` (`user_id`, `movie_id`),
+    CONSTRAINT `fk_reviews_movies_56a147b9` FOREIGN KEY (`movie_id`) REFERENCES `movies` (`id`) ON DELETE CASCADE,
+    CONSTRAINT `fk_reviews_users_8aed0759` FOREIGN KEY (`user_id`) REFERENCES `users` (`id`) ON DELETE CASCADE
+) CHARACTER SET utf8mb4;
+        CREATE TABLE IF NOT EXISTS `review_likes` (
+    `id` BIGINT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `created_at` DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+    `is_liked` BOOL NOT NULL DEFAULT 1,
+    `review_id` BIGINT NOT NULL,
+    `user_id` BIGINT NOT NULL,
+    UNIQUE KEY `uid_review_like_user_id_c69f9e` (`user_id`, `review_id`),
+    CONSTRAINT `fk_review_l_reviews_6cb49859` FOREIGN KEY (`review_id`) REFERENCES `reviews` (`id`) ON DELETE CASCADE,
+    CONSTRAINT `fk_review_l_users_5cd4a3e1` FOREIGN KEY (`user_id`) REFERENCES `users` (`id`) ON DELETE CASCADE
+) CHARACTER SET utf8mb4;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        DROP TABLE IF EXISTS `reviews`;
+        DROP TABLE IF EXISTS `review_likes`;"""

migrations/models/3_20250403172912_update.py

@@ -0,0 +1,37 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        CREATE TABLE IF NOT EXISTS `genres` (
+    `id` BIGINT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `created_at` DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+    `external_id` INT NOT NULL UNIQUE,
+    `name` VARCHAR(255) NOT NULL
+) CHARACTER SET utf8mb4;
+        ALTER TABLE `movies` ADD `release_date` DATETIME(6) NOT NULL;
+        ALTER TABLE `movies` RENAME COLUMN `plot` TO `overview`;
+        ALTER TABLE `movies` RENAME COLUMN `playtime` TO `runtime`;
+        ALTER TABLE `movies` ADD `external_id` INT NOT NULL UNIQUE;
+        ALTER TABLE `movies` DROP COLUMN `genre`;
+        ALTER TABLE `movies` MODIFY COLUMN `cast` VARCHAR(255) NOT NULL;
+        CREATE TABLE IF NOT EXISTS `movie_genres` (
+    `id` BIGINT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `created_at` DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),
+    `genre_id` BIGINT NOT NULL,
+    `movie_id` BIGINT NOT NULL,
+    CONSTRAINT `fk_movie_ge_genres_c7d0356e` FOREIGN KEY (`genre_id`) REFERENCES `genres` (`id`) ON DELETE CASCADE,
+    CONSTRAINT `fk_movie_ge_movies_014c0d12` FOREIGN KEY (`movie_id`) REFERENCES `movies` (`id`) ON DELETE CASCADE
+) CHARACTER SET utf8mb4;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `movies` ADD `genre` VARCHAR(9) NOT NULL COMMENT 'SF: SF\nADVENTURE: Adventure\nROMANCE: Romance\nCOMIC: Comic\nFANTASY: Fantasy\nSCIENCE: Science\nMYSTERY: Mystery\nACTION: Action\nHORROR: Horror';
+        ALTER TABLE `movies` RENAME COLUMN `runtime` TO `playtime`;
+        ALTER TABLE `movies` RENAME COLUMN `overview` TO `plot`;
+        ALTER TABLE `movies` DROP COLUMN `release_date`;
+        ALTER TABLE `movies` DROP COLUMN `external_id`;
+        ALTER TABLE `movies` MODIFY COLUMN `cast` JSON NOT NULL;
+        DROP TABLE IF EXISTS `movie_genres`;
+        DROP TABLE IF EXISTS `genres`;"""

migrations/models/4_20250404140022_update.py

@@ -0,0 +1,11 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `movies` MODIFY COLUMN `external_id` INT;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `movies` MODIFY COLUMN `external_id` INT NOT NULL;"""

migrations/models/5_20250404140200_update.py

@@ -0,0 +1,15 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        DROP TABLE IF EXISTS `movie_genres`;
+        CREATE TABLE `movies_genres` (
+    `genre_id` BIGINT NOT NULL REFERENCES `genres` (`id`) ON DELETE CASCADE,
+    `movies_id` BIGINT NOT NULL REFERENCES `movies` (`id`) ON DELETE CASCADE
+) CHARACTER SET utf8mb4;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        DROP TABLE IF EXISTS `movies_genres`;"""

migrations/models/6_20250408102900_update.py

@@ -0,0 +1,11 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `movies` MODIFY COLUMN `release_date` DATE NOT NULL;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE `movies` MODIFY COLUMN `release_date` DATETIME(6) NOT NULL;"""