import exception

from page.page import Page

from page.user import UserPage, UserCreatePage, UserListPage

class RootPage(Page):

  def __init__(self):

    super().__init__()

    self.user = UserPage()

    self.user_create = UserCreatePage()

    self.users = UserListPage()

  def process(self, request, path, prevpath):

    answer = request.answer

    answer.template = self.commontemplate

    if path:

      if path[0] == 'user':

        return self.user.sub_process(request, path, prevpath)

      if path[0] == 'user_create':

        return self.user_create.sub_process(request, path, prevpath)

      if path[0] == 'users':

        return self.users.sub_process(request, path, prevpath)

      if path[0] == 'auth':

        answer.headers.append( ('WWW-Authenticate', 'Basic realm="Repo", charset="UTF-8"') )

        answer.status = '401 Unauthorized'

      else:

        raise exception.HttpNotFound()

    answer.content += '
' + answer.te('root page') + '
'

    answer.content += '
' + answer.te('Welcome!') + '
'

    if request.model.users.can_list():

      answer.content += '
' + self.make_link(answer, prevpath + ['users'], 'Users list') + '
\n'

    answer.content += '
Env: \n' + answer.e(str(request.env)) + '
'

    tables = list(v[0] for v in request.connection.query_list('SHOW TABLES'))

    answer.content += '
DB tables: ' + answer.e(', '.join(tables)) + '
'

    rows = request.connection.query_dict('SELECT * FROM %T', 'test')

    answer.content += '
Rows of test table: ' + answer.e(str(rows)) + '
'

    return answer.complete_content()