python 计算 对象 占用大小

# 这里主要是计算文件内容(str)的大小即: 统计空间占用情况, 并转换宜读单位 K,M

def gen_atta_size(con):  # 参数可以是任意数据类型
    if con:
        size_b = sys.getsizeof(con)
        size = str(size_b) + 'B'  # 不要小数位
        size_k = size_b / 1024
        if size_k > 1:
            size = '%.1f' % size_k + 'K'  # 一位小数
            size_m = size_k / 1024
            if size_m > 1:
                size = '%.2f' % size_m + 'M'  # 两位小数
    else:  # "", {}, [], 都是占用空间的, 这里忽略, 主要统计文本大小
        size = "0B"
    return size