dr_py/controllers/service.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# File  : service.py.py
# Author: DaShenHan&道长-----先苦后甜，任凭晚风拂柳颜------
# Date  : 2022/9/6

from base.R import copy_utils
from models.storage import Storage
from models.ruleclass import RuleClass
from models.vipParse import VipParse
from utils.cfg import cfg
from base.database import db
from datetime import datetime,timedelta

class storage_service(object):

    @staticmethod
    def query_all():
        # 查询所有
        res = Storage.query.all()
        return copy_utils.obj_to_list(res)

    def __init__(self):
        conf_list = ['LIVE_URL', 'USE_PY', 'JS_MODE','JS0_DISABLE','JS0_PASSWORD','PLAY_URL', 'PLAY_DISABLE', 'LAZYPARSE_MODE', 'WALL_PAPER_ENABLE',
                     'WALL_PAPER', 'UNAME', 'PWD', 'LIVE_MODE', 'CATE_EXCLUDE', 'TAB_EXCLUDE','SEARCH_TIMEOUT','SEARCH_LIMIT','MULTI_MODE','XR_MODE','JS_PROXY','ENV','ALI_TOKEN']
        for conf in conf_list:
            if not self.hasItem(conf):
                print(f'开始初始化{conf}')
                self.setItem(conf, cfg.get(conf))

    @classmethod
    def getStoreConf(self):
        # MAX_CONTENT_LENGTH 最大上传和端口ip一样是顶级配置,无法外部修改的
        conf_list = ['LIVE_URL', 'LIVE_MODE','PLAY_URL', 'PID_URL','USE_PY','JS_MODE', 'JS0_DISABLE','JS0_PASSWORD','PLAY_DISABLE', 'LAZYPARSE_MODE', 'WALL_PAPER_ENABLE',
                     'WALL_PAPER', 'UNAME', 'PWD',  'CATE_EXCLUDE', 'TAB_EXCLUDE','SEARCH_TIMEOUT','SEARCH_LIMIT','MULTI_MODE','XR_MODE','JS_PROXY','ENV','ALI_TOKEN']
        conf_name_list = ['直播地址', '直播模式','远程地址', '进程管理链接','启用py源', 'js模式','禁用js0','js0密码','禁用免嗅', '免嗅模式', '启用壁纸', '壁纸链接', '管理账号',
                          '管理密码',  '分类排除', '线路排除','聚搜超时','搜索条数','多源模式','仙人模式','源代理','环境变量','阿里tk']
        conf_lists = []
        for i in range(len(conf_list)):
            conf = conf_list[i]
            conf_lists.append({
                'key': conf,
                'value': self.getItem(conf),
                'name': conf_name_list[i]
            })
        return conf_lists

    @classmethod
    def getStoreConfDict(self):
        store_conf = self.getStoreConf()
        store_conf_dict = {}
        for stc in store_conf:
            store_conf_dict[stc['key']] = stc['value']
        return store_conf_dict

    @classmethod
    def getItem(self, key, value=''):
        res = Storage.getItem(key,value)
        if str(res) == '0' or str(res) == 'false' or str(res) == 'False':
            return 0
        return res

    @classmethod
    def hasItem(self, key):
        return Storage.hasItem(key)

    @classmethod
    def setItem(self,key, value):
        return Storage.setItem(key, value)

    @classmethod
    def clearItem(self,key):
        return Storage.clearItem(key)

