import sys
import telnetlib


CRLF = "\r"+"\n"


class Hal:
    def __init__(self, env):
        self.winuser    = "svcqa"
        self.winpass    = "T35t123"
        self.haluser    = "markf"
        self.halpass    = "pre252r"
        self.host   = "waphal" + env
        self.tn     = telnetlib.Telnet(self.host)
        self.screen = ""

    def log_screen(self):
        logfile = open("logfile.txt", 'w')
        screen = self.tn.read_until("(*)", 2)
        logfile.write(screen)
        logfile.close()

    def connect(self):
        self.tn.read_until("login: ", 5)
        self.tn.write(self.winuser + "\n")
        if self.winuser:
            self.tn.read_until("Password: ", 5)
            self.tn.write(self.winpass + "\n")

        winerr = CRLF+"Login incorrect"
        winmsg = self.tn.read_until('Login incorrect', 2)

        if winerr == winmsg:
            self.tn.close()
            raise AssertionError('*ERROR* Windows Login Failed!')

        if self.tn.read_until("Connect", 2):
            self.tn.write("\n")

        self.tn.read_until("Enter your user id:", 2)
        self.tn.write(self.haluser + "\n")
        self.tn.read_until("user password:")
        self.tn.write(self.halpass + "\n")

        halerr = CRLF+"Invalid user or password"
        halmsg = self.tn.read_until('Invalid user or password', 2)

        if halerr == halmsg:
            self.tn.close()
            raise AssertionError('*ERROR* Hal Login Failed!')

        # Send extra carraige return for the "queue?" prompt
        self.tn.write("\n")

        print "Current Screen = %s" % (self.tn.read_until("(*)", 1))
        self.screen = str(self.tn.read_until("(*)", 1))

# New telnet session being used here
    def disconnect(self):

        self.tn.write("\n")

        stuff = self.tn.read_until("(*)", 5)
        print "Stuff on screen = %s" % (stuff)

        self.tn.write("off" + "\n")
# ... and here
    def navigate_to(self, menu, num):

        menu = str(menu)
        num = str(num)

        print "Current Screen = %s" % (self.screen)

        self.disconnect()