def setUp(self): mapper = ClassMapper([Dog, Cat, Person, JackRussell, Monkey, Monarch]) # self.db = FirestoreWrapper(mapper, creds_path) cred = credentials.Certificate(CREDS) app = firebase_admin.initialize_app(cred) uid = 'some-uid' additional_claims = { 'premiumAccount': True } custom_token = auth.create_custom_token(uid, additional_claims) firebase_admin.delete_app(app) self.db = FirestoreWrapper(mapper, None, project_id=CREDS["project_id"], custom_token=custom_token.decode("utf-8"), api_key=API_KEY, namespace=NAMESPACE )
def tearDownClass(cls): firebase_admin.delete_app(cls.db.app)
def tearDownClass(cls): if cls.db.app is not None: firebase_admin.delete_app(cls.db.app)
def tearDown(self): firebase_admin.delete_app(self.db.app)
def initialize_sdk_with_application_default(): # [START initialize_sdk_with_application_default] default_app = firebase_admin.initialize_app() # [END initialize_sdk_with_application_default] firebase_admin.delete_app(default_app)
def initialize_sdk_with_refresh_token(): # [START initialize_sdk_with_refresh_token] cred = credentials.RefreshToken('path/to/refreshToken.json') default_app = firebase_admin.initialize_app(cred) # [END initialize_sdk_with_refresh_token] firebase_admin.delete_app(default_app)
def test_fail_on_no_project_id(self): app = firebase_admin.initialize_app(testutils.MockCredential(), name='userMgt2') with pytest.raises(ValueError): auth._get_auth_service(app) firebase_admin.delete_app(app)
def user_mgt_app(): app = firebase_admin.initialize_app(testutils.MockCredential(), name='userMgt', options={'projectId': 'mock-project-id'}) yield app firebase_admin.delete_app(app)
def test_invalid_app_delete(self, app): with pytest.raises(ValueError): firebase_admin.delete_app(app)
def cleanup_apps(): with firebase_admin._apps_lock: apps = list(firebase_admin._apps.values()) for app in apps: firebase_admin.delete_app(app)
print('The stegosaurus is the shortest dino') # [END complex_query] service_account = 'path/to/serviceAccount.json' database_url = 'https://databaseName.firebaseio.com' cred = credentials.Certificate(service_account) firebase_admin.initialize_app(cred, {'databaseURL': database_url}) get_reference() set_value() set_child_value() update_child() update_children() overwrite_value() push_value() push_and_set_value() get_push_key() read_value() order_by_child() #order_by_nested_child() order_by_key() order_by_value() limit_query() range_query() complex_query() firebase_admin.delete_app(firebase_admin.get_app())
import lib.utils as utils import lib.services as firebase def get_admins(): with open(utils.dir.admins) as file: return { row[0]: row[1:] for row in csv.reader(file) } def set_claims(admins): for email, scopes in admins.items(): if not all(scope in firebase.SCOPES for scope in scopes): raise ValueError(f"Unrecognized scopes for {email}: {scopes}") utils.logger.verbose(f"Setting claims for {email}") if not scopes: utils.logger.warning(f"Removing admin priveleges for {email}") if utils.args.debug: utils.logger.debug(f" Claims for {email}", firebase.get_claims(email)) if utils.args.should_upload: firebase.set_scopes(email, scopes) if not utils.args.should_upload: utils.logger.warning("Did not upload admin claims. Use the --upload flag.") if __name__ == '__main__': utils.logger.info("Setting up admins...") admins = utils.logger.log_value("admins", get_admins) set_claims(admins) delete_app(firebase.app) utils.logger.info("Finished settig up admins")