#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author: <EMAIL>
"""

"""

import yaml
import os

from enum import Enum
from wush.common import constants
from wush.common import utils
from wush.common.files import FileUtils
from wush.common.functions import load_module
from wush.common.functions import super_function
from wush.common.functions import Function
from wush.common.functions import FunctionFactory
from wush.common.loggers import create_logger
from wush.common.exceptions import RequestException
from wush.models import ModuleModel
from wush.models import RequestModel

class Env():
    # 参数地址
    body_path = ''

    def __init__(self, **kw):
        self.add(**kw)

    def add(self, **kw):
        """添加环境变量"""
        for k, v in kw.items():
            setattr(self, k, v)
            if k.isupper():
                os.environ[k] = v

    def dict(self):
        data = self.__dict__
        return data

class Config():
    logger = create_logger('Config')

    _root = None
    env_root = ''
    body_root = ''
    module_root = ''
    data_root = ''
    #  response_root = ''
    #  request_root = ''
    #  version_root = ''
    space_name = ''
    function_modules = []
    modules = []
    _config = {}

    function  = None
    _env = Env()

    def __init__(self, ):
        """初始化"""
        self._root = self.get_default_root()
        default_config = self.get_default_config()
        for k, v in default_config.items():
            self._setattr(k, v)

    def _setattr(self, k, v):
        """设置属性"""
        if k == 'env':
            k = '_env'
            v = Env(**v)
        setattr(self, k, v)

    @classmethod
    def load(cls, fileroot):
        """
        加载配置
        :param fileroot: 如果是地址则获取内容，如果是数据则直接加载
        """
        cls.logger.info('Config load path: %s', fileroot)
        item = cls()
        data = {}
        if isinstance(fileroot, dict):
            data = dict(fileroot)
        else:
            item._root = fileroot
            filepath = os.path.join(fileroot, 'wapi.yml')
            with open(filepath, 'r') as f:
                data = yaml.safe_load(f)
        item._config = data

        if data:
            for k, v in data.items():
                item._setattr(k, v)

        # 格式化各个 root 配置
        for _root in ('env_root', 'module_root', 'body_root', 'data_root'):
            item._setattr(_root, item.fmt_path(getattr(item, _root)))

        item._load_functions()

        # 创建保持地址
        #  for _root in ('response_root', 'request_root', 'version_root'):
            #  setattr(item, _root, os.path.join(item.data_root,
                #  _root.replace('_root', '')))
            #  if not os.path.exists(getattr(item, _root)):
                #  os.makedirs(getattr(item, _root))

        return item

    @property
    def env(self):
        """获取 env 信息"""
        return self._env

    def _load_functions(self):
        """加载方法"""
        self.logger.info('Config functions %s', self.function_modules)
        if not self.function_modules:
            self.function = super_function
            return
        for module_name in self.function_modules:
            load_module(module_name)

        functions = FunctionFactory.get_factory()
        self.logger.info('load functions %s', functions)

        f = Function(functions)
        self.function = f

    def get_module(self, module_name):
        """获取模块"""
        _config = {}
        if self.modules:
            for m in self.modules:
                if m.get("module") == module_name:
                    _config = m
        elif self.module_root:
            self.logger.info('Module: %s', module_name)
            _path = self.get_module_path(module_name)
            self.logger.info('Module path: %s', _path)
            if not os.path.exists(_path):
                raise RequestException('can not found request config {}'.format(
                    _path))

            _config = FileUtils.read_dict(_path)
        # 获取 env 信息
        _config['functions'] = self.function.get_functions()

        # 父 env
        parent_env = { "env": self.env.dict() }
        _config = ModuleModel._merge_config(parent_env, _config)

        # 获取父配置
        parent = _config.get("parent")
        if parent:
            parent_path = self.get_module_path(parent)
            parent_config = FileUtils.read_dict(parent_path) or {}
            ModuleModel._merge_config(parent_config, _config)
            _config = parent_config
        module = ModuleModel.load(_config)
        return module

    def set_space_name(self, space_name):
        self.space_name = space_name
        self._reload_env()

    def _reload_env(self):
        """重新加载 env"""
        env_path = self.get_env_path(self.space_name)
        self.logger.info('env_path %s', env_path)
        if os.path.exists(env_path):
            space_env = FileUtils.read_dict(env_path) or {}
            self.env.add(**space_env)

    def get_env_path(self, space_name=None):
        """获取 env 配置地址"""
        if not space_name:
            space_name = self.space_name
        if not space_name:
            space_name = super_function.get_current_space_name()
        path = os.path.join(self.env_root, '{}.yml'.format(space_name))
        self.logger.info('Env path %s', path)
        return path

    def get_module_path(self, module_name=None):
        """获取 request 配置地址"""
        if not module_name:
            module_name = self.get_default_module_name()
        return os.path.join(self.module_root, '{}.yml'.format(module_name))

    def get_body_path(self, body_name):
        """获取 body 配置地址"""
        return os.path.join(self.body_root, body_name)

    def get_modules(self):
        """获取模块名称列表"""
        return list(filter(lambda x: not x.startswith('.'), [
            o.replace('.yml', '') for o in os.listdir(self.module_root)]))

    def get_requests(self, module_name):
        """获取请求名称列表"""
        path = self.get_module_path(module_name)
        data = {}
        with open(path, 'r') as f:
            data = yaml.safe_load(f)

        return data.get("requests") or []

    def get_function(self):
        return self.function

    def dict(self):
        return self._config

    @classmethod
    def get_default_root(cls):
        return constants.CONFIG_ROOT

    def fmt_path(self, path):
        if not path:
            return path
        path = os.path.expanduser(path)
        if os.path.isabs(path):
            return path
        return os.path.join(self._root, path)

    @classmethod
    def get_current_body_name(cls, space_name, module_name, request_name):
        """获取当前 body 文件名称"""
        body_name = '{space}_{module}_{request}'.format(
            space = space_name, module=module_name, request = request_name)
        return body_name

    @classmethod
    def get_current_env_name(self, space_name):
        """获取当前 env 文件名称"""
        return space_name

    @classmethod
    def get_default_module_name(cls):
        """获取默认 module 名称"""
        return constants.DEFAULT_MODULE_NAME

    @classmethod
    def get_default_config(cls):
        """获取默认配置"""
        return constants.DEFAULT_CONFIG

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author: <EMAIL>
"""

"""

