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 tearDownClass(cls):
firebase_admin.delete_app(cls.db.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")
