示例#1
0
def chat_list():
    return jsonify(rooms=Room.getall())
示例#2
0
def index():
    current_user.modules = list(Module.getall(current_user))
    return render_template('index.html', user=current_user, page='index', rooms=list(Room.getall()))
示例#3
0
class Config:
    DEBUG = True
    SECRET_KEY = 'abc'
    MODULE_PATH = './modules/'
    LOG_DIR = './logs/'

app = Flask(__name__)
app.config.from_object('mercury.Config')
app.config.from_envvar('MERCURY_CONFIG_PATH')
Module.PATH = app.config['MODULE_PATH']

pool = redis.ConnectionPool()
Room.REDIS = redis.Redis(connection_pool=pool)
User.REDIS = redis.Redis(connection_pool=pool)

for r in Room.getall():
    l = logging.getLogger(r.name)
    h = logging.FileHandler(os.path.join(app.config['LOG_DIR'], r.name + '.log'))
    l.setLevel(logging.INFO)
    l.addHandler(h)


login_manager = LoginManager()
login_manager.login_view = 'login'
login_manager.login_message = u'Please log in to access this page.'


@login_manager.user_loader
def load_user(username):
    try:
        return User(username)