a8df578921
- funciton for calculating disk usuage - function for fake disk allocation - created an object model for df keyword to parse the various data info Change-Id: Id362e184d223470104dfc97bd3811df8172f9dbf Signed-off-by: Haadia Mufti <haadia.mufti@windriver.com>
83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
"""Df command output collection."""
|
|
|
|
from typing import List
|
|
|
|
from keywords.linux.df.df_object import DfObject
|
|
|
|
|
|
class DfOutput:
|
|
"""Collection of df command results."""
|
|
|
|
def __init__(self, raw_output: List[str]):
|
|
"""Initialize df output collection.
|
|
|
|
Args:
|
|
raw_output (List[str]): Raw output lines from df command.
|
|
"""
|
|
# Initialize empty collection to store parsed df entries
|
|
self.df_entries: List[DfObject] = []
|
|
self._parse_df_output(raw_output)
|
|
|
|
def _parse_df_output(self, raw_output: List[str]) -> None:
|
|
"""Parse df command output into DfObject instances.
|
|
|
|
Args:
|
|
raw_output (List[str]): Raw output lines from df command.
|
|
"""
|
|
# Skip header line (first line contains column names)
|
|
for line in raw_output[1:]:
|
|
if line.strip():
|
|
parts = line.strip().split()
|
|
# Ensure we have all required fields (filesystem, size, used, avail, use%, mounted)
|
|
if len(parts) >= 6:
|
|
filesystem = parts[0] # Device/filesystem name
|
|
total_kb = int(parts[1]) # Total space in KB
|
|
used_kb = int(parts[2]) # Used space in KB
|
|
available_kb = int(parts[3]) # Available space in KB
|
|
usage_percent = int(parts[4].rstrip("%")) # Usage percentage (remove % symbol)
|
|
mount_point = parts[5] # Mount point path
|
|
|
|
# Create df object and add to collection
|
|
df_obj = DfObject(filesystem, total_kb, used_kb, available_kb, usage_percent, mount_point)
|
|
self.df_entries.append(df_obj)
|
|
|
|
def get_df_entries(self) -> List[DfObject]:
|
|
"""Get all df entries.
|
|
|
|
Returns:
|
|
List[DfObject]: List of all df entries.
|
|
"""
|
|
return self.df_entries
|
|
|
|
def get_df_by_mount_point(self, mount_point: str) -> DfObject:
|
|
"""Get df entry by mount point.
|
|
|
|
Args:
|
|
mount_point (str): Mount point path.
|
|
|
|
Returns:
|
|
DfObject: Df entry for the specified mount point.
|
|
|
|
Raises:
|
|
ValueError: If mount point not found.
|
|
"""
|
|
# Search through all entries for matching mount point
|
|
for entry in self.df_entries:
|
|
if entry.get_mount_point() == mount_point:
|
|
return entry
|
|
raise ValueError(f"Mount point '{mount_point}' not found")
|
|
|
|
def get_first_entry(self) -> DfObject:
|
|
"""Get first df entry.
|
|
|
|
Returns:
|
|
DfObject: First df entry.
|
|
|
|
Raises:
|
|
ValueError: If no entries available.
|
|
"""
|
|
# Return first entry (useful for single filesystem queries)
|
|
if not self.df_entries:
|
|
raise ValueError("No df entries available")
|
|
return self.df_entries[0]
|