def get_by_id(id, fields=None, load_org=False):
from pritunl import organization
user = User(id=id, fields=fields)
if not user:
return None
if load_org:
user.org = organization.get_by_id(user.id)
return user
def get_by_id(id, fields=None, load_org=False):
from pritunl import organization
user = User(id=id, fields=fields)
if not user:
return None
if load_org:
user.org = organization.get_by_id(user.id)
return user
def reserve_pooled_user(org,
name=None,
email=None,
pin=None,
type=CERT_CLIENT,
groups=None,
auth_type=None,
yubico_id=None,
disabled=None,
resource_id=None,
dns_servers=None,
dns_suffix=None,
bypass_secondary=None,
client_to_client=None,
port_forwarding=None):
doc = {}
if name is not None:
doc['name'] = name
if email is not None:
doc['email'] = email
if pin is not None:
doc['pin'] = pin
if type is not None:
doc['type'] = type
if groups is not None:
doc['groups'] = groups
if auth_type is not None:
doc['auth_type'] = auth_type
if yubico_id is not None:
doc['yubico_id'] = yubico_id
if disabled is not None:
doc['disabled'] = disabled
if resource_id is not None:
doc['resource_id'] = resource_id
if dns_servers is not None:
doc['dns_servers'] = dns_servers
if dns_suffix is not None:
doc['dns_suffix'] = dns_suffix
if bypass_secondary is not None:
doc['bypass_secondary'] = bypass_secondary
if client_to_client is not None:
doc['client_to_client'] = client_to_client
if port_forwarding is not None:
doc['port_forwarding'] = port_forwarding
doc = User.collection.find_and_modify(
{
'org_id': org.id,
'type': {
CERT_SERVER: CERT_SERVER_POOL,
CERT_CLIENT: CERT_CLIENT_POOL,
}[type],
}, {
'$set': doc,
},
new=True)
if doc:
return User(org=org, doc=doc)
def reserve_pooled_user(org, name=None, email=None,
type=CERT_CLIENT, disabled=None, resource_id=None):
doc = {}
if name is not None:
doc['name'] = name
if email is not None:
doc['email'] = email
if type is not None:
doc['type'] = type
if disabled is not None:
doc['disabled'] = disabled
if resource_id is not None:
doc['resource_id'] = resource_id
doc = User.collection.find_and_modify({
'org_id': org.id,
'type': {
CERT_SERVER: CERT_SERVER_POOL,
CERT_CLIENT: CERT_CLIENT_POOL,
}[type],
}, {
'$set': doc,
}, new=True)
if doc:
return User(org=org, doc=doc)
def find_user_auth(name, auth_type):
from pritunl import organization
spec = {
'name': name,
'auth_type': auth_type,
}
usr = User(None, spec=spec)
if not usr:
return None
usr.org = organization.get_by_id(usr.org_id)
if not usr.org:
return None
return usr
def find_user(org, name=None, type=None, resource_id=None):
spec = {
'org_id': org.id,
}
if name is not None:
spec['name'] = name
if type is not None:
spec['type'] = type
if resource_id is not None:
spec['resource_id'] = resource_id
return User(org, spec=spec)
def find_user_auth(name, auth_type):
from pritunl import organization
spec = {
'name': {
'$regex': '^%s$' % re.escape(name),
'$options': 'i',
},
'auth_type': auth_type,
}
usr = User(None, spec=spec)
if not usr:
return None
usr.org = organization.get_by_id(usr.org_id)
if not usr.org:
return None
return usr
def find_user(org, name=None, type=None, resource_id=None):
spec = {
'org_id': org.id,
}
if name is not None:
spec['name'] = {
'$regex': '^%s$' % re.escape(name),
'$options': 'i',
}
if type is not None:
spec['type'] = type
if resource_id is not None:
spec['resource_id'] = resource_id
return User(org, spec=spec)
def get_user(org, id, fields=None):
return User(org=org, id=id, fields=fields)
