mongoqueries.py

import pymongo 
import vidtotext
client = None
db = None
collection = None

def mongo_connect():
    global client, collection, db
    connection_string = "mongodb://mongoadmin:mongopassword@localhost:35945/"
    client = pymongo.MongoClient(connection_string)
    db = client["slang_data"]
    collection = db["word_dataset"]
    print("we got a connection ;]")

def mongo_disconnect():
    global client
    if client == None:
        print("No client created stupid")
    else:
        client.close()
        print("*indian accent* he disconnected (the client)")

def insert_transcripts(transcripts):
    global collection
    if collection == None:
        print("connect to the db my brotha")
    else:
        collection.insert_many(transcripts)
        print("it went in :D")

def get_transcripts():
    global collection
    if collection == None:
        print("collection could not be found")
        return None
    else:
        documents = collection.find()
        for doc in documents:
            print(doc)
        return documents
            
def delete_data():
    global client, db
    if db == None:
        print("db does not exist")
    else: 
        client.drop_database(db)
        print("db deleted")

'''
mongo_connect()
delete_data()
get_transcripts()
mongo_disconnect()
'''