summaryrefslogtreecommitdiffstats
path: root/command-stubs/list-harddrives-stub
blob: 73f86c2e650de8f7851c793df77d4330146dcba9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#!/usr/bin/python
#
# scan system for harddrives and output device name/size
#
# Copyright (C) 2007  Red Hat, Inc.  All rights reserved.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

import os
import sys

# for testing
if (os.path.exists('isys')):
    sys.path.append('isys')

sys.path.append('/usr/lib/anaconda')

import anaconda_log
import parted
import partedUtils
import isys

drives = isys.hardDriveDict()

driveList = drives.keys()
driveList.sort()
for drive in driveList:

    if not isys.mediaPresent(drive):
	continue

    # try to open and get size
    skip = 0
    deviceFile =  "/tmp/%s" % (drive,)
    isys.makeDevInode(drive, deviceFile)
    try:
	dev = parted.PedDevice.get(deviceFile)
    except:
	skip = 1
    os.remove(deviceFile)

    if skip:
	continue

    sizeMB = (float(dev.heads * dev.cylinders * dev.sectors) / (1024 * 1024)
		            * dev.sector_size)
	
    print drive, sizeMB