def fillup_db():
DB['rooms'].append(Room(42, 1, 'closed', 1000))
DB['rooms'].append(Room(7, 1, 'available', 50))
DB['rooms'].append(Room(69, 7, 'available', 9000))
DB['rooms'].append(Room(11, 3, 'closed', 10))
DB['tenants'].append(Tenant('Neo', 'zz0101', 33, 'M',
{
"city": "Zeon",
"street": "Street 255"
}, 42))
DB['tenants'].append(Tenant('Janis', 'bb2222', 40, 'W',
{
"city": "Los Angeles",
"street": "Beverly Hills, 90210"
}, 7))
DB['tenants'].append(Tenant('Sherlock', 'cc3333', 51, 'M',
{
"city": "London",
"street": "Bakerstreet, 221B"
}, 11))
DB['staff'].append(Staff('Taylor', 'aa1111', 'waiter', 100))
DB['staff'].append(Staff('Hannibal', 'et0000', 'cook', 1000))
DB['staff'].append(Staff('Hilton', 'ad1234', 'Administrator', 9000))
def main():
tenant_list = all_tenant()
for tenant in tenant_list:
if can_insert(tenant['id']):
tn = Tenant(tenant['id'], tenant['name'], tenant['user_id'])
db.add(tn)
print "tenant :", tenant['name'], " added to db"
db.flush()
db.commit()
def save(self, commit=True):
tenant = Tenant(rent_date=self.cleaned_data['rentDate'],\
remind_date=self.cleaned_data['remindDate'],rent_money=self.cleaned_data['rentMoney'],\
house_number=self.cleaned_data['houseNumber'],unit=self.cleaned_data['unit'],\
community=self.cleaned_data['community'],mobile_number=self.cleaned_data['mobileNumber'],\
sex=self.cleaned_data['sex'],name=self.cleaned_data['name'])
tenant.status = '1'
tenant.user_id = self.userId
tenant.done_date = datetime.datetime.now()
if commit:
tenant.save()
def post(self):
data = data_valid_for('POST')
logging.debug(data)
tenants_already_exist = list(
filter(lambda t: data['passport_id'] == t.passport_id,
DB['tenants']))
if not tenants_already_exist:
DB['tenants'].append(Tenant(**data))
new_passport_id = str(DB['tenants'][-1].passport_id)
location = {"Location": f'/api/v0.1/tenants/{new_passport_id}'}
return {"message": "tenant was added successfully"}, 201, location
msg = f"tenant with passport_id {data['passport_id']} already exists"
return {"message": msg}, 400
def __init__(self, args):
self.args = args
self.tenant = Tenant(
self.args.username,
self.args.api_key,
Tenant.SMALL, # this doesn't mean anything for our purposes
self.args.auth_endpoint,
)
self.tenant.get_token()
print "%s has token %s" % (self.tenant.id, self.tenant.get_token())
self.client = DesignateClient(
client=HttpSession(self.args.endpoint),
tenant=self.tenant,
use_project_id=self.args.no_repose,
tenant_id_in_url=not self.args.no_repose,
)
def userRegisteration(request: User, db: Session = Depends(get_db)):
id_uid = id_uuids()
organization = Organization(id=id_uid, name=request.organization_name)
db.add(organization)
db.commit()
db.refresh(organization)
id_uid = id_uuids()
tenant = Tenant(id=id_uid,
name=request.tenant_name,
company_logo=request.company_logo,
organization_id=organization.id)
db.add(tenant)
db.commit()
db.refresh(tenant)
role = db.query(
models.Role).filter(models.Role.name == "tenentAdmin").first()
if role:
print(role.id)
else:
role = Role(id=id_uid, name="tenentAdmin")
db.add(role)
db.commit()
db.refresh(role)
# atuo generate password
a = randoms()
password = str(a)
Tenantuser = TenantUser(id=id_uid,
username=request.username,
role_id=role.id,
password=password,
tenant_id=tenant.id)
db.add(Tenantuser)
db.commit()
db.refresh(Tenantuser)
return {organization, tenant, Tenantuser}
def userRegisteration(request: User, db: Session = Depends(get_db)):
id_uid = id_uuids()
organization = Organization(id=id_uid, name=request.organization_name)
db.add(organization)
db.commit()
db.refresh(organization)
id_uid = id_uuids()
tenant = Tenant(id=id_uid,
name=request.tenant_name,
organization_id=organization.id)
db.add(tenant)
db.commit()
db.refresh(tenant)
role = db.query(models.Role).filter(models.Role.name == "admin").first()
if role:
role_id = role.id
else:
role_id = Role(id=id_uid, name="tenentAdmin")
db.add(role_id)
db.commit()
db.refresh(role_id)
print(role_id)
tenantuser = TenantUser(id=id_uid,
username=request.username,
first_name=request.firstname,
last_name=request.lastname,
email=request.email,
role_id=role_id,
tenant_id=tenant.id)
db.add(tenantuser)
db.commit()
db.refresh(tenantuser)
return {organization, tenant, tenantuser}
def validate_checkdomain(request):
TenantObj = Tenant().getUserBySubdomain(request.GET.get('company_domain'))
if TenantObj == None:
return HttpResponse(json.dumps(True))
return HttpResponse(json.dumps(False))
#Push sample data with GraphQL #Drop existing tables if any before running #initialize db with first tenants and sensors import random from models import engine, db_session, Base, Tenant, Data, Sensor Base.metadata.create_all(bind=engine) #add first user kate = Tenant(name='kate') db_session.add(kate) #add second user kate_multitenancy = Tenant(name='kate_multitenancy') db_session.add(kate_multitenancy) #add sensors t = Sensor(name='temperature') db_session.add(t) e = Sensor(name='electricity') db_session.add(e) p = Sensor(name='pressure') db_session.add(p) users = [kate, kate_multitenancy] sensors = [e,p,t] #add 100 entries n = 100 for i in range(n):
# save a report when the test finishes
persistence.setup_persistence()
locust.config.RESET_STATS_AFTER_HATCHING = CONFIG.reset_stats
# use digaas to externally poll the nameserver(s) to record zone propagation
# times through Designate's stack
if CONFIG.use_digaas and not insight.is_slave():
_digaas_client = digaas_integration.DigaasClient(CONFIG.digaas_endpoint)
digaas_integration.setup(_digaas_client)
if not insight.is_master():
# TODO: the tenant id is actually the username. it should be named so.
SMALL_TENANTS = [
Tenant(id=id, api_key=api_key, type=Tenant.SMALL)
for id, api_key in CONFIG.small_tenants
]
LARGE_TENANTS = [
Tenant(id=id, api_key=api_key, type=Tenant.LARGE)
for id, api_key in CONFIG.large_tenants
]
ALL_TENANTS = SMALL_TENANTS + LARGE_TENANTS
# the greenlet_manager keeps track of greenlets spawned for polling
# todo: it's hard to ensure cleanup_greenlets gets run before the stats
# are persisted to a file...
# ensure cleanup when the test is stopped
locust.events.locust_stop_hatching += \
lambda: GreenletManager.get().cleanup_greenlets()