status.py

from db import Db
from mode import Mode

class Status(Mode):

	def validate(self, args):
		if len(args) != 2:
			raise ValueError('Usage: %s status' % args[0])

	def execute(self):
		db = Db()
		return db.get_doctype_counts().items()

	def output(self, results):
		bar = '=' * 40
		print('%s\nStatus:\n%s\n' % (bar, bar))

		if results:
			for doctype, count in results:
				print('%s: %s' % (doctype, count))
		else:
			print('No data')

		print('\n%s' % bar)