用OOP设计一个简单的IMDB风格的数据库

class Movie:
    def __init__(self, name, director, release_date, rating):
        self.name = name
        self.director = director
        self.release_date = release_date
        self.rating = rating
        self.actors = []

    def get_info(self):
        return f"Movie: {self.name}\nDirector: {self.director}\nRelease Date: {self.release_date}\nRating: {self.rating}"

    def add_actor(self, actor):
        self.actors.append(actor)

    def update_rating(self, rating):
        self.rating = rating


class Actor:
    def __init__(self, name, gender, birth_date):
        self.name = name
        self.gender = gender
        self.birth_date = birth_date

    def get_info(self):
        return f"Actor: {self.name}\nGender: {self.gender}\nBirth Date: {self.birth_date}"


class Database:
    def __init__(self):
        self.movies = []
        self.actors = []

    def add_movie(self, movie):
        self.movies.append(movie)

    def remove_movie(self, movie):
        self.movies.remove(movie)

    def search_movie(self, keyword):
        result = []
        for movie in self.movies:
            if keyword.lower() in movie.name.lower():
                result.append(movie)
        return result

    def add_actor(self, actor):
        self.actors.append(actor)

    def remove_actor(self, actor):
        self.actors.remove(actor)

# 创建电影和演员对象
movie1 = Movie("The Shawshank Redemption", "Frank Darabont", "1994-09-23", 9.3)
actor1 = Actor("Tim Robbins", "Male", "1958-10-16")

# 添加演员到电影
movie1.add_actor(actor1)

# 创建数据库对象
database = Database()

# 添加电影和演员到数据库
database.add_movie(movie1)
database.add_actor(actor1)

# 搜索电影
results = database.search_movie("redemption")
for movie in results:
    print(movie.get_info())

# 输出：
# Movie: The Shawshank Redemption
# Director: Frank Darabont
# Release Date: 1994-09-23
# Rating: 9.3