追跡対象動画バルク UPSERT (#22) (#23)

#22 インポート漏れ修正

#22

#22

Co-authored-by: miteruzo <miteruzo@naver.com>
Reviewed-on: #23

db/models.py

@@ -64,6 +64,12 @@ class TrackedVideo (Model):
 
     __timestamps__ = False
 
+    def upsert (
+            self,
+            *args:  str,
+    ) -> None:
+        super ().upsert ('code')
+
 
 class User (Model):
     id:     int

tracked_videos/put_bulk_upsert.py

@@ -0,0 +1,19 @@
+import sys
+
+from db.config import DB
+from db.models import TrackedVideo
+
+DB
+
+
+def main (
+        video_codes:    list[str],
+) -> None:
+    for code in video_codes:
+        tv = TrackedVideo ()
+        tv.code = code
+        tv.upsert ()
+
+
+if __name__ == '__main__':
+    main (sys.argv[1:])