Python小工具之消耗系统指定大小内存的方法

#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import re
import time
 
def print_help():
  print 'Usage: '
  print ' python mem.py 100MB'
  print ' python mem.py 1GB'
 
if __name__ == "__main__":
  if len(sys.argv) == 2: #用来判断运行.py文件后面有没有跟 100M，2的话就是跟了一个参数，因为该程序本身占一个参数，所以  *.py  100M 一共两个参数
    pattern = re.compile('^(\d*)([M|G]B)$') #正则
    match = pattern.match(sys.argv[1].upper()) # 匹配正则，取出参数 100M
    if match:  #如果有带参数
      num = int(match.group(1))   #取出100 值
      unit = match.group(2)  #取出但位 M 或者G
      if unit == 'MB': #判断输入的单位是MB 还是G
        s = ' ' * (num * 1024 * 1024) #空格乘以一个数赋值给变量，来达到扩大内存？
      else:
        s = ' ' * (num * 1024 * 1024 * 1024)
 
      time.sleep(10000)#内存保留的时间？
    else:
      print_help()
  else:
    print_help()

使用方法如下：

python mem.py 100M
python mem.py 1G