from mongokit import Document from project.services.database import Database from project import utils from project.config import config from bson.objectid import ObjectId import random Invites = Database['Invites'] connection = Database.connection() @connection.register class Invite(Document): __collection__ = 'Invites' __database__ = config['DATABASE_NAME'] structure = { 'producerObjectId': ObjectId, 'consumerObjectId': ObjectId, 'code': basestring, 'scratchedOut': bool } default_values = {'scratchedOut': False} required_fields = ['code'] basic_fields = { '_id', 'code', 'producerObjectId', 'consumerObjectId', 'scratchedOut' } use_dot_notation = True
from flask import Flask from flask import jsonify from flask.ext.api.status import HTTP_404_NOT_FOUND from json_rest import CustomJSONEncoder from config import config from project.services.database import Database Database.connect(config) from project.services.elastic import Elastic from project.services.auth import Auth from project.services.cors import Cors from project.services.api import API app = Flask(__name__) Cors.init_app(app, config) app.config['CORS_HEADERS'] = 'Content-Type' app.secret_key = config['SECRET_KEY'] app.debug = config['DEBUG'] app.json_encoder = CustomJSONEncoder # Init services Elastic.connect(config) Auth.init_app(app, config)
from bson import ObjectId from flask_login import current_user from flask_mongokit import Document from models.user import findMultipleUsers, findUserByID from project.services.database import Database from project.config import config from project.services.auth import Auth connection = Database.connection() Admins = Database['Admins'] @connection.register class Admin(Document): __collection__ = 'Admins' __database__ = config['DATABASE_NAME'] structure = { "activatedBy": ObjectId } def basic_user_with_email(user): basic_user = user.to_basic() basic_user["email"] = user["email"] return basic_user def activate_user(user_id): user = findUserByID(user_id) user["permissionLevel"] = Auth.USER user.save()
from project.config import config from project.services.database import Database db_name = raw_input('Enter the DB name on which you wish to operate: ') config['DATABASE_NAME'] = db_name Database.connect(config) import models.invite import models.user def make_invite(): inv = models.invite.create_invite(None) print 'New invite code is: ', inv['code'] def set_permission(user_id, level): user_data = models.user.findUserByID(user_id) user_data['permissionLevel'] = level user_data.save()