# userGroup.py
# Endpoints for managing returning corresponding user/group relationships
# Monica Mahon, 11/26/2019
from flask import Flask, request, render_template, redirect, url_for, session, jsonify
from flask_restful import Resource, request
import bcrypt
from SeniorProject import database
from bson.objectid import ObjectId
db = database.conn_DB()
class UserGroup(Resource):
def get(self, f_id, g_id):
res = {'msg': [], 'err': []}
facilities = db.facilities
users = db.users
current_facility = facilities.find_one({'_id': ObjectId(f_id)})
if current_facility:
groups = current_facility.get('groups')
group_id = next(g for g in groups
if g['_id'] == ObjectId(g_id))['_id']
if group_id:
users_in_group = list(
users.find({'groupID': ObjectId(g_id)}, {
'password': 0,
'answer': 0
}))
return jsonify(users_in_group)
else:
from SeniorProject.resources.facility import Facility
from SeniorProject.resources.room import Room
from SeniorProject.resources.building import Building
from SeniorProject.resources.maintenance import Maintenance
from SeniorProject.new_json_encoder import New_JSON_Encoder
from SeniorProject.resources.group import Group
from SeniorProject.resources.userGroup import UserGroup
from SeniorProject.user_check import *
from SeniorProject.resources.userFacility import UserFacility
from SeniorProject.resources.userMaintenance import UserMaintenance
app = Flask(__name__)
api = Api(app)
app.json_encoder = New_JSON_Encoder
db = conn_DB()
@app.route('/')
@app.route('/home')
def home():
return render_template('Home.html')
@app.route('/login', methods=['POST'])
def login():
data = request.json
users = db.users
login_user = users.find_one({'email':
data.get('email')}) # find user in db