请帮写一个“根据指定大小，自动填充文件字节”的代码

[feiyuncxd]

例如一个文件名为aaa.dat的文件，原大小为216byte，十六进制如下图


我想通过我指定的一个字节数，在文件尾部自动补足00
例如通过执行 python xxx.py aaa.dat 256 这个命令
就会自动将aaa.dat这个文件数据变为下图的样子


也就是在aaa.dat文件尾部，自动填充40个字节（256-216），全部是00。
如果指定的字节数（如示例命令里的256）小于或等于原文件大小，那么执行这个命令后，文件不变化。

请问这个代码应该怎么写，请各位帮忙！

[chuifengde]

import os
s = int(input(">>"))

file = r"c:\aaa.dat"
size = os.path.getsize(filename=file)
if s > size:
    with open(file, 'ab+') as f :
        g = b'\x00' * (s - size)
        f.write(g)