class rules_service(object):

    @staticmethod
    def query_all():
        # 查询所有
        res = RuleClass.query.order_by(RuleClass.order.asc(),RuleClass.write_date.desc()).all()
        # print(res)
        # res = RuleClass.query.order_by(RuleClass.write_date.asc()).all()
        return copy_utils.obj_to_list(res)

    @classmethod
    def hasItem(self, key):
        return RuleClass.hasItem(key)

    def getState(self,key):
        res = RuleClass.query.filter(RuleClass.name == key).first()
        if not res:
            return 1
        # print(res)
        state = res.state
        if state is None:
            state = 1
        return state or 0


    def setState(self,key,state=0):
        res = RuleClass.query.filter(RuleClass.name == key).first()
        if res:
            res.state = state
            db.session.add(res)
        else:
            res = RuleClass(name=key, state=state)
            db.session.add(res)
            db.session.flush()  # 获取id
        try:
            db.session.commit()
            return res.id
        except Exception as e:
            print(f'发生了错误:{e}')
            return None

    def setOrder(self,key,order=0):
        res = RuleClass.query.filter(RuleClass.name == key).first()
        if res:
            res.order = order
            # print(f'{res.name}设置order为:{order}')
            if res.order == order:
                res.write_date = datetime.now()
                # res.write_date = res.write_date + timedelta(hours=2)
            db.session.add(res)
        else:
            res = RuleClass(name=key, order=order)
            db.session.add(res)
            db.session.flush()  # 获取id
        try:
            db.session.commit()
            return res.id
        except Exception as e:
            print(f'发生了错误:{e}')
            return None

    @staticmethod
    def getHideRules():
        res = RuleClass.query.filter(RuleClass.state == 0).all()
        return copy_utils.obj_to_list(res)

class parse_service(object):

    @staticmethod
    def query_all():
        # 查询所有
        res = VipParse.query.order_by(VipParse.order.asc(),VipParse.write_date.desc()).all()
        # print(res)
        # res = RuleClass.query.order_by(RuleClass.write_date.asc()).all()
        return copy_utils.obj_to_list(res)

    @classmethod
    def hasItem(self, key):
        return VipParse.hasItem(key)

    def getState(self,key):
        res = VipParse.query.filter(VipParse.url == key).first()
        if not res:
            return 1
        # print(res)
        state = res.state
        if state is None:
            state = 1
        return state or 0


    def setState(self,key,state=0):
        res = VipParse.query.filter(VipParse.url == key).first()
        if res:
            res.state = state
            db.session.add(res)
        else:
            res = VipParse(url=key, state=state)
            db.session.add(res)
            db.session.flush()  # 获取id
        try:
            db.session.commit()
            return res.id
        except Exception as e:
            print(f'发生了错误:{e}')
            return None

    def setOrder(self,key,order=0):
        res = VipParse.query.filter(VipParse.url == key).first()
        if res:
            res.order = order
            # print(f'{res.name}设置order为:{order}')
            if res.order == order:
                res.write_date = datetime.now()
                # res.write_date = res.write_date + timedelta(hours=2)
            db.session.add(res)
        else:
            res = VipParse(url=key, order=order)
            db.session.add(res)
            db.session.flush()  # 获取id
        try:
            db.session.commit()
            return res.id
        except Exception as e:
            print(f'发生了错误:{e}')
            return None

    def setEverything(self,key,name,state,typeno,order,ext,header):
        res = VipParse.query.filter(VipParse.url == key).first()
        if res:
            res.name = name
            res.state = state
            res.type = typeno
            res.order = order
            res.ext = ext
            res.header = header
            res.write_date = datetime.now()
            db.session.add(res)
        else:
            res = VipParse(name=name,url=key,state=state,type=typeno,order=order,ext=ext,header=header)
            db.session.add(res)
            db.session.flush()  # 获取id
        try:
            db.session.commit()
            return res.id
        except Exception as e:
            print(f'发生了错误:{e}')
            return None

    def saveData(self,obj):
        """
        db.session.add_all([]) 可以一次性保存多条数据,但是这里用不到,因为涉及修改和新增一起的
        :param obj:
        :return:
        """
        # res = VipParse.query.filter(VipParse.url == obj['url']).first()
        res = VipParse.query.filter_by(url=obj['url']).first()
        if res:
            # res.update(obj)
            res.name = obj['name']
            res.state = obj['state']
            res.type = obj['type']
            res.order = obj['order']
            res.ext = obj['ext']
            res.header = obj['header']
            db.session.add(res)
        else:
            res = VipParse(**obj)
            db.session.add(res)
            db.session.flush()  # 获取id
        try:
            db.session.commit()
            return res.id
        except Exception as e:
            print(f'发生了错误:{e}')
            return None

    @staticmethod
    def getHideRules():
        res = VipParse.query.filter(VipParse.state == 0).all()
        return copy_utils.obj_to_list(res)