import pprint

import re

import os

import lxml.html

import univention.management.console as umc

import univention.management.console.modules as umcm

from univention.management.console.modules import UMC_CommandError

import univention.config_registry

from univention.lib.i18n import Translation

import re

from univention.management.console.base import Base

from univention.management.console.protocol.definitions import *

from univention.management.console.config import ucr

from univention.management.console.modules import UMC_Error

_ = umc.Translation('univention-management-console-module-printers').translate

_ = Translation('univention-management-console-module-printers').translate

class Instance(umcm.Base):

class Instance(Base):

		if not os.path.exists('/usr/bin/pkusers'):

		try:

			raise UMC_CommandError(_('The print quota settings are currently disabled. Please install the package univention-printquota to enable them.'))

			from pykota.tool import PyKotaTool

			from pykota import reporter

			from pykota.storages.pgstorage import PGError

		except ImportError:

			raise UMC_Error(_('The print quota settings are currently disabled. Please install the package univention-printquota to enable them.'))

		(stdout, stderr, status) = self._shell_command(['/usr/bin/pkusers', '--list'], {'LANG':'C'})

		reportTool = PyKotaTool()

		users = []

		try:

		expr = re.compile('^\s*(.*?)\s+\-\s\<')

			reportTool.deferredInit()

		if status == 0:

			printers = reportTool.storage.getMatchingPrinters(printer)

			for line in stdout.split("\n"):

			reportingtool = reporter.openReporter(reportTool, 'html', printers, '*', 0)

				match = expr.match(line)

			status = reportingtool.generateReport()

				if match:

		except PGError as exc:

					users.append(match.group(1))

			MODULE.error('Cannot connect to postgres: %s' % (exc,))

			raise UMC_Error(_('The connection to the print quota postgres database failed. Please make sure the postgres service is running and reachable.'))

		finally:

			reportTool.regainPriv()

		result = []

		if status:

		for user in users:

			tree = lxml.html.fromstring(status)

			if not os.path.exists('/usr/bin/repykota'):

			table = tree.find_class('pykotatable')

				raise UMC_CommandError(_('The print quota settings are currently disabled. Please install the package univention-printquota to enable them.'))

			for i in table:

				for a in i.iterchildren(tag='tr'):

					data = list()

					for b in a.iterchildren(tag='td'):

						data.append(b.text_content().strip())

					if data and len(data) >= 11:

						user = data[0]

						#limitby = data[1]

						#overcharge = data[2]

						used = data[3]

						soft = data[4]

						hard = data[5]

						#balance = data[6]

						#grace = data[7]

						total = data[8]

						#paid = data[9]

						#warn = data[10]

						result.append(dict(

							user=user,

							used=used,

							soft=soft,

							hard=hard,

							total=total,

						))

		if status or len(stderr):

		if status or stderr:

		if printer in result:

		return result.get(printer, False)

			return result[printer]

		return False