count_drives.py

#!/usr/bin/env python3

import os
import re
import sys

from ridlib import *

"""

Determine if any RIDS have bounced, blipped, or otherwise vanished.

It does this by looking at three quantities:

* Num_NS  = number of rids the depot is supposed to have according to /etc/number_of_storage_drives
* Num_IBP = number of rids visible to ibp_server
* Num_BLK = number of rids according to /sys/block

Complain if there is a discrepancy.

"""

Num_NS = int(SysExec("cat /etc/number_of_storage_drives"))

# Get a list of RIDs attached to the IBP Server daemon
IBP_Rids = []
Count_Pending = 0

for line in SysExec("get_version -a").splitlines():

	if re.search("^RID: ", line):
		IBP_Rids.append(int(line.split(" ")[1]))

	if re.search("Pending RID list", line):

#		print("DEBUG:  line = " + str(line))

		Rids = line.split(":")[1].strip().split(" ")

#		print("DEBUG:  Rids = " + str(Rids))

		if Rids[0] == "0":
			continue

		for Rid in Rids:
			IBP_Rids.append(int(Rid))
			Count_Pending += 1

#print("DEBUG:  IBP_Rids = " + str(IBP_Rids))

IBP_Rids.sort()
Num_IBP = len(IBP_Rids)

# Get a list of RIDs visible to the OS via /sys/block
OS_Rids = []
LS = os.listdir("/dev/disk/by-label/")
for disk in LS:

	if not re.search("rid-data", disk):
		continue

	OS_Rids.append(int(disk.split("-")[2]))
OS_Rids.sort()
Num_BLK = len(OS_Rids)

#print("DEBUG:  OS_Rids = " + str(OS_Rids))

# See if there descrepancies between OS and IBP Server
Diff_Rids = list(set(OS_Rids) - set(IBP_Rids))

#print("DEBUG:  Diff_Rids = " + str(Diff_Rids))

# If so, see if they are sequestered
Num_Seq = 0
if Diff_Rids:
	for Rid in list(Diff_Rids):
#		print("DEBUG:  Checking Rid " + str(Rid))
		Status = RID_Check_Sequester(str(Rid))
#		print("DEBUG:  Rid " + str(Rid) + " status = " + str(Status))
		if Status.startswith("SEQUESTERED"):
			Num_Seq += 1
			Diff_Rids.remove(Rid)
#			print("DEBUG:  Removing " + str(Rid) + " from Diff_Rids, leaving " + str(Diff_Rids))

#print("DEBUG:  Diff_Rids = " + str(Diff_Rids))

#print("DEBUG:  Num_NS = " + str(Num_NS) + " and Num_IBP + Num_Seq = " + str(Num_IBP + Num_Seq) + " and Num_BLK = " + str(Num_BLK))

if Num_NS == (Num_IBP + Num_Seq) and Num_NS == Num_BLK:

	if Num_Seq == 0:

		print("All Drives = " + str(Num_NS))
	else:
		print("All Drives = " + str(Num_NS) + " (including " + str(Num_Seq) + " sequestered drives)")
else:
	print("GV = " + str(Num_IBP) + " and NS = " + str(Num_NS) + " and SN = " + str(Num_BLK) + " and SEQUESTERED = " + str(Num_Seq) + " and PENDING = " + str(Count_Pending))
	if Diff_Rids:
		print("DIFF_RID = " + str(Diff_Rids))