python生成固定大小随机内容文件

最新推荐文章于 2025-06-03 09:06:33 发布

原创最新推荐文章于 2025-06-03 09:06:33 发布 · 3.4k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #工具

python 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个使用Python编写的简单脚本，该脚本能够生成指定大小的随机内容文件，支持从字节(B)到太字节(TB)级别的文件大小。

python生成固定大小随机内容文件

用法 gen.py size unit e.g. 100 m

支持从B到TB大小

#!/bin/env python
import os,sys
def createfile(filename,size):
    with open(filename,'wb') as f:
        f.write(os.urandom(size))

if len(sys.argv)<3:
    print "gen.py size unit e.g. 100 m"
    sys.exit(1)
size = int(sys.argv[1])
unit = sys.argv[2][0].lower()
sm = {'b':1,'k':1024,'m':1024*1024,'g':1024*1024*1024,'t':1024*1024*1024*1024}
createfile("data/test"+sys.argv[1]+unit,size*sm[unit])
print "gen test"+sys.argv[1]+unit,"done"

原创转载请先告知我