anilife update 2022.10.24(03.)

logic_anilife.py

@@ -756,6 +756,8 @@ class LogicAniLife(LogicModuleBase):
         arg = P.ModelSetting.to_dict()
         arg["sub"] = self.name
         if sub in ["setting", "queue", "list", "category", "request"]:
+            if sub == "request" and req.args.get("content_code") is not None:
+                arg["anilife_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))
@@ -794,6 +796,37 @@ class LogicAniLife(LogicModuleBase):
                 return jsonify(
                     {"ret": "success", "cate": cate, "page": page, "data": data}
                 )
+            elif sub == "complete_list":
+                data = []
+
+                cate = request.form["type"]
+                logger.debug("cate:: %s", cate)
+                page = request.form["page"]
+
+                data = self.get_anime_info(cate, page)
+                # self.current_data = data
+                return jsonify(
+                    {"ret": "success", "cate": cate, "page": page, "data": data}
+                )
+            elif sub == "search":
+                data = []
+                # cate = request.form["type"]
+                # page = request.form["page"]
+                cate = request.form["type"]
+                query = request.form["query"]
+                page = request.form["page"]
+
+                data = self.get_search_result(query, page, cate)
+                # self.current_data = data
+                return jsonify(
+                    {
+                        "ret": "success",
+                        "cate": cate,
+                        "page": page,
+                        "query": query,
+                        "data": data,
+                    }
+                )
             elif sub == "add_queue":
                 logger.debug(f"add_queue routine ===============")
                 ret = {}
@@ -832,11 +865,65 @@ class LogicAniLife(LogicModuleBase):
                 return jsonify("")
             elif sub == "web_list":
                 return jsonify(ModelAniLifeItem.web_list(request))
-
+            elif sub == "db_remove":
+                return jsonify(ModelAniLifeItem.delete_by_id(req.form["id"]))
         except Exception as e:
             P.logger.error("Exception:%s", e)
             P.logger.error(traceback.format_exc())
 
+    @staticmethod
+    def add_whitelist(*args):
+        ret = {}
+
+        logger.debug(f"args: {args}")
+        try:
+
+            if len(args) == 0:
+                code = str(LogicAniLife.current_data["code"])
+            else:
+                code = str(args[0])
+
+            print(code)
+
+            whitelist_program = P.ModelSetting.get("anilife_auto_code_list")
+            # whitelist_programs = [
+            #     str(x.strip().replace(" ", ""))
+            #     for x in whitelist_program.replace("\n", "|").split("|")
+            # ]
+            whitelist_programs = [
+                str(x.strip()) for x in whitelist_program.replace("\n", "|").split("|")
+            ]
+
+            if code not in whitelist_programs:
+                whitelist_programs.append(code)
+                whitelist_programs = filter(
+                    lambda x: x != "", whitelist_programs
+                )  # remove blank code
+                whitelist_program = "|".join(whitelist_programs)
+                entity = (
+                    db.session.query(P.ModelSetting)
+                    .filter_by(key="anilife_auto_code_list")
+                    .with_for_update()
+                    .first()
+                )
+                entity.value = whitelist_program
+                db.session.commit()
+                ret["ret"] = True
+                ret["code"] = code
+                if len(args) == 0:
+                    return LogicAniLife.current_data
+                else:
+                    return ret
+            else:
+                ret["ret"] = False
+                ret["log"] = "이미 추가되어 있습니다."
+        except Exception as e:
+            logger.error("Exception:%s", e)
+            logger.error(traceback.format_exc())
+            ret["ret"] = False
+            ret["log"] = str(e)
+        return ret
+
     def setting_save_after(self):
         if self.queue.get_max_ffmpeg_count() != P.ModelSetting.get_int(
             "anilife_max_ffmpeg_process_count"
@@ -846,7 +933,12 @@ class LogicAniLife(LogicModuleBase):
             )
 
     def scheduler_function(self):
-        pass
+        logger.debug(f"ohli24 scheduler_function::=========================")
+
+        content_code_list = P.ModelSetting.get_list("ohli24_auto_code_list", "|")
+        url = f'{P.ModelSetting.get("anilife_url")}/dailyani'
+        if "all" in content_code_list:
+            ret_data = LogicAniLife.get_auto_anime_info(self, url=url)
 
     def plugin_load(self):
         self.queue = FfmpegQueue(
@@ -882,8 +974,7 @@ class LogicAniLife(LogicModuleBase):
                 .text_content()
                 .strip()
             )
-            # print(des_items1)
-            # print(len(des_items))
+
             des = {}
             des_key = [
                 "_otit",
@@ -1315,6 +1406,10 @@ class ModelAniLifeItem(db.Model):
         )
         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()
@@ -1323,9 +1418,11 @@ class ModelAniLifeItem(db.Model):
     def get_by_anilife_id(cls, anilife_id):
         return db.session.query(cls).filter_by(anilife_id=anilife_id).first()
 
-    def save(self):
-        db.session.add(self)
+    @classmethod
+    def delete_by_id(cls, idx):
+        db.session.query(cls).filter_by(id=idx).delete()
         db.session.commit()
+        return True
 
     @classmethod
     def web_list(cls, req):
@@ -1370,6 +1467,10 @@ class ModelAniLifeItem(db.Model):
         )
         return query
 
+    @classmethod
+    def get_list_uncompleted(cls):
+        return db.session.query(cls).filter(cls.status != "completed").all()
+
     @classmethod
     def append(cls, q):
         item = ModelAniLifeItem()