updated_code

from ucsmsdk.ucshandle import UcsHandle
class UCS(object):
    def __init__(self, ucsm_ip="", ucsm_pw="", ucsm_login=""):
        self.ucsm_ip = ucsm_ip
        self.ucsm_pw = ucsm_pw
        self.ucsm_login = ucsm_login
        self.handle = UcsHandle(self.ucsm_ip, self.ucsm_login, self.ucsm_pw)

    def login(self):
        login_handle = self.handle.login()
        return self.logged_in()


    def logout(self):
        log_out = login_handle.logout()
        return log_out

ucs_login = UCS(ucsm_ip = "X.X.X.X", ucsm_login = "ucspe", ucsm_pw = "ucspe")
ucs_login.login()
ucs_login.logout()