Sybren A. Stüvel
e9cb235640
It looks like crap, but it allows you to edit the details and the members.
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
import logging
|
|
|
|
import attr
|
|
from flask import Blueprint, render_template, request, jsonify
|
|
import flask_wtf.csrf
|
|
import werkzeug.exceptions as wz_exceptions
|
|
|
|
from pillarsdk import User
|
|
|
|
import pillar.flask_extra
|
|
from pillar import current_app
|
|
from pillar.api.utils import authorization, str2id, gravatar
|
|
from pillar.web.system_util import pillar_api
|
|
from pillar.api.utils.authentication import current_user
|
|
|
|
|
|
from pillarsdk import Organization
|
|
|
|
log = logging.getLogger(__name__)
|
|
blueprint = Blueprint('pillar.web.organizations', __name__, url_prefix='/organizations')
|
|
|
|
|
|
@blueprint.route('/', endpoint='index')
|
|
def index(organization_id: str = None):
|
|
api = pillar_api()
|
|
|
|
organizations = Organization.all(api=api)
|
|
|
|
if not organization_id and organizations['_items']:
|
|
organization_id = organizations['_items'][0]._id
|
|
|
|
can_create_organization = current_user().has_cap('create-organization')
|
|
|
|
return render_template('organizations/index.html',
|
|
can_create_organization=can_create_organization,
|
|
organizations=organizations,
|
|
open_organization_id=organization_id)
|
|
|
|
|
|
@blueprint.route('/<organization_id>')
|
|
@pillar.flask_extra.vary_xhr()
|
|
def view_embed(organization_id: str):
|
|
if not request.is_xhr:
|
|
return index(organization_id)
|
|
|
|
api = pillar_api()
|
|
|
|
organization: Organization = Organization.find(organization_id, api=api)
|
|
|
|
om = current_app.org_manager
|
|
organization_oid = str2id(organization_id)
|
|
|
|
members = om.org_members(organization.members)
|
|
for member in members:
|
|
member['avatar'] = gravatar(member.get('email'))
|
|
member['_id'] = str(member['_id'])
|
|
|
|
can_edit = om.user_is_admin(organization_oid)
|
|
|
|
csrf = flask_wtf.csrf.generate_csrf()
|
|
|
|
return render_template('organizations/view_embed.html',
|
|
organization=organization,
|
|
members=members,
|
|
can_edit=can_edit,
|
|
seats_used=len(members) + len(organization.unknown_members),
|
|
csrf=csrf)
|
|
|
|
|
|
@blueprint.route('/create-new', methods=['POST'])
|
|
@authorization.require_login(require_cap='create-organization')
|
|
def create_new():
|
|
"""Creates a new Organization, owned by the currently logged-in user."""
|
|
|
|
user_id = current_user().user_id
|
|
log.info('Creating new organization for user %s', user_id)
|
|
|
|
name = request.form['name']
|
|
seat_count = int(request.form['seat_count'], 10)
|
|
|
|
org_doc = current_app.org_manager.create_new_org(name, user_id, seat_count)
|
|
|
|
return jsonify({'_id': org_doc['_id']}), 201
|