#!/usr/bin/python

import commands
import os
import string
import sys

script_name = sys.argv[0]

output = commands.getoutput('/sbin/fdisk -l')

if "sda" in output: 
	scsi = "yes"
else:
	scsi = "no"
	
if "sdb" in output: 
	scsi2 = "yes"
else:
	scsi2 = "no"

if "hda" in output: 
	ide = "yes"
else:
	ide = "no"

if "hdb" in output: 
	ide2 = "yes"
else:
	ide2 = "no"	

if scsi == "yes":
	drive1 = "/dev/sda"
	drive1_size = commands.getoutput('fdisk -l /dev/sda|cut -d " " -f 5|head -2').strip()
if scsi2 == "yes":
	drive2 = "/dev/sdb"
elif ide == "yes":
	drive2 = "/dev/hda"

if ide == "yes":
	drive1 = "/dev/hda"
	drive1_size = commands.getoutput('fdisk -l /dev/hda|cut -d " " -f 5|head -2').strip()
if ide2 == "yes":
	drive2 = "/dev/hdb"

drive_size_m = int(drive1_size) * 1024	

boot_size = 100;
var_size = int(drive_size_m)/15
usr_size = int(drive_size_m)/15
root_size = 1024
tmp_size = 1024

f = open('PARTITIONS', 'w')

f.write("""# Drive partitioning information determined from %s
clearpart --all
part /boot --fstype ext3 --size=%s --ondisk=%s --asprimary
part /var --fstype ext3 --size=%s --ondisk=%s
part /usr --fstype ext3 --size=%s --ondisk=%s --asprimary
part / --fstype ext3 --size=%s --ondisk=%s
part swap --recommended --ondisk=%s --asprimary
part /tmp --fstype ext3 --size=%s --ondisk=%s
part /home --fstype ext3 --size=1 --grow --ondisk=%s

""" %(script_name, boot_size, drive1, var_size, drive1, usr_size, drive1, root_size, drive1, drive1, tmp_size, drive1, drive1) )

f.write('part /backup --fstype ext3 --size=1 --grow --ondisk %s' % drive2)

f.close()