init

pt_gen/models/movie.py

@@ -0,0 +1,32 @@
+from typing import List, Optional
+from pydantic import BaseModel
+
+class MovieInfo(BaseModel):
+    douban_id: Optional[str] = None
+    imdb_id: Optional[str] = None
+    tmdb_id: Optional[int] = None
+    
+    chinese_title: Optional[str] = None
+    original_title: Optional[str] = None
+    aka_titles: List[str] = []
+    
+    year: Optional[str] = None
+    poster_url: Optional[str] = None
+    
+    countries: List[str] = []
+    genres: List[str] = []
+    release_date: Optional[str] = None
+    runtime: Optional[str] = None
+    douban_rating: Optional[str] = None
+    douban_link: Optional[str] = None
+    imdb_rating: Optional[str] = None
+    imdb_link: Optional[str] = None
+    
+    directors: List[str] = []
+    writers: List[str] = []
+    actors: List[str] = []
+    spoken_languages: List[str] = []
+    tagline: Optional[str] = None
+    synopsis: str = ""
+    awards: List[str] = []
+