bjorn-hommel's picture
refactor
28183db
raw
history blame
2.21 kB
import os
import json
import random
import logging
import firebase_admin
from firebase_admin import credentials, firestore
public_creds_path = 'public_creds.json'
def load_credentials():
try:
with open(public_creds_path) as f:
credentials_dict = json.load(f)
secret = {
'private_key_id': os.environ.get('private_key_id'),
'private_key': os.environ.get('private_key').replace(r'\n', '\n')
}
credentials_dict.update(secret)
return credentials_dict
except Exception as e:
logging.error(f'Error while loading credentials: {e}')
return None
def connect_to_db(credentials_dict):
try:
cred = credentials.Certificate(credentials_dict)
if not firebase_admin._apps:
firebase_admin.initialize_app(cred)
logging.info('Established connection to db!')
return firestore.client()
except Exception as e:
logging.error(f'Error while connecting to db: {e}')
return None
def get_statements_from_db(db):
try:
document = db.collection('ItemDesirability').document('Items')
statements = document.get().to_dict()['statements']
logging.info(f'Retrieved {len(statements)} statements from db!')
return statements
except Exception as e:
logging.error(f'Error while retrieving items from db: {e}')
return None
def pick_random(input_list):
try:
return random.choice(input_list)
except Exception as e:
logging.error(f'Error while picking random statement: {e}')
return None
def write_to_db(db, payload):
try:
collection_ref = db.collection('ItemDesirability')
doc_ref = collection_ref.document('Responses')
doc = doc_ref.get()
if doc.exists:
doc_ref.update({
'Data': firestore.ArrayUnion([payload])
})
else:
doc_ref.set({
'Data': [payload]
})
logging.info(f'Sent payload to db!')
return True
except Exception as e:
logging.error(f'Error while sending payload to db: {e}')
return False