def checkOrganizations():
json = request.json
for org_id in json['organizations']:
org = Organization.queryById(org_id)
if org == None:
return False
return True
def _organizations_create():
orgs = current_user.orgs.all()
create_form = CreateOrganizationForm()
edit_form = EditOrganizationForm()
if create_form.validate_on_submit():
org = Organization(name=create_form.name.data,
description=create_form.description.data)
db.session.add(org)
for main_org in create_form.main_orgs.data:
org.add_organization(main_org)
current_user.add_organization(org)
db.session.commit()
flash('Your organization has been created')
return redirect(url_for('organization.organizations'))
return render_template('organization/organizations.html',
orgs=orgs,
create_form=create_form,
edit_form=edit_form)
def create_organization(organization_name: str, address: str) -> Organization:
"""Add new organization.
Returns:
Organization -- New organization.
"""
return OrganizationRepository.create(
Organization(name=organization_name,
name_slug=get_name_slug(organization_name),
address=address,
status=OrganizationStatus.ACTIVE))
def list_organizations():
allOrgs = Organization.queryAll()
res = [{'_id': org._id, 'name': org.name} for org in allOrgs]
return {'organizations': res}
def find_organization(org_id):
org = Organization.queryById(org_id)
if org == None:
abort(404)
return org.__dict__
def create_organization(name, url, category):
org = Organization(name, url, category=category)
org.save()
return "Success"
def session(db, request, seed_data):
"""Create a new database session for the tests"""
connection = db.engine.connect()
transaction = connection.begin()
options = dict(bind=connection, binds={})
session = db.create_scoped_session(options=options)
db.session = session
client_industry_seed = seed_data['client_industry']
client_industry_1 = ClientIndustry(**client_industry_seed)
organization_seed = seed_data['organization']
organization_1 = Organization(**organization_seed)
user_account_seed = seed_data['user_account']
user_account_seed['organization_id'] = organization_1.id
user_1 = UserAccount(**user_account_seed)
client_seed = seed_data['client']
client_seed['organization_id'] = organization_1.id
client_seed['client_industry_id'] = client_industry_1.id
client_1 = Client(**client_seed)
line_of_service_seed = seed_data['line_of_service']
line_of_service_seed['organization_id'] = organization_1.id
line_of_service_1 = LineOfService(**line_of_service_seed)
engagement_seed = seed_data['engagement']
engagement_seed['client_id'] = client_1.id
engagement_seed['line_of_service_id'] = line_of_service_1.id
engagement_seed['organization_id'] = organization_1.id
engagement_1 = Engagement(**engagement_seed)
employee_seed = seed_data['employee']
employee_seed['user_account_id'] = user_1.id
employee_seed['organization_id'] = organization_1.id
employee_1 = Employee(**employee_seed)
administrator_role = RoleRepository.get_by_name('administrator')
if not administrator_role:
role_1 = Role(name='Administrator', normalized_name='administrator')
db.session.add(role_1)
db.session.add(organization_1)
db.session.add(user_1)
db.session.add(client_industry_1)
db.session.add(client_1)
db.session.add(line_of_service_1)
db.session.add(engagement_1)
db.session.add(employee_1)
db.session.commit()
def teardown():
"""Clean up test suite."""
transaction.rollback()
connection.close()
session.remove()
request.addfinalizer(teardown)
return session
from app import db, create_app
from app.organization.models import Organization
from app.auth.models import User
username = '******'
main = Organization(name='main_org')
if __name__ == "__main__":
app = create_app()
with app.app_context():
db.session.add(main)
db.session.commit()