fix new cloud verification process

This commit is contained in:
jneilliii 2024-12-07 00:17:57 -05:00
parent 8950778146
commit 383f0c3cb3
3 changed files with 35 additions and 22 deletions

View File

@ -130,9 +130,7 @@ class BambuPrintPlugin(
and "auth_token" in data and "auth_token" in data
): ):
self._logger.info(f"Registering user {data['email']}") self._logger.info(f"Registering user {data['email']}")
self._bambu_cloud = BambuCloud( self._bambu_cloud = BambuCloud(data["region"], data["email"], data["password"], data["auth_token"])
data["region"], data["email"], data["password"], data["auth_token"]
)
auth_response = self._bambu_cloud.login(data["region"], data["email"], data["password"]) auth_response = self._bambu_cloud.login(data["region"], data["email"], data["password"])
return flask.jsonify( return flask.jsonify(
{ {

View File

@ -192,12 +192,14 @@ class BambuCloud:
return ValueError(0) # FIXME return ValueError(0) # FIXME
elif loginType == 'verifyCode': elif loginType == 'verifyCode':
LOGGER.debug(f"Received verifyCode response") LOGGER.debug(f"Received verifyCode response")
raise EmailCodeRequiredError() # raise EmailCodeRequiredError()
return loginType
elif loginType == 'tfa': elif loginType == 'tfa':
# Store the tfaKey for later use # Store the tfaKey for later use
LOGGER.debug(f"Received tfa response") LOGGER.debug(f"Received tfa response")
self._tfaKey = auth_json.get("tfaKey") self._tfaKey = auth_json.get("tfaKey")
raise TfaCodeRequiredError() # raise TfaCodeRequiredError()
return loginType
else: else:
LOGGER.debug(f"Did not understand json. loginType = '{loginType}'") LOGGER.debug(f"Did not understand json. loginType = '{loginType}'")
LOGGER.error(f"Response not understood: '{response.text}'") LOGGER.error(f"Response not understood: '{response.text}'")
@ -358,18 +360,31 @@ class BambuCloud:
self._password = password self._password = password
result = self._get_authentication_token() result = self._get_authentication_token()
if result is None:
LOGGER.error("Unable to authenticate.")
return None
elif len(result) < 20:
return result
else:
self._auth_token = result self._auth_token = result
self._username = self._get_username_from_authentication_token() self._username = self._get_username_from_authentication_token()
return 'success'
# self._auth_token = result
# self._username = self._get_username_from_authentication_token()
def login_with_verification_code(self, code: str): def login_with_verification_code(self, code: str):
result = self._get_authentication_token_with_verification_code(code) result = self._get_authentication_token_with_verification_code(code)
self._auth_token = result self._auth_token = result
self._username = self._get_username_from_authentication_token() self._username = self._get_username_from_authentication_token()
if self._auth_token != "" and self._username != "" and self._auth_token != None and self._username != None:
return "success"
def login_with_2fa_code(self, code: str): def login_with_2fa_code(self, code: str):
result = self._get_authentication_token_with_2fa_code(code) result = self._get_authentication_token_with_2fa_code(code)
self._auth_token = result self._auth_token = result
self._username = self._get_username_from_authentication_token() self._username = self._get_username_from_authentication_token()
if self._auth_token != "" and self._username != "" and self._auth_token != None and self._username != None:
return "success"
def get_device_list(self) -> dict: def get_device_list(self) -> dict:
LOGGER.debug("Getting device list from Bambu Cloud") LOGGER.debug("Getting device list from Bambu Cloud")

View File

@ -14,7 +14,7 @@ plugin_package = "octoprint_bambu_printer"
plugin_name = "OctoPrint-BambuPrinter" plugin_name = "OctoPrint-BambuPrinter"
# The plugin's version. Can be overwritten within OctoPrint's internal data via __plugin_version__ in the plugin module # The plugin's version. Can be overwritten within OctoPrint's internal data via __plugin_version__ in the plugin module
plugin_version = "0.1.8rc13" plugin_version = "0.1.8rc14"
# The plugin's description. Can be overwritten within OctoPrint's internal data via __plugin_description__ in the plugin # The plugin's description. Can be overwritten within OctoPrint's internal data via __plugin_description__ in the plugin
# module # module