清空分区和创建分区脚本

Magicloud Script

清空分区

脚本名称

clean_disks.py

使用说明

• python clean_disks.py [disk_name1, disk_name2 ...]
• 支持单磁盘和多磁盘清空
• 支持脚本参数，如果没有参数，默认清空所有磁盘分区，加磁盘参数则可清空指定磁盘分区, 多磁盘则累加多位磁盘参数
• 为了保证安全，脚本不会清空系统磁盘，脚本会做过滤工作

使用范例

• 清空所有磁盘分区

python clean_disks.py

• 清空sdb的分区

python clean_disks.py sdb

• 清空sdb和sdc的分区

python clean_disks.py sdb sdc

脚本代码

import sys
from mcs3.command import do_cmd
def clean_disk(disk_name):
    print 'disk: {} clean start ...'.format(disk_name)
    do_cmd('sgdisk -Zog /dev/{}'.format(disk_name))
    print 'disk: {} clean done ...'.format(disk_name)
mount_info = do_cmd('mount')
sys_disk_name = mount_info[5:8]
if len(sys.argv) > 1:
    disks = sys.argv[1:]
    for disk_name in disks:
        if disk_name == sys_disk_name:
            print '{} is system disk, can\'t be clean!'.format(disk_name)
        else:
            clean_disk(disk_name)
else:
    all_disks = do_cmd('lsblk').splitlines()
    for disk in all_disks:
        if (disk.startswith('sd')):
            disk_name = disk.split()[0]
            if disk_name != sys_disk_name:
                clean_disk(disk_name)

创建分区

脚本名称

create_partitions.py

使用说明

• python create_partitions.py size num [disk_name1, disk_name2 ...], size为分区大小，num为分区数
• 支持单磁盘和多磁盘创建分区
• 支持脚本参数，如果没有参数，默认为所有磁盘创建分区，加磁盘参数则可清空指定磁盘分区, 多磁盘则累加多位磁盘参数
• 为了保证安全，脚本不会为系统磁盘创建磁盘分区，脚本会做过滤工作

使用范例

• 为sdb和sdc分别都创建两个大小为100G的分区

python create_partitions.py 100G 2 sdb sdc 

• 为所有磁盘创建三个大小为10G的分区

python create_partitions.py 10G 3

• 为sdc创建四个大小为1T的分区

python create_partitions.py 1024G 4 sdc

脚本代码

import sys
from mcs3.command import do_cmd
size = sys.argv[1]
num = sys.argv[2]
mount_info = do_cmd('mount')
sys_disk_name = mount_info[5:8]
def parted_disks(num, size, disk):
    for i in range(int(num)):
        do_cmd('sgdisk -n {}:0:+{} /dev/{}'.format(i+1, size, disk))
        print 'disk {} partition {} done ...'.format(disk, i+1)
if len(sys.argv) > 3:
    disks = sys.argv[3:]
    for disk in disks:
        if disk == sys_disk_name:
            print '{} is system disk, can\'t be parted!'.format(disk)
        else:
            parted_disks(num, size, disk)
else:
    all_disks = do_cmd('lsblk').splitlines()
    for disk in all_disks:
        if (disk.startswith('sd')):
            disk_name = disk.split()[0]
            if disk_name != sys_disk_name:
                parted_disks(num, size, disk)