def save_creds(self, cred_mc: OAuth2Credentials, cred_mm: OAuth2Credentials): cred_mc_entity = datastore.Entity( self._KEY_CRED_MC, exclude_from_indexes=('credentials', )) cred_mc_entity.update({'credentials': cred_mc.to_json()}) cred_mm_entity = datastore.Entity( self._KEY_CRED_MM, exclude_from_indexes=('credentials', )) cred_mm_entity.update({'credentials': cred_mm.to_json()}) self._ds.put_multi([cred_mc_entity, cred_mm_entity])
async def _update_creds(self, creds: OAuth2Credentials): cred_json = creds.to_json() json_dict = json.loads(cred_json) # update cache of new Credentials self.photo_library_creds = json_dict # Save it to the database await self.config.photo_library_creds.set(json_dict)
def config(cleandir, drive_creds: OAuth2Credentials): with open(os.path.join(cleandir, "secrets.yaml"), "w") as f: f.write("for_unit_tests: \"password value\"\n") with open(os.path.join(cleandir, "credentials.dat"), "w") as f: f.write(drive_creds.to_json()) config = Config() config.override(Setting.DRIVE_URL, "http://localhost:1234") config.override(Setting.HASSIO_URL, "http://localhost:1234/") config.override(Setting.HOME_ASSISTANT_URL, "http://localhost:1234/homeassistant/api/") config.override(Setting.AUTHENTICATE_URL, "http://localhost:1234/external/drivecreds/") config.override(Setting.HASSIO_TOKEN, "test_header") config.override(Setting.SECRETS_FILE_PATH, "secrets.yaml") config.override(Setting.CREDENTIALS_FILE_PATH, "credentials.dat") config.override(Setting.FOLDER_FILE_PATH, "folder.dat") config.override(Setting.RETAINED_FILE_PATH, "retained.json") config.override(Setting.INGRESS_TOKEN_FILE_PATH, "ingress.dat") return config
def save_creds(self, cred_mc: OAuth2Credentials, cred_mm: OAuth2Credentials): with self._PATH_CRED_MC.open('w') as fmc: fmc.write(cred_mc.to_json()) with self._PATH_CRED_MM.open('w') as fmm: fmm.write(cred_mm.to_json())