#!/usr/bin/python
import os
import subprocess
import sys

PURPLE = '\033[95m'
CYAN = '\033[96m'
DARKCYAN = '\033[36m'
BLUE = '\033[94m'
GREEN = '\033[92m'
YELLOW = '\033[93m'
RED = '\033[91m'  
BOLD = '\033[1m'
UNDERLINE = '\033[4m'
END = '\033[0m'

def make_bold(string):
	return '{:s}{:s}{:s}'.format(BOLD, string, END)

def get_ipv4_addr(interface):
	cmd = 'ifconfig {:s}'.format(interface)
	for line in subprocess.check_output(cmd, shell=True).split('\n'):
		if line.strip().startswith('inet addr:'):
			return line.split()[1][5:]
	else:
		return '{:s}NO IP ASSIGNED{:s}'.format(RED, END)

def get_fs_usage(mount_point):
	cmd = 'df -h {:s}'.format(mount_point)
	try:
		line = subprocess.check_output(cmd, shell=True).split('\n')[1]
	except IndexError:
		return '{:s}MOUNT POINT NOT FOUND{:s}'.format(RED, END)
	line_tokens = line.split()
	output = '{:10s} ({:<9s}) - {:>4s} of {:>4s} [ {:s} ] used; '\
			 '{:>4s} available'.format(line_tokens[5], line_tokens[0], 
									 line_tokens[2], line_tokens[1],
									 line_tokens[4], line_tokens[3])
	return output


def main():
	""" This is a custom script I wrote to run on at login on phalanx
	and give me some system overview information.
	"""

	rows, columns = os.popen('stty size', 'r').read().split()
	rows = int(rows)
	columns = int(columns)

	hostname = subprocess.check_output('hostname', shell=True).strip()

	ip_addr = get_ipv4_addr('p1p1')
	root_stats = get_fs_usage('/dev/sda1')
	storage_stats = get_fs_usage('/dev/md0')

	print '       _           _                  '
	print '      | |         | |                 '
 	print ' _ __ | |__   __ _| | __ _ _ __ __  __'
	print "| '_ \| '_ \ / _` | |/ _` | '_ \\ \/ /"
	print '| |_) | | | | (_| | | (_| | | | |>  < '
	print "| .__/|_| |_|\__,_|_|\__,_|_| |_/_/\_\ "
	print '| |                                   '
	print '|_|                                   '
	print '-' * columns
	print '{:<16s} {:s}   {:s} {:s}'.format(make_bold('Hostname:'), hostname, 
										  make_bold('IP:'), ip_addr)
	print '{:<16s} {:s}'.format(make_bold('Storage:'), root_stats)
	print '{:<16s} {:s}'.format(make_bold(' '), storage_stats)


	print '-' * columns
	print subprocess.check_output('w -i', shell=True)
	print

if __name__ == '__main__':
	sys.exit(main())