diff --git a/db/models.py b/db/models.py index b254c1b..918580d 100644 --- a/db/models.py +++ b/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 diff --git a/tracked_videos/put_bulk_upsert.py b/tracked_videos/put_bulk_upsert.py new file mode 100644 index 0000000..fa74d89 --- /dev/null +++ b/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:])