python穿透类 对象代理

python穿透类 对象代理

# coding:utf-8

import os
import logging

# 对象代理
# 用于解决需要在包调用与业务代码之间增加一些其他操作的场景
# 比如：应用程序中使用redis包对象，我们希望在redis包对象出错时尝试一次重连
# 又比如: 我们希望在调用包的某方法时纪录日志
# 使用此代理容器可以在不修改包及业务代码的情况下，实现以上功能

# 代理基类
# 该基类实现代理转发的基本功能，子类根据业务需要继承该类，并实现绑定方法
# 注意：为了避免透传的时候命名冲突，代理容器所有属性或者方法需要以abc_agent_开头



class AgentContainer(object):
    # 初始化类对象，将代理对象添加到属性中
    def __init__(self, current_object):
        self.abc_agent_current_object = current_object
        self.abc_agent_current_func = ""
        self.abc_agent_error = ""
        self.abc_agent_debug = True

    # 代理方法，用于代理对象方法的透传
    def abc_agent_container(self, *args, **kargs):
        result = None
        self.abc_agent_before_container(*args, **kargs)
        func = getattr(self.abc_agent_current_object, self.abc_agent_current_func)
        logging.error("func: %s" % func)
        try:
            result = func(*args, **kargs)
        except Exception, e:
            self.abc_agent_object_error(e)

        self.abc_agent_after_container(*args, **kargs)
        return result

    # 透传前的回调
    def abc_agent_before_container(self, *args, **kargs):
        self.abc_agent_log("before container")

    # 透传后的回调
    def abc_agent_after_container(self, *args, **kargs):
        self.abc_agent_log("after container")

    # 调用代理对象方法出错后触发
    def abc_agent_object_error(self, except_info):
        self.abc_agent_log("object error")
        self.abc_agent_error = except_info
        return except_info

    # 调用方法或者属性触发
    def __getattr__(self, func_name):
        logging.error("INTO __GETATTR__")
        self.abc_agent_log("container function is:",func_name)
        self.abc_agent_current_func = func_name
        func = getattr(self.abc_agent_current_object, func_name)
        if not callable(func):
            return func
        else:
            return self.abc_agent_container

    def __setattr__(self, attr_name, attr_value):
        logging.error("INTO __SETATTR__")
        pre_str = "abc_agent_"
        if attr_name.startswith(pre_str):
            object.__setattr__(self, attr_name, attr_value)
        else:
            self.abc_agent_before_container(attr_name, attr_value)
            setattr(self.abc_agent_current_object, attr_name, attr_value)
            self.abc_agent_after_container(attr_name, attr_value)

    def __get__(self, instance, owner):
        pass

    def abc_agent_log(self, *args):
        if self.abc_agent_debug:
            print "".join(args)


# 应用举例：
# 场景：使用代理容器代理redis，当调用redis出错时，重连redis

# 需要代理的类对象
class Redis(object):

    def hget(self, *args,**kargs):
        return "show is ok"

    def hset(self, *args):
        logging.error(args)
        #模拟redis hset方法报错
        raise IOError("ioerror")

    def incr(self, *args, **kwargs):
        logging.error("into incr")
        logging.error(args)
        logging.error(kwargs)
        return "success"

    def __init__(self):
        self.name = "polo"

# 基于代理基类派生出的redis高可用代理
# 当捕获到redis错误后对redis进行重连，完成后继续抛出错误
class HRedis(AgentContainer):
    def __init__(self):
        redis = Redis()
        AgentContainer.__init__(self, redis)
        # 是否打印调试信息，调试信息打印方法可以重载为其他日志输出方式
        self.abc_agent_debug = True

    # 重载error函数，可以通过判断redis报错是否为io类，对io类的报错进行处理：重新创建一个redis连接
    def abc_agent_object_error(self, except_info):
        self.abc_agent_current_object = Redis()
        # raise except_info
        return "error"


redis = HRedis()
# print redis.hget("ssssss")
logging.error('############')
redis.hset('a', 'b', 123)
logging.error("============")
logging.error(redis.incr(1, 2, 3, 4, arg1='arg'))