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()
