mirror of
https://github.com/corpnewt/gibMacOS.git
synced 2024-11-12 10:04:11 +01:00
96 lines
3.7 KiB
Python
96 lines
3.7 KiB
Python
import subprocess, plistlib, sys, os, time, json
|
|
sys.path.append(os.path.abspath(os.path.dirname(os.path.realpath(__file__))))
|
|
import run
|
|
|
|
class Disk:
|
|
|
|
def __init__(self):
|
|
self.r = run.Run()
|
|
self._update_disks()
|
|
|
|
def update(self):
|
|
self._update_disks()
|
|
|
|
def _update_disks(self):
|
|
self.disks = self.get_disks()
|
|
|
|
def get_disks(self):
|
|
# We hate windows... all of us.
|
|
#
|
|
# This has to be done in 3 commands,
|
|
# 1. To get the PHYSICALDISK entries, index, and model
|
|
# 2. To get the drive letter, volume name, fs, and size
|
|
# 3. To get some connection between them...
|
|
#
|
|
# May you all forgive me...
|
|
|
|
disks = self.r.run({"args":["wmic", "diskdrive", "get", "deviceid,model,index,size,partitions"]})[0]
|
|
disks = disks.replace("\r","").split("\n")[1:]
|
|
p_disks = {}
|
|
for d in disks:
|
|
ds = [x for x in d.split(" ") if len(x)]
|
|
if len(ds) < 5:
|
|
continue
|
|
p_disks[ds[1]] = {
|
|
"index":int(ds[1]),
|
|
"device":ds[0],
|
|
"model":" ".join(ds[2:-2]),
|
|
"size":int(ds[-1]),
|
|
"partitioncount":int(ds[-2]),
|
|
"type":0 # 0 = Unknown, 1 = No Root Dir, 2 = Removable, 3 = Local, 4 = Network, 5 = Disc, 6 = RAM disk
|
|
}
|
|
if not len(p_disks):
|
|
# Drat, nothing
|
|
return p_disks
|
|
# Let's find a shitty way to map this biz now
|
|
shit = self.r.run({"args":["wmic", "path", "Win32_LogicalDiskToPartition", "get", "antecedent,dependent"]})[0]
|
|
shit = shit.replace("\r","").split("\n")[1:]
|
|
for s in shit:
|
|
s = s.lower()
|
|
d = p = mp = None
|
|
try:
|
|
dp = s.split("deviceid=")[1].split('"')[1]
|
|
d = dp.split("disk #")[1].split(",")[0]
|
|
p = dp.split("partition #")[1]
|
|
mp = s.split("deviceid=")[2].split('"')[1].upper()
|
|
except:
|
|
pass
|
|
if any([d, p, mp]):
|
|
# Got *something*
|
|
if p_disks.get(d,None):
|
|
if not p_disks[d].get("partitions",None):
|
|
p_disks[d]["partitions"] = {}
|
|
p_disks[d]["partitions"][p] = {"letter":mp}
|
|
# Last attempt to do this - let's get the partition names!
|
|
parts = self.r.run({"args":["wmic", "logicaldisk", "get", "deviceid,filesystem,volumename,size,drivetype"]})[0]
|
|
parts = parts.replace("\r","").split("\n")[1:]
|
|
for p in parts:
|
|
ps = [x for x in p.split(" ") if len(x)]
|
|
if len(ps) < 2:
|
|
# Need the drive letter and disk type at minimum
|
|
continue
|
|
# Organize!
|
|
plt = ps[0] # get letter
|
|
ptp = ps[1] # get disk type
|
|
# Initialize
|
|
pfs = pnm = None
|
|
psz = -1 # Set to -1 initially for indeterminate size
|
|
try:
|
|
pfs = ps[2] # get file system
|
|
psz = ps[3] # get size
|
|
pnm = " ".join(ps[4:]) # get the rest in the name
|
|
except:
|
|
pass
|
|
for d in p_disks:
|
|
p_dict = p_disks[d]
|
|
for pr in p_dict.get("partitions",{}):
|
|
pr = p_dict["partitions"][pr]
|
|
if pr.get("letter","").upper() == plt.upper():
|
|
# Found it - set all attributes
|
|
pr["size"] = int(psz)
|
|
pr["file system"] = pfs
|
|
pr["name"] = pnm
|
|
# Also need to set the parent drive's type
|
|
p_dict["type"] = int(ptp)
|
|
break
|
|
return p_disks |