2022.10.15(01.)

logic_linkkf.py

@@ -33,6 +33,41 @@ from .plugin import P
 
 
 class LogicLinkkf(LogicModuleBase):
+    db_default = {
+        "linkkf_db_version": "1",
+        "linkkf_url": "https://linkkf.app",
+        "linkkf_download_path": os.path.join(path_data, P.package_name, "linkkf"),
+        "linkkf_auto_make_folder": "True",
+        "linkkf_auto_make_season_folder": "True",
+        "linkkf_finished_insert": "[완결]",
+        "linkkf_max_ffmpeg_process_count": "1",
+        "linkkf_order_desc": "False",
+        "linkkf_auto_start": "False",
+        "linkkf_interval": "* 5 * * *",
+        "linkkf_auto_mode_all": "False",
+        "linkkf_auto_code_list": "all",
+        "linkkf_current_code": "",
+        "linkkf_uncompleted_auto_enqueue": "False",
+        "linkkf_image_url_prefix_series": "",
+        "linkkf_image_url_prefix_episode": "",
+        "linkkf_discord_notify": "True",
+    }
+    current_headers = None
+    current_data = None
+
+    session = requests.Session()
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) "
+        "Chrome/71.0.3578.98 Safari/537.36",
+        "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
+        "Accept-Language": "ko-KR,ko;q=0.9,en-US;q=0.8,en;q=0.7",
+        "Referer": "",
+    }
+    useragent = {
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, "
+        "like Gecko) Chrome/96.0.4664.110 Whale/3.12.129.46 Safari/537.36"
+    }
+
     def __init__(self, P):
         super(LogicLinkkf, self).__init__(P, "setting", scheduler_desc="linkkf 자동 다운로드")
         self.name = "linkkf"
@@ -41,9 +76,9 @@ class LogicLinkkf(LogicModuleBase):
     def process_menu(self, sub, req):
         arg = P.ModelSetting.to_dict()
         arg["sub"] = self.name
-        if sub in ["setting", "queue", "list", "request"]:
+        if sub in ["setting", "queue", "category", "list", "request"]:
             if sub == "request" and req.args.get("content_code") is not None:
-                arg["ani365_current_code"] = req.args.get("content_code")
+                arg["linkkf_current_code"] = req.args.get("content_code")
             if sub == "setting":
                 job_id = "%s_%s" % (self.P.package_name, self.name)
                 arg["scheduler"] = str(scheduler.is_include(job_id))
@@ -56,4 +91,84 @@ class LogicLinkkf(LogicModuleBase):
             )
         return render_template("sample.html", title="%s - %s" % (P.package_name, sub))
 
+    def process_ajax(self, sub, req):
+        try:
+            if sub == "analysis":
+                pass
+            elif sub == "anime_list":
+                pass
+            elif sub == "complete_list":
+                pass
+            elif sub == "search":
+                pass
+            elif sub == "add_queue":
+                pass
+            elif sub == "entity_list":
+                pass
+            elif sub == "queue_command":
+                pass
+            elif sub == "add_queue_checked_list":
+                pass
+            elif sub == "web_list":
+                pass
+            elif sub == "db_remove":
+                pass
+            elif sub == "add_whitelist":
+                pass
+
+        except Exception as e:
+            P.logger.error("Exception:%s", e)
+            P.logger.error(traceback.format_exc())
+
     pass
+
+
+class ModelLinkkfItem(db.Model):
+    __tablename__ = "{package_name}_linkkf_item".format(package_name=P.package_name)
+    __table_args__ = {"mysql_collate": "utf8_general_ci"}
+    __bind_key__ = P.package_name
+    id = db.Column(db.Integer, primary_key=True)
+    created_time = db.Column(db.DateTime)
+    completed_time = db.Column(db.DateTime)
+    reserved = db.Column(db.JSON)
+    content_code = db.Column(db.String)
+    season = db.Column(db.Integer)
+    episode_no = db.Column(db.Integer)
+    title = db.Column(db.String)
+    episode_title = db.Column(db.String)
+    linkkf_va = db.Column(db.String)
+    linkkf_vi = db.Column(db.String)
+    linkkf_id = db.Column(db.String)
+    quality = db.Column(db.String)
+    filepath = db.Column(db.String)
+    filename = db.Column(db.String)
+    savepath = db.Column(db.String)
+    video_url = db.Column(db.String)
+    vtt_url = db.Column(db.String)
+    thumbnail = db.Column(db.String)
+    status = db.Column(db.String)
+    linkkf_info = db.Column(db.JSON)
+
+    def __int__(self):
+        self.created_time == datetime.now()
+
+    def __repr__(self):
+        return repr(self.as_dict())
+
+    def as_dict(self):
+        ret = {x.name: getattr(self, x.name) for x in self.__table__.columns}
+        ret["created_time"] = self.created_time.strftime("%Y-%m-%d %H:%M:%S")
+        ret["completed_time"] = (
+            self.completed_time.strftime("%Y-%m-%d %H:%M:%S")
+            if self.completed_time is not None
+            else None
+        )
+        return ret
+
+    def save(self):
+        db.session.add(self)
+        db.session.commit()
+
+    @classmethod
+    def get_by_id(cls, idx):
+        return db.session.query(cls).filter_by(id=idx).first()