コード例 #1
0
    def delete_from_all_collections_by_user_id(self, db_name: str,
                                               user_id: str) -> dict:
        print('In Method: delete_by_user_id()')

        client = MongoClient(db_config.connection_string)
        db = client.get_database(db_name)

        try:
            collection_names = [
                collection for collection in db.collection_names()
                if not collection.startswith('system.')
            ]
            for collection_name in collection_names:
                db[collection_name].delete_many({'user_id': user_id})
        except Exception as e:
            print('Exception Thrown:')
            print(e)

            client.close()

            return {'success': False}

        client.close()

        return {'success': True}
コード例 #2
0
    def open_collection(self, db_name: str, collection_name: str):
        client = MongoClient(db_config.connection_string)
        db = client.get_database(db_name)
        collection = db[collection_name]

        return client, collection
コード例 #3
0
ファイル: app.py プロジェクト: raeray/persproj
app = Flask(__name__)
from flask_pymongo import MongoClient
SECRET_KEY = os.urandom(32)
app.config['SECRET_KEY'] = SECRET_KEY

# app.config["MONGO_URI"] = "mongodb://localhost:27017/Tags.Corpus"
# mongo = PyMongo(app)

#################################################
# HARD CODED CONSTANTS
#################################################

client = MongoClient('mongodb://127.0.0.1:27017')
settings = Settings()
db_geo = client.get_database(settings.GEO_DATABASE)
collection_reviews = db_geo.get_collection(settings.REVIEW_COLLECTION)
collection_topics = db_geo.get_collection(settings.TOPIC_COLLECTION)
collection_topic_values = db_geo.get_collection(
    settings.TOPIC_VALUES_COLLECTION)

db_business = client.get_database(settings.BUSINESS_DATABASE)
collection_business = db_business.get_collection(settings.BUSINESS_COLLECTION)

print('count', collection_business.count())

#################################################
# Geospatial query
#################################################

コード例 #4
0
ファイル: server.py プロジェクト: AkshatJain1/Simple-Login
jwt = JWTManager(app)

CORS(app)



'''
    Create Database from MongoDB Atlas
'''
try:
    user_password = open("./credentials.txt", "r").read()
except FileNotFoundError:
    sys.exit("No password found. You are not an authenticated user. Please ask Akshat for the password.")

client = MongoClient("mongodb+srv://dev_user:"******"@simple-login-5a4hv.gcp.mongodb.net/test?retryWrites=true&w=majority")
db = client.get_database("simpleloginreg")




'''
    Authentication functions
'''

def admin_required(fn):
    @wraps(fn)
    def wrapper(*args, **kwargs):
        verify_jwt_in_request()
        claims = get_jwt_claims()
        if claims['role'] != 'admin':
            return jsonify({'msg':'Admins only!'}), 403
コード例 #5
0
from bcrypt import hashpw, gensalt
from flask_pymongo import MongoClient
from .forms import RegisterForm, DetailForm, LoginForm
from flask_login import login_user, current_user, logout_user, login_required
from flask import render_template, redirect, url_for, flash, request, jsonify, Blueprint, session, make_response

# initialise blueprint
main = Blueprint('main', __name__)

# load env variables from the .env file
load_dotenv(verbose=True)

# mongodb configurations
uri = os.getenv('MONGO_URI')
client = MongoClient(uri, connect=False, connectTimeoutMS=30000)
mongo = client.get_database('offers')


# routes
@main.route('/adduser', methods=['POST', 'GET'])
@login_required
def adduser():
    form = RegisterForm(request.form)
    if request.method == 'POST':
        # user details
        name = form.name.data
        username = form.username.data
        email = form.email.data
        passw = form.password.data
        role = form.role.data
コード例 #6
0
import os
from bson.objectid import ObjectId
from bson.objectid import ObjectId
from dotenv.main import load_dotenv
from flask_pymongo import MongoClient

# MongoDB Details Saved In ENV
load_dotenv('../utils/.env')
MONGO_DB_CREDENTIAL = os.getenv('MONGO_DB_CREDENTIAL')

client = MongoClient(MONGO_DB_CREDENTIAL)
db = client.get_database('myFirstDatabase')
diabetic = db.Diabetic['username']
files = db.fs.files
chunks = db.fs.chunks


def save_users_images(mongo, image, username):
    mongo.save_file(image.filename, image)
    mongo.db.Diabetic.insert({
        'username': username,
        'filename': image.filename,
        'diabetic_result': 'None'
    })
    return True


def get_user_data(mongo, username):
    filename = (mongo.db.Diabetic.find_one_or_404({'username':
                                                   username}))['filename']
    return filename