import yaml
import os

from enum import Enum
from wush.common import constants
from wush.common import utils
from wush.common.files import FileUtils
from wush.common.functions import load_module
from wush.common.functions import super_function
from wush.common.functions import Function
from wush.common.functions import FunctionFactory
from wush.common.loggers import create_logger
from wush.common.exceptions import RequestException
from wush.models import ModuleModel
from wush.models import RequestModel

class Env():
    # 参数地址
    body_path = ''

    def __init__(self, **kw):
        self.add(**kw)

    def add(self, **kw):
        """添加环境变量"""
        for k, v in kw.items():
            setattr(self, k, v)
            if k.isupper():
                os.environ[k] = v

    def dict(self):
        data = self.__dict__
        return data

class Config():
    logger = create_logger('Config')

    _root = None
    env_root = ''
    body_root = ''
    module_root = ''
    data_root = ''
    #  response_root = ''
    #  request_root = ''
    #  version_root = ''
    space_name = ''
    function_modules = []
    modules = []
    _config = {}

    function  = None
    _env = Env()

    def __init__(self, ):
        """初始化"""
        self._root = self.get_default_root()
        default_config = self.get_default_config()
        for k, v in default_config.items():
            self._setattr(k, v)

    def _setattr(self, k, v):
        """设置属性"""
        if k == 'env':
            k = '_env'
            v = Env(**v)
        setattr(self, k, v)

    @classmethod
    def load(cls, fileroot):
        """
        加载配置
        :param fileroot: 如果是地址则获取内容，如果是数据则直接加载
        """
        cls.logger.info('Config load path: %s', fileroot)
        item = cls()
        data = {}
        if isinstance(fileroot, dict):
            data = dict(fileroot)
        else:
            item._root = fileroot
            filepath = os.path.join(fileroot, 'wapi.yml')
            with open(filepath, 'r') as f:
                data = yaml.safe_load(f)
        item._config = data

        if data:
            for k, v in data.items():
                item._setattr(k, v)

        # 格式化各个 root 配置
        for _root in ('env_root', 'module_root', 'body_root', 'data_root'):
            item._setattr(_root, item.fmt_path(getattr(item, _root)))

        item._load_functions()

        # 创建保持地址
        #  for _root in ('response_root', 'request_root', 'version_root'):
            #  setattr(item, _root, os.path.join(item.data_root,
                #  _root.replace('_root', '')))
            #  if not os.path.exists(getattr(item, _root)):
                #  os.makedirs(getattr(item, _root))

        return item

    @property
    def env(self):
        """获取 env 信息"""
        return self._env

    def _load_functions(self):
        """加载方法"""
        self.logger.info('Config functions %s', self.function_modules)
        if not self.function_modules:
            self.function = super_function
            return
        for module_name in self.function_modules:
            load_module(module_name)

        functions = FunctionFactory.get_factory()
        self.logger.info('load functions %s', functions)

        f = Function(functions)
        self.function = f

    def get_module(self, module_name):
        """获取模块"""
        _config = {}
        if self.modules:
            for m in self.modules:
                if m.get("module") == module_name:
                    _config = m
        elif self.module_root:
            self.logger.info('Module: %s', module_name)
            _path = self.get_module_path(module_name)
            self.logger.info('Module path: %s', _path)
            if not os.path.exists(_path):
                raise RequestException('can not found request config {}'.format(
                    _path))

            _config = FileUtils.read_dict(_path)
        # 获取 env 信息
        _config['functions'] = self.function.get_functions()

        # 父 env
        parent_env = { "env": self.env.dict() }
        _config = ModuleModel._merge_config(parent_env, _config)

        # 获取父配置
        parent = _config.get("parent")
        if parent:
            parent_path = self.get_module_path(parent)
            parent_config = FileUtils.read_dict(parent_path) or {}
            ModuleModel._merge_config(parent_config, _config)
            _config = parent_config
        module = ModuleModel.load(_config)
        return module

    def set_space_name(self, space_name):
        self.space_name = space_name
        self._reload_env()

    def _reload_env(self):
        """重新加载 env"""
        env_path = self.get_env_path(self.space_name)
        self.logger.info('env_path %s', env_path)
        if os.path.exists(env_path):
            space_env = FileUtils.read_dict(env_path) or {}
            self.env.add(**space_env)

    def get_env_path(self, space_name=None):
        """获取 env 配置地址"""
        if not space_name:
            space_name = self.space_name
        if not space_name:
            space_name = super_function.get_current_space_name()
        path = os.path.join(self.env_root, '{}.yml'.format(space_name))
        self.logger.info('Env path %s', path)
        return path

    def get_module_path(self, module_name=None):
        """获取 request 配置地址"""
        if not module_name:
            module_name = self.get_default_module_name()
        return os.path.join(self.module_root, '{}.yml'.format(module_name))

    def get_body_path(self, body_name):
        """获取 body 配置地址"""
        return os.path.join(self.body_root, body_name)

    def get_modules(self):
        """获取模块名称列表"""
        return list(filter(lambda x: not x.startswith('.'), [
            o.replace('.yml', '') for o in os.listdir(self.module_root)]))

    def get_requests(self, module_name):
        """获取请求名称列表"""
        path = self.get_module_path(module_name)
        data = {}
        with open(path, 'r') as f:
            data = yaml.safe_load(f)

        return data.get("requests") or []

    def get_function(self):
        return self.function

    def dict(self):
        return self._config

    @classmethod
    def get_default_root(cls):
        return constants.CONFIG_ROOT

    def fmt_path(self, path):
        if not path:
            return path
        path = os.path.expanduser(path)
        if os.path.isabs(path):
            return path
        return os.path.join(self._root, path)

    @classmethod
    def get_current_body_name(cls, space_name, module_name, request_name):
        """获取当前 body 文件名称"""
        body_name = '{space}_{module}_{request}'.format(
            space = space_name, module=module_name, request = request_name)
        return body_name

    @classmethod
    def get_current_env_name(self, space_name):
        """获取当前 env 文件名称"""
        return space_name

    @classmethod
    def get_default_module_name(cls):
        """获取默认 module 名称"""
        return constants.DEFAULT_MODULE_NAME

    @classmethod
    def get_default_config(cls):
        """获取默认配置"""
        return constants.DEFAULT_CONFIG

#!/usr/bin/env python # -*- coding:utf-8 -*- # Author: <EMAIL> # 参数地址 添加环境变量 #  response_root = '' #  request_root = '' #  version_root = '' 初始化 设置属性 加载配置         :param fileroot: 如果是地址则获取内容，如果是数据则直接加载 # 格式化各个 root 配置 # 创建保持地址 #  for _root in ('response_root', 'request_root', 'version_root'): #  setattr(item, _root, os.path.join(item.data_root, #  _root.replace('_root', ''))) #  if not os.path.exists(getattr(item, _root)): #  os.makedirs(getattr(item, _root)) 获取 env 信息 加载方法 获取模块 # 获取 env 信息 # 父 env # 获取父配置 重新加载 env 获取 env 配置地址 获取 request 配置地址 获取 body 配置地址 获取模块名称列表 获取请求名称列表 获取当前 body 文件名称 获取当前 env 文件名称 获取默认 module 名称 获取默认配置