add method for import data

kontor-scripts/db/models/media.py

@@ -2,7 +2,7 @@ import re
 import subprocess
 from datetime import datetime
 from pathlib import Path
-
+from typing import Any, AnyStr, Dict
 import requests
 from bs4 import BeautifulSoup
 from sqlalchemy import Boolean, Column, False_, String, ForeignKey
@@ -21,6 +21,19 @@ class MediaFile(Base, BaseMixin, BaseVideoMixin):
     def __str__(self):
         return f'{self.title}({self.id})'
 
+    def import_dict(self, import_data: Dict[AnyStr, Any]):
+        self.id = import_data['id']
+        self.created_date = import_data['created_date']
+        self.last_modified_date = import_data['last_modified_date']
+        self.version = import_data['version']
+        self.cloud_link = import_data['cloud_link']
+        self.file_name = import_data['file_name']
+        self.path = import_data['path']
+        self.review = import_data['review']
+        self.title = import_data['title']
+        self.url = import_data['url']
+        self.should_download = import_data['should_download']
+
     def update_title(self) -> None:
         print(f"update title for {self.url}")
         try:
@@ -72,6 +85,13 @@ class MediaActor(Base, BaseMixin):
     name = Column(String)
     media_actor_files = relationship("MediaActorFile")
 
+    def import_dict(self, import_data: Dict[AnyStr, Any]):
+        self.id = import_data['id']
+        self.created_date = import_data['created_date']
+        self.last_modified_date = import_data['last_modified_date']
+        self.version = import_data['version']
+        self.name = import_data['name']
+
 
 class MediaActorFile(Base, BaseMixin):
     __tablename__ = 'media_actor_file'
@@ -80,6 +100,15 @@ class MediaActorFile(Base, BaseMixin):
     media_file_id = Column(String, ForeignKey("media_file.id"), nullable=True)
     media_file = relationship("MediaFile", back_populates="media_actor_files")
 
+    def import_dict(self, import_data: Dict[AnyStr, Any]):
+        self.id = import_data['id']
+        self.created_date = import_data['created_date']
+        self.last_modified_date = import_data['last_modified_date']
+        self.version = import_data['version']
+        self.media_actor_id = import_data['media_actor_id']
+        self.media_file_id = import_data['media_file_id']
+
+
 
 class MediaArticle(Base, BaseMixin):
     __tablename__ = 'media_article'
@@ -87,6 +116,16 @@ class MediaArticle(Base, BaseMixin):
     title = Column(String)
     url = Column(String, unique=True)
 
+    def import_dict(self, import_data: Dict[AnyStr, Any]):
+        self.id = import_data['id']
+        self.created_date = import_data['created_date']
+        self.last_modified_date = import_data['last_modified_date']
+        self.version = import_data['version']
+        self.review = import_data['review']
+        self.title = import_data['title']
+        self.url = import_data['url']
+
+
 
 class MediaVideo(Base, BaseMixin):
     __tablename__ = 'media_video'
@@ -97,3 +136,16 @@ class MediaVideo(Base, BaseMixin):
     title = Column(String)
     url = Column(String, unique=True)
     should_download = Column(Boolean)
+
+    def import_dict(self, import_data: Dict[AnyStr, Any]):
+        self.id = import_data['id']
+        self.created_date = import_data['created_date']
+        self.last_modified_date = import_data['last_modified_date']
+        self.version = import_data['version']
+        self.cloud_link = import_data['cloud_link']
+        self.file_name = import_data['file_name']
+        self.path = import_data['path']
+        self.review = import_data['review']
+        self.title = import_data['title']
+        self.url = import_data['url']
+        self.should_download = import_data['should_download']