db.py

import os
from pymongo import MongoClient

# define on heroku settings tab
MONGODB_URI = os.environ['MONGODB_URI']

cluster = MongoClient(MONGODB_URI)
db = cluster['QnA']
collection = db['QnA']


def insert(message, response):
    # EST
    time = (datetime.fromtimestamp(int(str(message['timestamp'])[:-3])) - timedelta(hours=4)).strftime(
        '%Y-%m-%d %H:%M:%S')
    collection.insert_one({"question": message['message'].get('text'),
                           "answer": response, 'time': time, 'source':'Facebook Messenger'})