如何创建一个类似字符串的类？

2019-07-22 00:59:16 阅读：219 来源： 互联网

我有一个上下文管理器,它将输出捕获到一个字符串,用于在with语句下缩进的代码块.此上下文管理器生成一个自定义结果对象,当块完成执行时,该对象将包含捕获的输出.

from contextlib import contextmanager

@contextmanager
def capturing():
    "Captures output within a 'with' block."
    from cStringIO import StringIO

    class result(object):
        def __init__(self):
            self._result = None
        def __str__(self):
            return self._result

    try:
        stringio = StringIO()
        out, err, sys.stdout, sys.stderr = sys.stdout, sys.stderr, stringio, stringio
        output = result()
        yield output
    finally:
        output._result, sys.stdout, sys.stderr = stringio.getvalue(), out, err
        stringio.close()

with capturing() as text:
    print "foo bar baz",

print str(text)   # prints "foo bar baz"

当然,我不能只返回一个字符串,因为字符串是不可变的,因此用户从with语句返回的字符串在代码块运行后无法更改.但是,在使用str之后显式地将结果对象转换为字符串是一种拖累(我还使用了将对象调用为一点语法糖).

那么是否可以使结果实例像字符串一样,因为它确实在命名时返回一个字符串？我尝试实现__get__,但似乎只适用于属性.或者我想要的是不可能的？

解决方法:

如何创建一个类似字符串的类？
子类str

import os
class LikeAStr(str):
    '''Making a class like a str object; or more precisely
    making a str subclass with added contextmanager functionality.'''

    def __init__(self, diff_directory):
        self._iwd = os.getcwd()
        self._cwd = diff_directory

    def __enter__(self):
        return self

    def __exit__(self, ext_typ, exc_value, traceback):
        try: os.chdir(self._iwd) # might get deleted within the "with" statement
        except: pass

    def __str__(self):
        return self._cwd

    def __repr__(self):
        return repr(self._cwd)


astr = LikeAStr('C:\\')

with LikeAStr('C:\\') as astr:
    print 1, os.getcwd()
    os.chdir( astr ) # expects str() or unicode() not some other class
    print 2, os.getcwd()
    #

# out of with block
print 3, os.getcwd()
print 4, astr == 'C:\\'

输出：

1 D:\Projects\Python\
2 C:\
3 D:\Projects\Python\
4 True

标签：python,string,with-statement
来源： https://codeday.me/bug/20190721/1497351.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

如何创建一个类似字符串的类？